Skip to content

Files

Latest commit

 

History

History
128 lines (110 loc) · 13 KB

x2paddle_model_zoo.md

File metadata and controls

128 lines (110 loc) · 13 KB

X2Paddle转换库

TensorFlow预测模型

模型 代码 类型 误差
SqueezeNet code 视觉 atol@1e-05, rtol@1e-05
MobileNet_V1 code 视觉 atol@1e-05, rtol@1e-05
MobileNet_V2 code 视觉 atol@1e-05, rtol@1e-05
ShuffleNet code 视觉 atol@1e-05, rtol@1e-05
mNASNet code 视觉 atol@1e-05, rtol@1e-05
EfficientNet code 视觉 atol@1e-05, rtol@1e-05
Inception_V3 code 视觉 -
Inception_V4 code 视觉 atol@1e-05, rtol@1e-05
Inception_ResNet_V2 code 视觉 atol@1e-05, rtol@1e-05
VGG16 code 视觉 atol@1e-05, rtol@1e-05
ResNet_V1_101 code 视觉 atol@1e-05, rtol@1e-05
ResNet_V2_101 code 视觉 atol@1e-05, rtol@1e-05
UNet code1/code2 视觉 atol@1e-04, rtol@1e-04
MTCNN code 视觉 atol@1e-05, rtol@1e-05
YOLO-V3 code 视觉 atol@1e-04, rtol@1e-04
FALSR code 视觉 -
DCSCN code 视觉 -
Bert（albert） code 自然语言处理 atol@1e-04
Bert（chinese_L-12_H-768_A-12） code 自然语言处理 -
Bert（multi_cased_L-12_H-768_A-12） code 自然语言处理 -

【备注】- 代表源模型已无法获取，或未测试精度。

Caffe预测模型

模型 代码 类型 误差
SqueezeNet code 视觉 atol@1e-05, rtol@1e-05
MobileNet_V1 code 视觉 atol@1e-05, rtol@1e-05
MobileNet_V2 code 视觉 atol@1e-05, rtol@1e-05
ShuffleNet_v2 code 视觉 atol@1e-05, rtol@1e-05
InceptionV3 code 视觉 -
InceptionV4 code 视觉 -
mNASNet code 视觉 atol@1e-05, rtol@1e-05
MTCNN code 视觉 atol@1e-05, rtol@1e-05
Mobilenet_SSD code 视觉 atol@1e-05
ResNet18 code 视觉 atol@1e-05, rtol@1e-05
ResNet50 code 视觉 atol@1e-05, rtol@1e-05
Unet code 视觉 atol@1e-05, rtol@1e-05
VGGNet code 视觉 atol@1e-05, rtol@1e-05
FaceDetection - 视觉 -

【备注】- 代表源模型已无法获取，或未测试精度。

ONNX预测模型

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

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

【备注】- 代表源模型已无法获取，或未测试精度。

PyTorch预测模型

模型 代码 类型 误差 备注
AlexNet code 视觉 atol@1e-04 -
MNasNet code 视觉 atol@1e-04 -
MobileNetV2 code 视觉 atol@1e-04 -
ResNet18 code 视觉 atol@1e-04 -
ShuffleNetV2 code 视觉 atol@1e-04 -
SqueezeNet code 视觉 atol@1e-04 -
VGG16 code 视觉 atol@1e-04 -
InceptionV3 code 视觉 atol@1e-04 -
DeepLabv3_ResNet50 code 视觉 atol@1e-04 -
FCN_ResNet50 code 视觉 atol@1e-04 -
CamembertForQuestionAnswering code 自然语言处理 atol@1e-04 只支持trace模式
DPRContextEncoder code 自然语言处理 atol@1e-04 只支持trace模式
ElectraModel code 自然语言处理 atol@1e-04 只支持trace模式
FlaubertModel code 自然语言处理 atol@1e-04 只支持trace模式
Roberta code 自然语言处理 atol@1e-04 只支持trace模式
XLMRobertaForTokenClassification code 自然语言处理 - 只支持trace模式
EasyOCR_detector code 视觉 atol@1e-04 -
EasyOCR_recognizer code 视觉 atol@1e-04 -
SwinTransformer code 视觉 atol@1e-04 -
BASNet code 视觉 atol@1e-04 -
DBFace code 视觉 atol@1e-04 -
SOLAR code 自然语言处理 - 只支持trace模式
YOLOX code 视觉 - 只支持trace模式
YOLOv5 code 视觉 atol@3e-03 只支持trace模式
MockingBird code 语音 - 只支持trace模式
GPT2 code 自然语言处理 atol@1e-04 只支持trace模式
MT5_small code 自然语言处理 atol@2e-04 只支持trace模式

【备注】- 代表源模型已无法获取，或未测试精度。

PyTorch训练项目

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

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