『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 |
类似资料：