【YOLOX训练部署】将自己训练的YOLOX权重转化成ONNX 并进行推理

YOLOX 训练自己的VOC数据集
【YOLOX训练部署】YOLOX训练自己的VOC数据集_乐亦亦乐的博客-CSDN博客YOLOX 环境安装与训练自己标注的VOC数据集;https://blog.csdn.net/qq_41251963/article/details/122262738
训练好的模型:模型所在路径YOLOX/YOLOX_outputs/yolox_voc_s 

将自己的模型转化成ONNX
参考redeme:YOLOX/demo/ONNXRuntime at main · Megvii-BaseDetection/YOLOX · GitHub

pip install onnx
pip install onnxsim-simplifier

运行以下命令进行模型转化:
python3 tools/export_onnx.py --output-name your_yolox.onnx -f exps/your_dir/your_yolox.py -c your_yolox.pth
python

【YOLOX训练部署】将自己训练的YOLOX权重转化成ONNX 并进行推理最先出现在Python成神之路

版权声明:
作者:dingding
链接:https://www.techfm.club/p/11687.html
来源:TechFM
文章版权归作者所有,未经允许请勿转载。

THE END
分享
二维码
< <上一篇
下一篇>>