admin 管理员组

文章数量: 887021

yolov5/v7训练报错build target indices.append((b, a, gj.clamp

torch 1.12.0运行yolov5 6.0版本时报错：
build target indices.append((b, a, gj.clamp_(0, gain[3] - 1), gi.clamp_(0, gain[2] - 1))) # image, anchor, grid indices runtimeerror: result type float can’t be cast to the desired output type long int
解决方案

RuntimeError: result type Float can’t be cast to the desired output type long int · Issue #35 · WongKinYiu/yolov7 · GitHub
在loss.py文件下修改

gain = torch.ones(7, device = device)

替换为

gain = torch.ones(7, device = device).long()

参考：

本文标签： yolov5v7训练报错build target indicesappend((b A gjclamp

版权声明：本文标题：yolov5v7训练报错build target indices.append((b, a, gj.clamp 内容由网友自发贡献，该文观点仅代表作者本人， 转载请联系作者并注明出处：http://www.freenas.com.cn/jishu/1730915334h1405879.html， 本站仅提供信息存储空间服务，不拥有所有权，不承担相关法律责任。如发现本站有涉嫌抄袭侵权/违法违规的内容，一经查实，本站将立刻删除。