Skip to content

Latest commit

 

History

History
125 lines (124 loc) · 4.3 KB

device_support.md

File metadata and controls

125 lines (124 loc) · 4.3 KB

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