List of the Supported Models
Currently, the MindX SDK supports both the new and old model post-processing frameworks. For details, see Inference Model Post-processing Development. Use the corresponding post-processing dynamic library based on the actual framework.
Type |
Framework |
Post-processing Dynamic Library |
How to Obtain |
|---|---|---|---|
YOLOv3 |
TensorFlow |
(New framework) modelpostprocessors/libyolov3postprocess.so |
None |
Caffe |
The model is under internal debugging. You are advised to use the model provided later. |
||
ResNet-50 |
TensorFlow |
(New framework) modelpostprocessors/libresnet50postprocess.so |
None |
Faster R-CNN |
TensorFlow |
(New framework) modelpostprocessors/libfasterrcnnpostprocess.so |
None |
MindSpore |
|||
YOLOv4 |
PyTorch |
(New framework) modelpostprocessors/libyolov3postprocess.so |
|
MindSpore |
None |
||
SSD-VGG16 |
Caffe |
(New framework) modelpostprocessors/libssdvggpostprocess.so |
The model is under internal debugging. You are advised to use the model provided later. |
SSD MobileNet v1 FPN |
TensorFlow |
(New framework) modelpostprocessors/libssdmobilenetfpnpostprocess.so |
|
CRNN |
TensorFlow |
(New framework) modelpostprocessors/libcrnnpostprocess.so |
|
YOLOv5 |
PyTorch |
(New framework) modelpostprocessors/libyolov3postprocess.so |
|
FasterRCNN-FPN/CascadeRCNN-FPN |
PyTorch |
(New framework) modelpostprocessors/libfasterrcnnpostprocess.so |
|
ResNet-18 |
TensorFlow |
(New framework) modelpostprocessors/libresnet50postprocess.so |
|
DeepLabv3+ |
TensorFlow |
(New framework) modelpostprocessors/libdeeplabv3post.so |
None |
CTPN |
TensorFlow |
(New framework) modelpostprocessors/libctpnpostprocess.so |
None |
CTPN |
MindSpore |
(New framework) modelpostprocessors/libctpnpostprocess.so |
None |
DeepLabv3 |
MindSpore |
(New framework) modelpostprocessors/libdeeplabv3post.so |
None |
BERT-Base (Uncased) |
TensorFlow |
(New framework) modelpostprocessors/libresnet50postprocess.so |
None |
DeepLabv3+ |
PyTorch |
(New framework) modelpostprocessors/libdeeplabv3post.so |
None |
U-Net |
MindSpore |
(New framework) modelpostprocessors/libunetmindsporepostprocess.so |
None |
Mask R-CNN |
PyTorch |
(New framework) modelpostprocessors/libmaskrcnnmindsporepost.so |
None |
FaceNet |
TensorFlow |
Post-processing is not required. |
None |
SSD MobileNet v1 FPN |
MindSpore |
(New framework) modelpostprocessors/libSsdMobilenetFpn_MindsporePost.so |
None |
OpenPose |
TensorFlow |
(New framework) modelpostprocessors/libopenposepostprocess.so |
The model is under internal debugging. You are advised to use the model provided later. |
UNet++ |
MindSpore |
(New framework) modelpostprocessors/libunetmindsporepostprocess.so |
None |
RetinaNet |
TensorFlow |
(New framework) modelpostprocessors/retinanetpostprocess.so |
None |
HigherHRnet |
PyTorch |
(New framework) modelpostprocessors/libhigherhrnetpostprocess.so |
None |