Public Pre-Trained Models Device Support

View on GitHub

Model NameCPUGPU
aclnetYES
aclnet-int8YES
anti-spoof-mn3YESYES
background-matting-mobilenetv2YES
bert-base-nerYES
brain-tumor-segmentation-0002YES
cocosnetYES
colorization-siggraphYESYES
colorization-v2YESYES
common-sign-language-0001YES
convnext-tinyYES
ctdet_coco_dlav0_512YESYES
ctpnYESYES
deeplabv3YESYES
densenet-121-tfYESYES
detr-resnet50YES
dla-34YESYES
drn-d-38YES
efficientdet-d0-tfYESYES
efficientdet-d1-tfYESYES
efficientnet-b0YESYES
efficientnet-b0-pytorchYESYES
efficientnet-v2-b0YES
efficientnet-v2-sYES
erfnetYES
f3netYESYES
face-recognition-resnet100-arcface-onnxYES
faceboxes-pytorchYES
facenet-20180408-102900YESYES
fast-neural-style-mosaic-onnxYESYES
faster_rcnn_inception_resnet_v2_atrous_cocoYES
faster_rcnn_resnet50_cocoYESYES
fastseg-largeYES
fastseg-smallYES
fbcnnYESYES
fcrn-dp-nyu-depth-v2-tfYES
forward-tacotron-duration-predictionYES
forward-tacotron-regressionYES
gmcnn-places2-tfYESYES
googlenet-v1-tfYESYES
googlenet-v2-tfYESYES
googlenet-v3YESYES
googlenet-v3-pytorchYESYES
googlenet-v4-tfYESYES
gpt-2YES
hbonet-0.25YESYES
hbonet-1.0YESYES
higher-hrnet-w32-human-pose-estimationYES
hrnet-v2-c1-segmentationYES
human-pose-estimation-3d-0001YESYES
hybrid-cs-model-mriYES
i3d-rgb-tfYES
inception-resnet-v2-tfYESYES
levit-128sYES
license-plate-recognition-barrier-0007YES
mask_rcnn_inception_resnet_v2_atrous_cocoYES
mask_rcnn_resnet50_atrous_cocoYES
midasnetYES
mixnet-lYESYES
mobilenet-v1-0.25-128YESYES
mobilenet-v1-1.0-224-tfYESYES
mobilenet-v2-1.0-224YESYES
mobilenet-v2-1.4-224YESYES
mobilenet-v2-pytorchYESYES
mobilenet-v3-large-1.0-224-tfYESYES
mobilenet-v3-small-1.0-224-tfYESYES
mobilenet-yolo-v4-sygYES
modnet-photographic-portrait-mattingYESYES
modnet-webcam-portrait-mattingYESYES
mozilla-deepspeech-0.6.1YES
mozilla-deepspeech-0.8.2YES
nanodet-m-1.5x-416YES
nanodet-plus-m-1.5x-416YES
netvlad-tfYES
nfnet-f0YESYES
open-closed-eye-0001YES
pspnet-pytorchYES
quartznet-15x5-enYES
regnetx-3.2gfYESYES
repvgg-a0YESYES
repvgg-b1YESYES
repvgg-b3YESYES
resnest-50-pytorchYESYES
resnet-18-pytorchYESYES
resnet-34-pytorchYESYES
resnet-50-pytorchYESYES
resnet-50-tfYESYES
retinaface-resnet50-pytorchYESYES
retinanet-tfYES
rexnet-v1-x1.0YES
rfcn-resnet101-coco-tfYESYES
robust-video-matting-mobilenetv3
shufflenet-v2-x1.0YESYES
single-human-pose-estimation-0001YESYES
ssd-resnet34-1200-onnxYESYES
ssd_mobilenet_v1_cocoYESYES
ssd_mobilenet_v1_fpn_cocoYESYES
ssdlite_mobilenet_v2YESYES
swin-tiny-patch4-window7-224YESYES
t2t-vit-14YES
text-recognition-resnet-fcYESYES
ultra-lightweight-face-detection-rfb-320YESYES
ultra-lightweight-face-detection-slim-320YESYES
vehicle-license-plate-detection-barrier-0123YESYES
vehicle-reid-0001YESYES
vitstr-small-patch16-224YES
wav2vec2-baseYES
wavernn-rnnYESYES
wavernn-upsamplerYES
yolact-resnet50-fpn-pytorchYESYES
yolo-v1-tiny-tfYESYES
yolo-v2-tfYESYES
yolo-v2-tiny-tfYESYES
yolo-v3-onnxYES
yolo-v3-tfYESYES
yolo-v3-tiny-onnxYES
yolo-v3-tiny-tfYESYES
yolo-v4-tfYESYES
yolo-v4-tiny-tfYESYES
yolofYES
yolox-tinyYES