DataXujing / YOLOv9

:fire: YOLOv9 paper解析,训练自己的数据集,TensorRT端到端部署, NCNN安卓手机部署
GNU General Public License v3.0
57 stars 11 forks source link

onnx导出问题 #3

Closed stackByStack closed 6 months ago

stackByStack commented 6 months ago

建议将 python export.py --data=./data/xray.yaml --weights=./runs/train/exp/weights/last.pt --opset=13 --include=onnx --simplify 替换为python export.py --data=./data/xray.yaml --weights=./runs/train/exp/weights/last.pt --include=onnx --simplify 默认opset为12 经过多次测试,使用前者导出的模型无法使用,效果很差.

DataXujing commented 6 months ago

我暂时没有复现出你说的问题,如果使用NCNN部署,建议使用PNNX.

stackByStack commented 6 months ago

好的,我再试试