×

签到

分享到微信

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

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

【PyTorch mlu】yolov5移植时算子yolov5_detection_output有BUG 已完结 Xinqian2022-11-30 11:12:05 回复 2 查看 使用求助
【PyTorch mlu】yolov5移植时算子yolov5_detection_output有BUG
分享到:

【寒武纪硬件产品型号】必填*:
MLU270


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


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


【出错信息】必填*:

目前情况是将官方的yolov5模型按照社区里的教程进行了移植,然后用新的数据集进行训练。发现在用量化模式在CPU上跑推理结果是正确的,但是将model和data放到MLU上结果就不对了。比如同一张有目标的图,CPU上会检测出4个目标框。但是用MLU上运行就会检测出多于4个框,有7,8个框,唯一的区别就是用了yolov5_detection_output算子作为输出,但多余的框里也能包含检测正确的框,请问这个yolov5_detection_output是否存在bug呢

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


【相关日志文档】选填


【出错代码链接】选填:

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