Skip to content

Latest commit

 

History

History
73 lines (62 loc) · 6.1 KB

x2paddle_model_zoo.md

File metadata and controls

73 lines (62 loc) · 6.1 KB

X2Paddle模型测试库

目前X2Paddle支持50+的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 转换需要关闭NHWC->NCHW的优化,见文档Q2
FALSR code -
DCSCN 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