diff --git a/README.md b/README.md index f1822eec5b03ece53492ebdea2485b8a6ed99822..3d14a0c8093ab9f3d497bbde2aa6eaf7fd5acae9 100644 --- a/README.md +++ b/README.md @@ -2,11 +2,16 @@ OneFlow相关的模型转换工具 -### onnx +### oneflow_onnx + +[![PyPI version](https://img.shields.io/pypi/v/oneflow-onnx.svg)](https://pypi.python.org/pypi/oneflow-onnx/) +[![PyPI pyversions](https://img.shields.io/pypi/pyversions/oneflow-onnx.svg)](https://pypi.python.org/pypi/oneflow-onnx/) +[![PyPI license](https://img.shields.io/pypi/l/oneflow-onnx.svg)](https://pypi.python.org/pypi/oneflow-onnx/) +[![PRs Welcome](https://img.shields.io/badge/PRs-welcome-brightgreen.svg)](https://github.com/Oneflow-Inc/oneflow_convert_tools/pulls) #### 简介 -onnx工具包含两个功能,一个是将OneFlow导出ONNX,另外一个是将各个训练框架导出的ONNX模型转换为OneFlow的模型。本工程已经适配了TensorFlow/Pytorch/PaddlePaddle框架的预训练模型通过导出ONNX转换为OneFlow(我们将这一功能叫作X2OneFlow)。 +oneflow_onnx工具包含两个功能,一个是将OneFlow导出ONNX,另外一个是将各个训练框架导出的ONNX模型转换为OneFlow的模型。本工程已经适配了TensorFlow/Pytorch/PaddlePaddle框架的预训练模型通过导出ONNX转换为OneFlow(我们将这一功能叫作X2OneFlow)。 - OneFlow2ONNX模型支持,支持OneFlow静态图模型转为ONNX,可转换由[flow.checkpoint.save](https://docs.oneflow.org/basics_topics/model_load_save.html)方法保存下来的OneFlow模型,详情可以参考[OneFlow2ONNX模型列表](docs/oneflow2onnx/oneflow2onnx_model_zoo.md)。 - X2OneFlow模型支持,支持将TensorFlow/Pytorch/PaddlePaddle的模型通过ONNX转换为OneFlow的模型,详情可以参考[X2OneFlow模型列表](docs/x2oneflow/x2oneflow_model_zoo.md)。