欢迎来到尧图网

客户服务 关于我们

您的位置:首页 > 汽车 > 新车 > mmdeploy 转换torch为onnx

mmdeploy 转换torch为onnx

2025/5/5 1:34:02 来源:https://blog.csdn.net/xihuanniNI/article/details/147627833  浏览:    关键词:mmdeploy 转换torch为onnx

安装 mmdeploy

以前已经安装了相关的mmsegment、mmcv 和 mmdetection 等库

方式安装 mmdeploy

 

pip install mmdeploy

或者,也可以从源代码安装,注意分支版本:

git clone https://github.com/open-mmlab/mmdeploy.git

cd mmdeploy

pip install -e .

应用

torch转onnx 使用MMDeploy 中的工具 tools/deploy.py

python tools/deploy.py \$DEPLOY_CFG \$MODEL_CFG \$PTH_MODEL_PATH \--work-dir $OUT_PATH \--show --device cuda:0 --dump-info

其中:

DEPLOY_PATH为mmdeploy工程下./mmedeploy/configs/mmseg/XXX.py的config文件路径

MODEL_CFG为自己训练时的config文件,一般在pth模型同目录下就有

PTH_MODEL_PATH为需要转的pth模型文件地址

OUT_PATH为输出的onnx模型文件地址和对应的json存放地址

python ./tools/deploy.py configs/mmseg/segmentation_onnxruntime_static-640x640.py ~/prjs/mmsegmentation/configs/hrnet/fcn-HR48.py ~/prjs/mmsegmentation/work_dirs/fcn-HR48/iter_32000.pth ~/prjs/mmsegmentation/demo/demo.jpg --work-dir work_dir_hr --show --device cuda:0

 

版权声明:

本网仅为发布的内容提供存储空间,不对发表、转载的内容提供任何形式的保证。凡本网注明“来源:XXX网络”的作品,均转载自其它媒体,著作权归作者所有,商业转载请联系作者获得授权,非商业转载请注明出处。

我们尊重并感谢每一位作者,均已注明文章来源和作者。如因作品内容、版权或其它问题,请及时与我们联系,联系邮箱:809451989@qq.com,投稿邮箱:809451989@qq.com

热搜词