YOLOv6_pro icon indicating copy to clipboard operation
YOLOv6_pro copied to clipboard

请问如何导出模型为onnx呢

Open AI-M-BOT opened this issue 1 year ago • 3 comments

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

AI-M-BOT avatar Dec 29 '23 02:12 AI-M-BOT

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

yang-0201 avatar Dec 29 '23 02:12 yang-0201

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

AI-M-BOT avatar Dec 30 '23 21:12 AI-M-BOT

这个这个。https://github.com/meituan/YOLOv6/tree/main/deploy/ONNX

yang-0201 avatar Jan 02 '24 02:01 yang-0201