error when exporting to openvino

tino926 commented 6 months ago

Before Asking

[X] I have read the README carefully. 我已经仔细阅读了README上的操作指引。
[ ] I want to train my custom dataset, and I have read the tutorials for training your custom data carefully and organize my dataset correctly; (FYI: We recommand you to apply the config files of xx_finetune.py.) 我想训练自定义数据集，我已经仔细阅读了训练自定义数据的教程，以及按照正确的目录结构存放数据集。（FYI: 我们推荐使用xx_finetune.py等配置文件训练自定义数据集。）
[X] I have pulled the latest code of main branch to run again and the problem still existed. 我已经拉取了主分支上最新的代码，重新运行之后，问题仍不能解决。

Search before asking

[X] I have searched the YOLOv6 issues and found no similar questions.

Question

I got error when I tried to export a model to openvino model.

this is the command:

python ./deploy/OpenVINO/export_openvino.py --weight weights/yolov6n.pt --img 480 --batch 1

thie is the error message:

...
Starting to export OpenVINO...
usage: main.py [options]
main.py: error: unrecognized arguments: --data_type FP32
OpenVINO export failure: Command '['mo', '--input_model', 'weights/yolov6n.onnx', '--output_dir', 'weights/yolov6n_openvino', '--data_type', 'FP32']' returned non-zero exit status 2.

I followed this page to install Openvino. Maybe yolov6 rely on an older openvino?

Additional

No response

LuLu2kANG commented 5 months ago

SO you can change your openvino version

tino926 commented 5 months ago

@LuLu2kANG, yes I can do that. In the meantime, it would be great if there is a suggested version number in this repo.

knoppmyth commented 1 month ago

@tino926 If you want to use OpenVino 2024, take a look at PR1054. Tested fine for export fp32 and fp16.

meituan / YOLOv6