请问如何导出模型为onnx呢

yang-0201 / YOLOv6_pro

Make it easier for yolov6 to change the network structure

GNU General Public License v3.0

67 stars 15 forks source link

Open AI-M-BOT opened 9 months ago

AI-M-BOT commented 9 months ago

在v5 exportde.py 的基础上要修改什么吗

yang-0201 commented 9 months ago

已经集成转换为onnx的代码了，可以使用如下命令进行： python ./deploy/ONNX/export_onnx.py \ --weights yolov6s.pt \ --img 640 \ --batch 1 \ --simplify

AI-M-BOT commented 9 months ago

额请问是哪个版本yolov6的deploy文件夹呢? 这个仓库里已经删除了deploy文件夹

yang-0201 commented 9 months ago