官网尽量选择新的yolov5,因为旧的版本需要自己改一些参数。我能力不够,不会修改。
当使用yolov5-6.1时,需要修改两处:
1、将下面的代码替换掉上图中的红圈部分:
用anchors, shape = self.anchors[i], p[i].shape替换文件中loss.py的相似部分。
2、indices.append((b, a, gj.clamp_(0, shape[2] - 1), gi.clamp_(0, shape[3] - 1))) # image, anchor, grid替换loss.py文件中的相似部分。
之后改改权重路径、模型路径以及coco128.yaml就可以训练模型了。
具体的可以看这里