这是从好奇开始的......我从公共谷歌驱动器下载了预训练的yolov5s.pt，并使用yolov5的models/export.py将其转换为输入形状为[1,3,640,640]的yolov5s.onnx文件。然后我使用openvino的部署工具/mo.py将yolov5s.onnx转换成openvino推理引擎文件（.xml+.bin）。转换成功，没有错误。最后，我尝试使用openvino的演示程序使用这些文件运行预测，预测成功返回结果。但是所有的结果都返回负数，并且数组级别是错误的。是否无法将 yolov5 权重转换为 openvino？
问问题
663 次
2 回答
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。有关更多详细信息，您可以在此处关注此线程。
于 2021-03-16T09:48:55.553 回答
0
试试这个：（python mo.py --input_model yolov5s.onnx -s 255 --reverse_input_channels --output Conv_245,Conv_261,Conv_277使用 Netron 检查你的架构）
于 2021-08-27T17:44:23.557 回答