List of the Supported Models
Type |
Framework |
Postprocessing Dynamic Library |
How to Obtain |
|---|---|---|---|
YOLOv3 |
TensorFlow |
|
|
ResNet-50 |
TensorFlow |
|
|
Faster Rcnn |
TensorFlow |
|
|
MindSpore |
(tensorinfer) modelpostprocessors/libfasterrcnnpostprocess.so |
||
YOLOv4 |
PyTorch |
(tensorinfer) modelpostprocessors/libyolov3postprocess.so |
|
SSD-VGG16 |
Caffe |
|
None |
SSD MobileNet v1 FPN |
TensorFlow |
|
|
CRNN |
TensorFlow |
|
|
YOLOv5 |
PyTorch |
|
|
FasterRCNN-FPN/CascadeRCNN-FPN |
PyTorch |
(tensorinfer) modelpostprocessors/libfasterrcnnpostprocess.so |
|
ResNet-18 |
TensorFlow |
(tensorinfer) modelpostprocessors/libresnet50postprocess.so |
|
CTPN |
TensorFlow |
|
|
CTPN |
MindSpore |
|
|
DeepLabv3 |
MindSpore |
(tensorinfer) modelpostprocessors/libdeeplabv3post.so |
|
BERT-Base (Uncased) |
TensorFlow |
(tensorinfer) modelpostprocessors/libresnet50postprocess.so |
|
DeepLabv3+ |
PyTorch |
(tensorinfer) modelpostprocessors/libdeeplabv3post.so |
|
U-Net |
MindSpore |
(tensorinfer) modelpostprocessors/libunetmindsporepostprocess.so |
|
Mask R-CNN |
PyTorch |
(tensorinfer) modelpostprocessors/libmaskrcnnmindsporepost.so |
|
FaceNet |
TensorFlow |
Postprocessing is not required. |
|
SSD MobileNet v1 FPN |
MindSpore |
(tensorinfer) modelpostprocessors/libSsdMobilenetFpn_MindsporePost.so |
|
OpenPose |
TensorFlow |
(tensorinfer) modelpostprocessors/libopenposepostprocess.so |
None |
UNet++ |
MindSpore |
(tensorinfer) modelpostprocessors/libunetmindsporepostprocess.so |
None |
RetinaNet |
TensorFlow |
(tensorinfer) modelpostprocessors/libretinanetpostprocess.so |
|
HigherHRnet |
PyTorch |
(tensorinfer) modelpostprocessors/libhigherhrnetpostprocess.so |
|
YoloV7Detection |
PyTorch |
None |
None |
PPYOLOEPlusDetection |
Paddle |
None |
None |