Yolov4训练bug 一步真实解决RuntimeError: result type Float can‘t be cast to the desired output type long int

在loss.py中用ctrl+F 找到 gain[2:] = torch.tensor(p[i].shape)[[3, 2, 3, 2]]  # xyxy gain 这一行

在其上面添加 shape = p[i].shape 如上图所添加的一样。

之后找到 indices.append 大概168行左右 换成下面的样子

indices.append((b, a, gj.clamp_(0, shape[2] - 1), gi.clamp_(0, shape[3] - 1)))  # image, anchor, grid indices  就ok了

 和yolov5解决思路一样 但具体位置不一样

 

  • 1
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值