diff --git a/README.md b/README.md index fc3fea0e8d56e56b69a6029c90e4c1776fef1cc3..6953faf521570de7ae0c859c9df2c3612b9e5517 100644 --- a/README.md +++ b/README.md @@ -5,12 +5,12 @@ X2Paddle支持将其余深度学习框架训练得到的模型,转换至Paddle X2Paddle is a toolkit for converting trained model to PaddlePaddle from other deep learning frameworks. ## 转换模型库 -X2Paddle在多个主流的CV模型上,测试过TensorFlow/Caffe/ONNX模型的转换,可以在[X2Paddle-Model-Zoo](x2paddle_model_zoo.md)查看我们的模型测试列表,可以在[OP-LIST](op_list.md)中查看目前X2Paddle支持的OP列表。如果你在新的模型上进行了测试转换,也欢迎继续补充该列表;如若无法转换,可通过ISSUE反馈给我们,我们会尽快跟进。 +X2Paddle在多个主流的CV模型上,测试过TensorFlow/Caffe/ONNX/PyTorch模型的转换,可以在[X2Paddle-Model-Zoo](x2paddle_model_zoo.md)查看我们的模型测试列表,可以在[OP-LIST](op_list.md)中查看目前X2Paddle支持的OP列表。如果你在新的模型上进行了测试转换,也欢迎继续补充该列表;如若无法转换,可通过ISSUE反馈给我们,我们会尽快跟进。 ## 环境依赖 python == 2.7 | python >= 3.5 -paddlepaddle >= 2.0.0 +paddlepaddle 2.0-rc 或者 develop **按需安装以下依赖** tensorflow : tensorflow == 1.14.0 @@ -30,7 +30,7 @@ python setup.py install ### 安装方式二 我们会定期更新pip源上的x2paddle版本 ``` -pip install x2paddle --index https://pypi.Python.org/simple/ +pip install x2paddle==1.0.0rc0 --index https://pypi.Python.org/simple/ ``` ## 使用方法 ### TensorFlow @@ -47,7 +47,7 @@ x2paddle --framework=onnx --model=onnx_model.onnx --save_dir=pd_model ``` ### PyTorch -> PyTorch不支持命令行使用方式,详见[PyTorch2Paddle](pytorch2paddle.md) +> PyTorch不支持命令行使用方式,详见[PyTorch2Paddle](./docs/user_guides/pytorch2paddle.md) ### Paddle2ONNX > Paddle2ONNX功能已迁移至新的github: https://github.com/PaddlePaddle/paddle2onnx, 欢迎大家去新的代码仓库查看详细介绍以及新功能。 @@ -62,7 +62,7 @@ x2paddle --framework=onnx --model=onnx_model.onnx --save_dir=pd_model |--save_dir | 指定转换后的模型保存目录路径 | |--model | 当framework为tensorflow/onnx时,该参数指定tensorflow的pb模型文件或onnx模型路径 | |--caffe_proto | **[可选]** 由caffe.proto编译成caffe_pb2.py文件的存放路径,当存在自定义Layer时使用,默认为None | -|--define_input_shape | **[可选]** For TensorFlow, 当指定该参数时,强制用户输入每个Placeholder的shape,见[文档Q2](FAQ.md) | +|--define_input_shape | **[可选]** For TensorFlow, 当指定该参数时,强制用户输入每个Placeholder的shape,见[文档Q2](./docs/user_guides/FAQ.md) | |--params_merge | **[可选]** 当指定该参数时,转换完成后,inference_model中的所有模型参数将合并保存为一个文件__params__ | @@ -78,11 +78,12 @@ X2Paddle提供了工具解决如下问题,详见[tools/README.md](tools/README 2. 合并模型参数文件 ## 相关文档 -1. [X2Paddle使用过程中常见问题](FAQ.md) -2. [如何导出TensorFlow的pb模型](export_tf_model.md) -3. [X2Paddle测试模型库](x2paddle_model_zoo.md) -4. [PyTorch模型导出为ONNX模型](pytorch_to_onnx.md) -5. [X2Paddle内置的Caffe自定义层](caffe_custom_layer.md) +1. [X2Paddle使用过程中常见问题](./docs/user_guides/FAQ.md) +2. [如何导出TensorFlow的pb模型](./docs/user_guides/export_tf_model.md) +3. [X2Paddle测试模型库](./docs/introduction/x2paddle_model_zoo.md) +4. [X2Paddle支持的op列表](./docs/introduction/op_list.md) +5. [PyTorch模型导出为ONNX模型](./docs/user_guides/pytorch2onnx.md) +6. [X2Paddle添加内置的Caffe自定义层](./docs/user_guides/add_caffe_custom_layer.md) ## 更新历史 2019.08.05 @@ -91,7 +92,15 @@ X2Paddle提供了工具解决如下问题,详见[tools/README.md](tools/README 3. 解决Windows上保存模型无法加载的问题 4. 新增optimizer,优化代码结构,合并conv、batch_norm的bias和激活函数 -**如果你需要之前版本的tensorflow2fluid/caffe2fluid/onnx2fluid,可以继续访问release-0.3分支,获取之前版本的代码使用。** +2020.12.09 +1. 新增PyTorch2Paddle转换方式,转换得到Paddle动态图代码,并动转静获得inference_model。 +方式一:trace方式,转换后的代码有模块划分,每个模块的功能与PyTorch相同。 +方式二:script方式,转换后的代码按执行顺序逐行出现。 +2. 新增Caffe/ONNX/Tensorflow到Paddle动态图的转换。 +3. 新增TensorFlow op(14个):Neg、Greater、FloorMod、LogicalAdd、Prd、Equal、Conv3D、Ceil、AddN、DivNoNan、Where、MirrorPad、Size、TopKv2 +4. 新增Optimizer模块,主要包括op融合、op消除功能,转换后的代码可读性更强,进行预测时耗时更短。 + +**如果你需要之前版本的tensorflow2fluid/caffe2fluid/onnx2fluid,可以继续访问release-0.8分支,获取之前版本的代码使用。** ## Acknowledgements