将yolov5s.pt转换为openvino IR格式成功，预测材料失败

Question

这是从好奇开始的……我从公共google drive下载了预先训练好的yolov5s.pt，并使用yolov5的model/export.py将其转换为输入形状为1,3640,640的yolov5s.onnx文件。然后，我使用openvino的部署工具/mo.py将yolov5s.onnx转换为openvino推理引擎文件(.xml+.bin)。转换成功，没有错误。最后，使用openvino的demo程序对这些文件进行了预测，并成功地返回了预测结果。但是所有的结果都返回负数，并且数组级别是错误的。openvino的yolov5权重转换是不可能的吗？

Answer 1

YOLOv5目前不是OpenVINO工具包正式支持的拓扑。请在此处查看经过验证的ONNX和PyTorch拓扑列表https://docs.openvinotoolkit.org/latest/openvino_docs_MO_DG_prepare_model_convert_model_Convert_Model_From_ONNX.html

然而，我们有一个建议供您尝试，但不能保证它会成功。您可以通过更改export.py以包含Detect layer：yolov5/models/export.py来完成此操作

a1c8406中的第28行

model.model[-1].export = True  # set Detect() layer export=True 

以上需要从True更改为False。有关更多详细信息，请访问此线程here。

Answer 2

试试这个:python mo.py --input_model yolov5s.onnx -s 255 --reverse_input_channels --output Conv_245,Conv_261,Conv_277 (使用Netron检查你的架构)