x2paddle_model_zoo.md

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	-
SOLAR	code	只支持trace模式

PyTorch训练项目

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

注： 受限于不同框架的差异，部分预测模型可能会存在目前无法转换的情况，如TensorFlow中包含控制流的模型等。对于常见的预测模型或PyTorch项目，如若您发现无法转换或转换失败，存在较大diff等问题，欢迎通过ISSUE反馈的方式告知我们(模型名，代码实现或模型获取方式)，我们会及时跟进。