『paddle』X2Paddle 学习笔记：支持的模型列表

TensorFlow预测模型

模型	代码
SqueezeNet	code
MobileNet_V1	code
MobileNet_V2	code
ShuffleNet	code
mNASNet	code
EfficientNet	code
Inception_V3	code
Inception_V4	code
Inception_ResNet_V2	code
VGG16	code
ResNet_V1_101	code
ResNet_V2_101	code
UNet	code1/code2
MTCNN	code
YOLO-V3	code
FALSR	code
DCSCN	code
Bert（albert）	code
Bert（chinese_L-12_H-768_A-12）	code
Bert（multi_cased_L-12_H-768_A-12）	code

Caffe预测模型

模型	代码
SqueezeNet	code
MobileNet_V1	code
MobileNet_V2	code
ShuffleNet_v2	code
InceptionV3	code
InceptionV4	code
mNASNet	code
MTCNN	code
Mobilenet_SSD	code
ResNet18	code
ResNet50	code
Unet	code
VGGNet	code
FaceDetection	-

【备注】代表源模型已无法获取。

ONNX预测模型

注： 部分模型来源于PyTorch，PyTorch的转换可参考 pytorch_to_onnx.md

模型	来源	operator version	备注
ResNet18	torchvison.model.resnet18	9
ResNet34	torchvison.model.resnet34	9
ResNet50	torchvison.model.resnet50	9
ResNet101	torchvison.model.resnet101	9
VGG11	torchvison.model.vgg11	9
VGG11_bn	torchvison.model.vgg11_bn	9
VGG19	torchvison.model.vgg19	9
DenseNet121	torchvison.model.densenet121	9
AlexNet	torchvison.model.alexnet	9
ShuffleNet	onnx official	9
Inception_V2	onnx official	9
MobileNet_V2	pytorch(personal practice)	9
mNASNet	pytorch(personal practice)	9
EfficientNet	pytorch(personal practice)	9
SqueezeNet	onnx official	9
Ultra-Light-Fast-Generic-Face-Detector-1MB	onnx_model	9
BERT	pytorch(huggingface)	11	转换时需指定input shape，见文档Q3
GPT2	pytorch(huggingface)	11	转换时需指定input shape，见文档Q3
CifarNet	tensorflow	9
Fcos	pytorch(mmdetection)	11
Yolov3	pytorch(mmdetection)	11
RetinaNet	pytorch(mmdetection)	11
FSAF	pytorch(mmdetection)	11
SSD	pytorch(mmdetection)	11
Faster R-CNN	pytorch(mmdetection)	11

PyTorch预测模型

模型	代码	备注
AlexNet	code	-
MNasNet	code	-
MobileNetV2	code	-
ResNet18	code	-
ShuffleNetV2	code	-
SqueezeNet	code	-
VGG16	code	-
InceptionV3	code	-
DeepLabv3_ResNet50	code	-
FCN_ResNet50	code	-
CamembertForQuestionAnswering	code	只支持trace模式
DPRContextEncoder	code	只支持trace模式
ElectraModel	code	只支持trace模式
FlaubertModel	code	只支持trace模式
Roberta	code	只支持trace模式
XLMRobertaForTokenClassification	code	只支持trace模式
EasyOCR_detector	code	-
EasyOCR_recognizer	code	-
SwinTransformer	code	-
BASNet	code	-
DBFace	code	-

PyTorch训练项目

模型	转换前代码	转换后代码
StarGAN	code	code
Ultra-Light-Fast-Generic-Face-Detector	code	code