Public Pre-Trained Models Device Support¶
Model Name |
CPU |
GPU |
|---|---|---|
aclnet |
YES |
|
aclnet-int8 |
YES |
|
alexnet |
YES |
YES |
anti-spoof-mn3 |
YES |
YES |
background-matting-mobilenetv2 |
YES |
|
bert-base-ner |
YES |
|
brain-tumor-segmentation-0001 |
YES |
|
brain-tumor-segmentation-0002 |
YES |
|
caffenet |
YES |
YES |
cocosnet |
YES |
|
colorization-siggraph |
YES |
YES |
colorization-v2 |
YES |
YES |
common-sign-language-0001 |
YES |
|
convnext-tiny |
YES |
|
ctdet_coco_dlav0_512 |
YES |
YES |
ctpn |
YES |
YES |
deblurgan-v2 |
YES |
YES |
deeplabv3 |
YES |
YES |
densenet-121 |
YES |
YES |
densenet-121-tf |
YES |
YES |
detr-resnet50 |
YES |
|
dla-34 |
YES |
YES |
drn-d-38 |
YES |
|
efficientdet-d0-tf |
YES |
YES |
efficientdet-d1-tf |
YES |
YES |
efficientnet-b0 |
YES |
YES |
efficientnet-b0-pytorch |
YES |
YES |
efficientnet-v2-b0 |
YES |
|
efficientnet-v2-s |
YES |
|
erfnet |
YES |
|
f3net |
YES |
YES |
face-detection-retail-0044 |
YES |
|
face-recognition-resnet100-arcface-onnx |
YES |
|
faceboxes-pytorch |
YES |
|
facenet-20180408-102900 |
YES |
YES |
fast-neural-style-mosaic-onnx |
YES |
YES |
faster_rcnn_inception_resnet_v2_atrous_coco |
YES |
|
faster_rcnn_resnet50_coco |
YES |
YES |
fastseg-large |
YES |
|
fastseg-small |
YES |
|
fbcnn |
YES |
YES |
fcrn-dp-nyu-depth-v2-tf |
YES |
|
forward-tacotron-duration-prediction |
YES |
|
forward-tacotron-regression |
YES |
|
gmcnn-places2-tf |
YES |
YES |
googlenet-v1 |
YES |
YES |
googlenet-v1-tf |
YES |
YES |
googlenet-v2 |
YES |
YES |
googlenet-v2-tf |
YES |
YES |
googlenet-v3 |
YES |
YES |
googlenet-v3-pytorch |
YES |
YES |
googlenet-v4-tf |
YES |
YES |
gpt-2 |
YES |
|
hbonet-0.25 |
YES |
YES |
hbonet-1.0 |
YES |
YES |
higher-hrnet-w32-human-pose-estimation |
YES |
|
hrnet-v2-c1-segmentation |
YES |
|
human-pose-estimation-3d-0001 |
YES |
YES |
hybrid-cs-model-mri |
YES |
|
i3d-rgb-tf |
YES |
|
inception-resnet-v2-tf |
YES |
YES |
levit-128s |
YES |
|
license-plate-recognition-barrier-0007 |
YES |
|
mask_rcnn_inception_resnet_v2_atrous_coco |
YES |
|
mask_rcnn_resnet50_atrous_coco |
YES |
|
midasnet |
YES |
|
mixnet-l |
YES |
YES |
mobilefacedet-v1-mxnet |
YES |
YES |
mobilenet-ssd |
YES |
YES |
mobilenet-v1-0.25-128 |
YES |
YES |
mobilenet-v1-1.0-224 |
YES |
YES |
mobilenet-v1-1.0-224-tf |
YES |
YES |
mobilenet-v2 |
YES |
YES |
mobilenet-v2-1.0-224 |
YES |
YES |
mobilenet-v2-1.4-224 |
YES |
YES |
mobilenet-v2-pytorch |
YES |
YES |
mobilenet-v3-large-1.0-224-tf |
YES |
YES |
mobilenet-v3-small-1.0-224-tf |
YES |
YES |
mobilenet-v3-large-1.0-224-paddle |
YES |
YES |
mobilenet-v3-small-1.0-224-paddle |
YES |
YES |
mobilenet-yolo-v4-syg |
YES |
|
modnet-photographic-portrait-matting |
YES |
YES |
modnet-webcam-portrait-matting |
YES |
YES |
mozilla-deepspeech-0.6.1 |
YES |
|
mozilla-deepspeech-0.8.2 |
YES |
|
mtcnn-o |
YES |
|
mtcnn-p |
YES |
|
mtcnn-r |
YES |
|
nanodet-m-1.5x-416 |
YES |
|
nanodet-plus-m-1.5x-416 |
YES |
|
netvlad-tf |
YES |
|
nfnet-f0 |
YES |
YES |
ocrnet-hrnet-w48-paddle |
YES |
|
octave-resnet-26-0.25 |
YES |
YES |
open-closed-eye-0001 |
YES |
|
pelee-coco |
YES |
YES |
pspnet-pytorch |
YES |
|
quartznet-15x5-en |
YES |
|
regnetx-3.2gf |
YES |
YES |
repvgg-a0 |
YES |
YES |
repvgg-b1 |
YES |
YES |
repvgg-b3 |
YES |
YES |
resnest-50-pytorch |
YES |
YES |
resnet-18-pytorch |
YES |
YES |
resnet-34-pytorch |
YES |
YES |
resnet-50-pytorch |
YES |
YES |
resnet-50-tf |
YES |
YES |
retinaface-resnet50-pytorch |
YES |
YES |
retinanet-tf |
YES |
|
rexnet-v1-x1.0 |
YES |
|
rfcn-resnet101-coco-tf |
YES |
YES |
robust-video-matting-mobilenetv3 |
||
se-inception |
YES |
YES |
se-resnet-50 |
YES |
YES |
se-resnext-50 |
YES |
YES |
shufflenet-v2-x0.5 |
YES |
YES |
shufflenet-v2-x1.0 |
YES |
YES |
single-human-pose-estimation-0001 |
YES |
YES |
Sphereface |
YES |
YES |
squeezenet1.0 |
YES |
YES |
squeezenet1.1 |
YES |
YES |
ssd-resnet34-1200-onnx |
YES |
YES |
ssd300 |
YES |
YES |
ssd512 |
YES |
YES |
ssd_mobilenet_v1_coco |
YES |
YES |
ssd_mobilenet_v1_fpn_coco |
YES |
YES |
ssdlite_mobilenet_v2 |
YES |
YES |
swin-tiny-patch4-window7-224 |
YES |
YES |
t2t-vit-14 |
YES |
|
text-recognition-resnet-fc |
YES |
YES |
ultra-lightweight-face-detection-rfb-320 |
YES |
YES |
ultra-lightweight-face-detection-slim-320 |
YES |
YES |
vehicle-license-plate-detection-barrier-0123 |
YES |
YES |
vehicle-reid-0001 |
YES |
YES |
vgg16 |
YES |
YES |
vgg19 |
YES |
YES |
vitstr-small-patch16-224 |
YES |
|
wav2vec2-base |
YES |
|
wavernn-rnn |
YES |
YES |
wavernn-upsampler |
YES |
|
yolact-resnet50-fpn-pytorch |
YES |
YES |
yolo-v1-tiny-tf |
YES |
YES |
yolo-v2-tf |
YES |
YES |
yolo-v2-tiny-tf |
YES |
YES |
yolo-v3-onnx |
YES |
|
yolo-v3-tf |
YES |
YES |
yolo-v3-tiny-onnx |
YES |
|
yolo-v3-tiny-tf |
YES |
YES |
yolo-v4-tf |
YES |
YES |
yolo-v4-tiny-tf |
YES |
YES |
yolof |
YES |
|
yolox-tiny |
YES |