当前位置:   article > 正文

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

yolov5 需要更换anchor

参考:(23条消息) yolov5 anchors设置详解_高祥xiang的博客-CSDN博客_yolov5锚框

工科本科生做大创,零基础,看这么多代码直接大脑发懵,然后照着链接改,一堆报错 

Exception: Error loading data from images/train2017: images\train2017 does not exist

其实是coco128.yaml里面没改,得把路径改了

这里我用我自己的数据集

  1. import utils.autoanchor as autoAC
  2. # 对数据集重新计算 anchors
  3. new_anchors = autoAC.kmean_anchors('D:\YOLOv7\yolov7-main\yolov7-main\Wu.v1i.yolov7pytorch\data.yaml', 12, 640, 5.0, 1000, True)
  4. print(new_anchors)

 然后对data.yaml里的train和valid路径都改成绝对路径

  1. train: D:\YOLOv7\yolov7-main\yolov7-main\Wu.v1i.yolov7pytorch\train
  2. val: D:\YOLOv7\yolov7-main\yolov7-main\Wu.v1i.yolov7pytorch\valid
  3. nc: 1
  4. names: ['裂缝']

就好了

题外话:当时改的时候我根本不知道这个images/train2017是啥,然后把utils文件夹全逛了一遍,发现有几个地方好像不太对,其中一个就是metrics.py, 我把import general给注释了,因为run general的时候,他说metrics重复调用,不知道为啥,先这样吧

  1. # Model validation metrics
  2. from pathlib import Path
  3. import matplotlib.pyplot as plt
  4. import numpy as np
  5. import torch
  6. # import general # (循环调用了)

再就是v5也是,import utils.xxxx    我一律把utils给删了,有点虎,因为import总是标红,v7就么这个毛病

希望指正

声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/笔触狂放9/article/detail/579118
推荐阅读
相关标签
  

闽ICP备14008679号