> 文章列表 > YOLOv5、YOLOv8实战系列:NEU钢材缺陷检测精度提升☆☆

YOLOv5、YOLOv8实战系列:NEU钢材缺陷检测精度提升☆☆

YOLOv5、YOLOv8实战系列:NEU钢材缺陷检测精度提升☆☆

yolov5检测钢材表面缺陷

  • NEU数据集
  • 训练
    • 训练参数及配置
    • NEU.yaml
    • 注意力对比改进1(yaml和代码)
  • 测试
    • 改进1,进行多种实验测试,涨点
    • 基于yolov5s模型训练,实验模型存储

本文记录的是YOLOV5的实战笔记,是一个用yolov5检测钢材表面缺陷的实战案例。

主要分为以下几个步骤:

  1. 将原始标签xml文件数据处理成yolo要求的txt格式。为了公平测试,对每类进行均分,训练240张,训练30张,测试30张张。只划分了训练和测试两部分。

NEU数据集

下载方式:
已上传百度网盘:链接

训练

训练参数及配置

CPU:i9-12900KF
GPU:RTX3080Ti
CUDA:11.1
pytorch:1.8.3
训练参数:epoch=200 imgz=224
另外,