msnhnet_onnx

简体中文 | English

MsnhNet 相关的模型转换工具

msnhnet_onnx

简介

msnhnet_onnx 转换工具支持将Pytorch/PaddlePaddle/TensorFlow2/OneFlow等框架构建的模型经过ONNX转换为MsnhNet可用的模型（.msnhnet和.msnhbin）

环境依赖

用户环境配置

python>=3.5
onnx>=1.8.0
onnx-simplifier>=0.3.3
onnxoptimizer>=0.2.5
onnxruntime>=1.6.0
pytorch>=1.7.0
paddlepaddle>=2.0.0
paddle2onnx>=0.6
tensorflow>=2.0.0
tf2onnx>=1.8.4

安装

安装方式1

pip install msnhnet_onnx

安装方式2

git clone https://github.com/BBuf/msnhnet-onnx
cd msnhnet_onnx
python3 setup.py install

使用方法

请参考使用示例

相关文档

• OneFlow2ONNX模型列表

项目进展

• 2021/10/5 初始化项目
• 2021/10/10 测试通过Pytorch的AlexNet，VGG16，ResNet18，MobileNetV2转为ONNX，发布0.1.0版本。