yolov5配置错误记录

最新推荐文章于 2025-03-27 22:03:41 发布

原创

最新推荐文章于 2025-03-27 22:03:41 发布 · 2.8k 阅读

8 ·

CC 4.0 BY-SA版权

文章标签：

#YOLO

在使用YOLOv5训练过程中遇到找不到数据集、标签错误、数据为0及FutureWarning: Non-finite norm等问题。通过检查数据集路径、标签命名、数据质量、学习率、梯度爆炸等多方面进行排查，发现数据集标签文件夹命名错误和训练数据可能存在异常值。目前问题尚未完全解决，解决方案将在后续文章中分享。

这里是直接没有找到数据集，说明是路径错误。经过设置yaml后，

# Train/val/test sets as 1) dir: path/to/imgs, 2) file: path/to/imgs.txt, or 3) list: [path/to/imgs1, path/to/imgs2, ..]
path: ../autodl-tmp/datasets/neu  # dataset root dir
train: train/images  # train images (relative to 'path') 118287 images
val: val/images  # val images (relative to 'path') 5000 images
test: test/images  # 20288 of 40670 images, submit to https://competitions.codalab.org/competitions/20794

# Classes
nc: 6
names:
  0: crazing
  1: inclusion
  2: patches
  3: pitted_surface
  4: rolled-in_scale
  5: scratches

出现了新的错误。