×

签到

分享到微信

打开微信,使用扫一扫进入页面后,点击右上角菜单,

点击“发送给朋友”或“分享到朋友圈”完成分享

MLU270-SDK1.7.0-移植单类别YOLOv5s时 多核离线模型推理报错 解决中 Sunshine2023-08-10 11:52:14 回复 8 查看 技术答疑
MLU270-SDK1.7.0-移植单类别YOLOv5s时 多核离线模型推理报错
分享到:

MLU270 SDK1.7.0 移植YOLOv5s-v6.0模型时报错

YOLOv5参考代码: https://github.com/ultralytics/yolov5

环境采用的是开发平台上的(http://devplatform.hengqinai.com:30080/server):

image.png


问题:使用 https://github.com/ultralytics/yolov5训练单类别识别模型,即仅检测car车辆。

生成离线模型时 core_number=1,离线推理推理结果正常,可以检测到图像中的car。

core_number=4时,离线模型推理结果不正确,检测不到car。两种情况检测结果如下图:

image.png



请问这是什么原因?是不是plugin_yolov5_detection_op算子多核并行和类别不匹配导致的,可以提供下正确的算子吗 


版权所有 © 2024 寒武纪 Cambricon.com 备案/许可证号:京ICP备17003415号-1
关闭