模型支持列表
模型种类 |
模型框架 |
使用后处理动态库 |
获取途径 |
---|---|---|---|
YOLOv3 |
TensorFlow |
|
|
ResNet-50 |
TensorFlow |
|
|
Faster Rcnn |
TensorFlow |
|
|
MindSpore |
(tensorinfer框架)modelpostprocessors/libfasterrcnnpostprocess.so |
||
YOLOv4 |
Pytorch |
(tensorinfer框架)modelpostprocessors/libyolov3postprocess.so |
|
SSD-VGG16 |
Caffe |
|
无 |
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 |
无需后处理。 |
|
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/libretinanetpostprocess.so |
|
HigherHRnet |
Pytorch |
(tensorinfer框架)modelpostprocessors/libhigherhrnetpostprocess.so |
|
YoloV7Detection |
Pytorch |
无 |
无 |
PPYOLOEPlusDetection |
Paddle |
无 |
无 |