Skip to content

Latest commit

 

History

History
101 lines (86 loc) · 8.3 KB

x2paddle_model_zoo.md

File metadata and controls

101 lines (86 loc) · 8.3 KB

X2Paddle模型测试库

目前X2Paddle支持70+的TensorFlow OP,40+的Caffe Layer,覆盖了大部分CV分类模型常用的操作。我们在如下模型列表中测试了X2Paddle的转换。

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

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 code

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

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 -