参考:(23条消息) yolov5 anchors设置详解_高祥xiang的博客-CSDN博客_yolov5锚框
工科本科生做大创,零基础,看这么多代码直接大脑发懵,然后照着链接改,一堆报错
Exception: Error loading data from images/train2017: images\train2017 does not exist
其实是coco128.yaml里面没改,得把路径改了
这里我用我自己的数据集
import utils.autoanchor as autoAC
# 对数据集重新计算 anchors
new_anchors = autoAC.kmean_anchors('D:\YOLOv7\yolov7-main\yolov7-main\Wu.v1i.yolov7pytorch\data.yaml', 12, 640, 5.0, 1000, True)
print(new_anchors)
然后对data.yaml里的train和valid路径都改成绝对路径
train: D:\YOLOv7\yolov7-main\yolov7-main\Wu.v1i.yolov7pytorch\train
val: D:\YOLOv7\yolov7-main\yolov7-main\Wu.v1i.yolov7pytorch\valid
nc: 1
names: ['裂缝']
就好了
题外话:当时改的时候我根本不知道这个images/train2017是啥,然后把utils文件夹全逛了一遍,发现有几个地方好像不太对,其中一个就是metrics.py, 我把import general给注释了,因为run general的时候,他说metrics重复调用,不知道为啥,先这样吧
# Model validation metrics
from pathlib import Path
import matplotlib.pyplot as plt
import numpy as np
import torch
# import general # (循环调用了)
再就是v5也是,import utils.xxxx 我一律把utils给删了,有点虎,因为import总是标红,v7就么这个毛病
希望指正