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

本文记录的是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
另外，