List of the Supported Models

Table 1 Supported models

Type

Framework

Postprocessing Dynamic Library

How to Obtain

YOLOv3

TensorFlow

  • (tensorinfer) modelpostprocessors/libyolov3postprocess.so
  • (modelinfer) libMpYOLOv3PostProcessor.so

Model code

Model code

ResNet-50

TensorFlow

  • (tensorinfer) modelpostprocessors/libresnet50postprocess.so
  • (modelinfer) libresnet50postprocessor.so

Model code

Model code

Faster Rcnn

TensorFlow

  • (tensorinfer) modelpostprocessors/libfasterrcnnpostprocess.so
  • (modelinfer) libfasterrcnnpostprocessor.so

Model code

MindSpore

(tensorinfer) modelpostprocessors/libfasterrcnnpostprocess.so

Model code

Model code

YOLOv4

PyTorch

(tensorinfer) modelpostprocessors/libyolov3postprocess.so

Model code

SSD-VGG16

Caffe

  • (tensorinfer) modelpostprocessors/libssdvgg16postprocess.so
  • (modelinfer) libssdvggpostprocessor.so

None

SSD MobileNet v1 FPN

TensorFlow

  • (tensorinfer) modelpostprocessors/libssdmobilenetv1fpnpostprocess.so
  • (modelinfer) libssdmobilenetfpnpostprocessor.so

CRNN

TensorFlow

  • (tensorinfer) modelpostprocessors/libcrnnpostprocess.so
  • (modelinfer) libcrnnpostprocessor.so

YOLOv5

PyTorch

  • (tensorinfer) modelpostprocessors/libyolov3postprocess.so
  • (modelinfer) libMpYOLOv5PostProcessor.so

FasterRCNN-FPN/CascadeRCNN-FPN

PyTorch

(tensorinfer) modelpostprocessors/libfasterrcnnpostprocess.so

ResNet-18

TensorFlow

(tensorinfer) modelpostprocessors/libresnet50postprocess.so

CTPN

TensorFlow

  • (tensorinfer) modelpostprocessors/libctpnpostprocess.so
  • (modelinfer) libMpCtpnPostProcessor.so

Model code

CTPN

MindSpore

  • (tensorinfer) modelpostprocessors/libctpnpostprocess.so
  • (modelinfer) libMpCtpnPostProcessor.so

Model code

Model code

DeepLabv3

MindSpore

(tensorinfer) modelpostprocessors/libdeeplabv3post.so

Model code

Model code

BERT-Base (Uncased)

TensorFlow

(tensorinfer) modelpostprocessors/libresnet50postprocess.so

Model code

DeepLabv3+

PyTorch

(tensorinfer) modelpostprocessors/libdeeplabv3post.so

Model code

U-Net

MindSpore

(tensorinfer) modelpostprocessors/libunetmindsporepostprocess.so

Model code

Mask R-CNN

PyTorch

(tensorinfer) modelpostprocessors/libmaskrcnnmindsporepost.so

Model code

FaceNet

TensorFlow

Postprocessing is not required.

Model code

SSD MobileNet v1 FPN

MindSpore

(tensorinfer) modelpostprocessors/libSsdMobilenetFpn_MindsporePost.so

Model code

Model code

OpenPose

TensorFlow

(tensorinfer) modelpostprocessors/libopenposepostprocess.so

None

UNet++

MindSpore

(tensorinfer) modelpostprocessors/libunetmindsporepostprocess.so

None

RetinaNet

TensorFlow

(tensorinfer) modelpostprocessors/libretinanetpostprocess.so

Model code

HigherHRnet

PyTorch

(tensorinfer) modelpostprocessors/libhigherhrnetpostprocess.so

Model code

YoloV7Detection

PyTorch

None

None

PPYOLOEPlusDetection

Paddle

None

None