search

meituan / YOLOv6

YOLOv6: a single-stage object detection framework dedicated to industrial applications.
GNU General Public License v3.0
5.72k stars 1.03k forks source link

Yolov6n export onnx have different #476

Closed YunlongHu closed 2 years ago

YunlongHu commented 2 years ago

The model i train on my dataset , and use "./deploy/ONNX/export_onnx.py" to export 1663349424813

But the onnx model downloaded from your website is different with mine.

1663349582341

How to solve this problem. The first onnx is difficult to depoly.

Chilicyy commented 2 years ago

hi @YunlongHu , did you add the arg --simplify when exporting onnx? You can refer to https://github.com/meituan/YOLOv6/tree/main/deploy/ONNX . Howerver, your exported onnx model seems normal, and it may work fine in your project as well.