官方代码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

能够直接通过训练。可能环境不同、显卡不同、各个模块的版本不同，每一个人遇到的问题都不一样，各位如果遇到这个问题解决不了，可以按我的办法尝试一下。

声明：本文内容由易百纳平台入驻作者撰写，文章观点仅代表作者本人，不代表易百纳立场。如有内容侵权或者其他问题，请联系本站进行删除。