error when exporting to openvino

[tino926]

Before Asking

• 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等配置文件训练自定义数据集。）

• I have pulled the latest code of main branch to run again and the problem still existed. 我已经拉取了主分支上最新的代码，重新运行之后，问题仍不能解决。

Search before asking

• 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]

SO you can change your openvino version

[tino926]

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

[knoppmyth]

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