openvino - 将 yolov5s.pt 转换为 openvino IR 格式成功，预测内容失败

Question

这是从好奇开始的......我从公共谷歌驱动器下载了预训练的yolov5s.pt，并使用yolov5的models/export.py将其转换为输入形状为[1,3,640,640]的yolov5s.onnx文件。然后我使用openvino的部署工具/mo.py将yolov5s.onnx转换成openvino推理引擎文件（.xml+.bin）。转换成功，没有错误。最后，我尝试使用openvino的演示程序使用这些文件运行预测，预测成功返回结果。但是所有的结果都返回负数，并且数组级别是错误的。是否无法将 yolov5 权重转换为 openvino？

标签： openvinoyolov5

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 以包含检测层来做到这一点： yolov5/models/export.py

a1c8406 中的第 28 行

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

以上需要从True更改为False。有关更多详细信息，您可以在此处关注此线程。