YOLOv5/v7修改初始锚定框anchor(以及utils文件夹的部分py文件报错)

参考:(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就么这个毛病

希望指正

猜你喜欢

转载自blog.csdn.net/m0_60461719/article/details/127067097