Public Pre-Trained Models Device Support#

Model Name

CPU

GPU

aclnet

YES

aclnet-int8

YES

anti-spoof-mn3

YES

YES

background-matting-mobilenetv2

YES

bert-base-ner

YES

brain-tumor-segmentation-0002

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

deeplabv3

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-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-tf

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

mobilenet-v1-0.25-128

YES

YES

mobilenet-v1-1.0-224-tf

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-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

nanodet-m-1.5x-416

YES

nanodet-plus-m-1.5x-416

YES

netvlad-tf

YES

nfnet-f0

YES

YES

open-closed-eye-0001

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

shufflenet-v2-x1.0

YES

YES

single-human-pose-estimation-0001

YES

YES

ssd-resnet34-1200-onnx

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

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