×
分享到微信

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

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

PyTorch 框架的 Yolov5 移植课程中的问题 解决中 Xinqian2022-09-22 15:57:53 回复 1 查看 使用求助
PyTorch 框架的 Yolov5 移植课程中的问题
分享到:

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


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


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


【出错信息】必填*:

请教各位大佬,在跑PyTorch 框架的 Yolov5 移植课程中的代码时,发现对于单张图片,比如输入img的维度为(1,3,640,640)时,结果是正确的。但是将img复制一下,将batch size改为2,img=torch.cat([img,img], dim=0),输入维度为(2,3,640,640)时,输出只有第一个图片的结果,而不是两倍的结果,请问是什么原因呢


【当前已做了哪些信息确认】选填:
例如交叉验证/CPU能运行/调试通过等


【相关日志文档】选填
如有,可附件


【出错代码链接】选填:
github的或gitee的代码的链接,

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