模型支持列表
模型种类 |
模型框架 |
使用后处理动态库 |
获取途径 |
|---|---|---|---|
YOLOv3 |
TensorFlow |
（tensorinfer框架）modelpostprocessors/libyolov3postprocess.so |
|
ResNet-50 |
TensorFlow |
（tensorinfer框架）modelpostprocessors/libresnet50postprocess.so |
|
Faster Rcnn |
TensorFlow |
（tensorinfer框架）modelpostprocessors/libfasterrcnnpostprocess.so |
|
MindSpore |
（tensorinfer框架）modelpostprocessors/libfasterrcnnpostprocess.so |
||
YOLOv4 |
Pytorch |
（tensorinfer框架）modelpostprocessors/libyolov3postprocess.so |
|
SSD-VGG16 |
Caffe |
（tensorinfer框架）modelpostprocessors/libssdvggpostprocess.so |
无 |
SSD MobileNet v1 FPN |
TensorFlow |
（tensorinfer框架）modelpostprocessors/libssdmobilenetfpnpostprocess.so |
|
CRNN |
TensorFlow |
（tensorinfer框架）modelpostprocessors/libcrnnpostprocess.so |
|
YOLOv5 |
Pytorch |
（tensorinfer框架）modelpostprocessors/libyolov3postprocess.so |
|
FasterRCNN-FPN/CascadeRCNN-FPN |
Pytorch |
（tensorinfer框架）modelpostprocessors/libfasterrcnnpostprocess.so |
|
ResNet-18 |
TensorFlow |
（tensorinfer框架）modelpostprocessors/libresnet50postprocess.so |
|
CTPN |
TensorFlow |
（tensorinfer框架）modelpostprocessors/libctpnpostprocess.so |
|
CTPN |
MindSpore |
（tensorinfer框架）modelpostprocessors/libctpnpostprocess.so |
|
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 |
无需后处理。 |
|
SSD MobileNet v1 FPN |
MindSpore |
（tensorinfer框架）modelpostprocessors/libSsdMobilenetFpn_MindsporePost.so |
|
OpenPose |
TensorFlow |
（tensorinfer框架）modelpostprocessors/libopenposepostprocess.so |
无 |
UNet++ |
MindSpore |
（tensorinfer框架）modelpostprocessors/libunetmindsporepostprocess.so |
|
RetinaNet |
TensorFlow |
（tensorinfer框架）modelpostprocessors/retinanetpostprocess.so |
|
HigherHRnet |
Pytorch |
（tensorinfer框架）modelpostprocessors/libhigherhrnetpostprocess.so |
|
YoloV7Detection |
Pytorch |
无 |
无 |
PPYOLOEPlusDetection |
Paddle |
无 |
无 |
