模型支持列表
当前MindX SDK支持新、老两种模型后处理框架,具体请参见推理模型后处理开发介绍,请根据实际框架使用对应的后处理动态库。
模型种类 |
模型框架 |
使用后处理动态库 |
获取途径 |
---|---|---|---|
YOLOv3 |
TensorFlow |
(新框架)modelpostprocessors/libyolov3postprocess.so |
无 |
Caffe |
内部调试中,暂请勿使用,请等待后续提供。 |
||
ResNet50 |
TensorFlow |
(新框架)modelpostprocessors/libresnet50postprocess.so |
无 |
Faster Rcnn |
TensorFlow |
(新框架)modelpostprocessors/libfasterrcnnpostprocess.so |
无 |
MindSpore |
|||
YOLOv4 |
Pytorch |
(新框架)modelpostprocessors/libyolov3postprocess.so |
|
MindSpore |
无 |
||
SSD-VGG16 |
Caffe |
(新框架)modelpostprocessors/libssdvggpostprocess.so |
内部调试中,暂请勿使用,请等待后续提供。 |
SSD MobileNet v1 FPN |
TensorFlow |
(新框架)modelpostprocessors/libssdmobilenetfpnpostprocess.so |
|
CRNN |
TensorFlow |
(新框架)modelpostprocessors/libcrnnpostprocess.so |
|
YOLOv5 |
Pytorch |
(新框架)modelpostprocessors/libyolov3postprocess.so |
|
FasterRCNN-FPN/CascadeRCNN-FPN |
Pytorch |
(新框架)modelpostprocessors/libfasterrcnnpostprocess.so |
|
ResNet-18 |
TensorFlow |
(新框架)modelpostprocessors/libresnet50postprocess.so |
|
DeepLabv3+ |
TensorFlow |
(新框架)modelpostprocessors/libdeeplabv3post.so |
无 |
CTPN |
TensorFlow |
(新框架)modelpostprocessors/libctpnpostprocess.so |
|
CTPN |
MindSpore |
(新框架)modelpostprocessors/libctpnpostprocess.so |
|
DeepLabv3 |
MindSpore |
(新框架)modelpostprocessors/libdeeplabv3post.so |
|
BERT-Base (Uncased) |
TensorFlow |
(新框架)modelpostprocessors/libresnet50postprocess.so |
|
DeepLabv3+ |
Pytorch |
(新框架)modelpostprocessors/libdeeplabv3post.so |
|
U-Net |
MindSpore |
(新框架)modelpostprocessors/libunetmindsporepostprocess.so |
|
Mask R-CNN |
Pytorch |
(新框架)modelpostprocessors/libmaskrcnnmindsporepost.so |
|
FaceNet |
TensorFlow |
无需后处理。 |
|
SSD MobileNet v1 FPN |
MindSpore |
(新框架)modelpostprocessors/libSsdMobilenetFpn_MindsporePost.so |
|
OpenPose |
TensorFlow |
(新框架)modelpostprocessors/libopenposepostprocess.so |
内部调试中,暂请勿使用,请等待后续提供。 |
Unet++ |
MindSpore |
(新框架)modelpostprocessors/libunetmindsporepostprocess.so |
内部调试中,暂请勿使用,请等待后续提供。 |
RetinaNet |
TensorFlow |
(新框架)modelpostprocessors/retinanetpostprocess.so |
|
HigherHRnet |
Pytorch |
(新框架)modelpostprocessors/libhigherhrnetpostprocess.so |