【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
共有 0 条评论