yolov5 报错The size of tensor a (80) must match the size of tensor b (56) at non-singleton dimension 3

一 把数据集处理好之后,直接端到端训练。结果报这个错误,网上搜寻后最终发现正确的答案有:

   batch_size不能整除,但设置为整除后还是报错。

二 再次狂看博客,最终发现是数据集图片的尺寸不一致导致的,即width和height不全是同一个尺寸。因此写程序处理尺寸后得以正常运行

三 附上python对目录操作的常见指令

  1. 删除目录 os.remove(path)  
  2. 新建目录 os.mkdir(path)
  3. 移动文件 shutil.copy("old_path", "new_path")支持右斜杠\
  4. 重命名文件 os.rename(old,new)
  5. 遍历目录 fileList=os.listdir(path)  而后遍历fileList即可
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值