×

签到

分享到微信

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

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

yolov5 v6.1融合模式设置4core预测框异常 已解决 ilovemlu2023-05-28 22:46:33 回复 2 查看 技术答疑 使用求助
yolov5 v6.1融合模式设置4core预测框异常
分享到:

【寒武纪硬件产品型号】必填*:MLU220 m.2

【使用操作系统】必填*:ubuntu 16.04

【使用驱动版本】必填*:v4.9.9

【出错信息】必填*:在pytorch:0.15.0-2-ubuntu16.04 docker容器中进行yolov5 v6.1模型转换,自己训练的模型有4个类别,当设置set_core_number(1)时,融合模式下可以正常预测出框;当set_core_number(4)时,预测框异常,但是框数量时对的;

【当前已做了哪些信息确认】选填:

1core下输出信息.txt

4core下输出结果.txt

yolov5 v6.1官方的80类yolov5s模型测试,1core或者4core都能正常检测出框

【相关日志文档】选填:

【出错代码链接】选填:

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