官方代码download下来，数据集准备好，这些步骤做过AI数据集训练的人都没问题，在这里不用多说。
在训练时，跳出来loss.py的错误，按照网上说的改部分版本的软件，无法解决。我通过摸索，已经成功训练处理了数据集。原来代码的问题如下： # Append a = t[:,6].long()# anchor indices indices.append((b, a, gj.clamp_(0, gain[3]–1), gi.clamp_(0, gain[2]–1)))# image, anchor, grid indices tbox.append(torch.cat((gxy – gij, gwh),1))# box anch.append(anchors_cfg[a])# anchors tcls.append(c)# class出错之后就无法运行，如果坛子里面朋友遇到这种问题，可以按我的办法尝试。
原因是clamp_函数需要接受整形数，而gain是浮点型的，直接改成如下即可： # Append a = t[:,6].long()# anchor indices indices.append((b, a, gj.clamp_(0,(gain[3]–1).long()), gi.clamp_(0,(gain[2]–1).long())))# image, anchor, grid indices tbox.append(torch.cat((gxy – gij, gwh),1))# box anch.append(anchors_cfg[a])# anchors tcls.append(c)# class能够直接通过训练。可能环境不同、显卡不同、各个模块的版本不同，每一个人遇到的问题都不一样，各位如果遇到这个问题解决不了，可以按我的办法尝试一下。
免责声明：文章内容来自互联网，本站不对其真实性负责，也不承担任何法律责任，如有侵权等情况，请与本站联系删除。
转载请注明出处：Yolo-FastestV2训练笔记-yolo训练集需要多大 https://www.yhzz.com.cn/a/9534.html