YOLOV5应用实战项目:钢材表面缺陷检测(数据集:NEU-CLS)笔记

简介: YOLOV5应用实战项目:钢材表面缺陷检测(数据集:NEU-CLS)笔记

1. NEU-CLS数据集

下载方式:

方式一:已上传至CSDN资源,链接如下

[钢材表面缺陷检测数据集:NEU-DET 用于钢材表面的6种缺陷检测]

方式二:

关注GZH:阿旭算法与机器学习,回复:“NEU”即可获取本文数据集

2. 实战视频链接如下

【YOLOV5应用实战项目系列】教程

3.YOLOV5模型配置及训练个人笔记

  1. 准备好数据集

图片数据集与Label数据集(txt格式)。图像名称与Label名称一一对应。我已将Labelss标签处理为txt格式文件,见上面数据集中。

标签Label的格式,必须是一个txt文件,一个txt文件中可存放多个标签,文件内容格式如下:

第一个0:代表缺陷的类别信息。从0-5:分别代表6种缺陷类别。

后面四个参数分别代表矩形框的中心点X, Y, 及矩形框with,height。(注:这些都是相对于原始图像的相对位置参数

  1. 添加配置文件data.yaml

Train:代表训练集路径
 
Val:代表验证集路径
 
Nc:代表分类类别数目
 
Names:代表各个分类的名称
  1. 配置train.py文件的运行参数(pycharm中的配置方式如下

--data ./datasets/NEU-DET/data.yaml
--cfg ./models/yolov5s.yaml
--batch-size 4

Data:代表之前配置的配置文件data.yaml路径

Cfg:代表使用的yolo模型路径

Batch-size:代表每一批训练的样本数。(根据电脑配置合理设置)4,8,16,32等

参数设置好后,就可以运行train.py文件训练模型了。

4.训练结果相关说明

训练过程中,会在runs目录下生成训练的一些过程文件以及训练好的模型。

Weights:中存放的是训练好的模型;

Results:会记录每一个epoch训练完后的模型结果。

5.使用模型进行检测

模型检测使用的是detect.py文件,

  1. 配置detect.py的运行参数

--source E:\CJX_DL\YOLO_shizhan\YOLOV5\yolov5-master\datasets\NEU-DET\valid\images\
--weights E:\CJX_DL\YOLO_shizhan\YOLOV5\yolov5-master\runs\train\exp2\weights\best.pt
--save-txt
--conf 0.6

Socurce:表示所需检测的image对象路径,可以指定单个img也可以指定img目录。

Weights:表示训练好的模型路径,一般在run/train目录下:best.pt表示训练过程中效果最好的模型,last.pt表示最后一次训练出的模型。

然后就可以运行detect.py监测文件了。监测结果会默认存放到runs/detect目录下


相关文章
|
6月前
|
机器学习/深度学习 编解码 人工智能
|
8月前
|
机器学习/深度学习 网络架构 计算机视觉
YOLOv5改进 | 检测头篇 | 利用DBB重参数化模块魔改检测头实现暴力涨点 (附代码 + 详细修改教程)
YOLOv5改进 | 检测头篇 | 利用DBB重参数化模块魔改检测头实现暴力涨点 (附代码 + 详细修改教程)
373 3
|
7月前
|
人工智能 计算机视觉 Python
【超详细】【YOLOV8使用说明】一套框架解决CV的5大任务:目标检测、分割、姿势估计、跟踪和分类任务【含源码】(1)
【超详细】【YOLOV8使用说明】一套框架解决CV的5大任务:目标检测、分割、姿势估计、跟踪和分类任务【含源码】
【超详细】【YOLOV8使用说明】一套框架解决CV的5大任务:目标检测、分割、姿势估计、跟踪和分类任务【含源码】(1)
|
3月前
|
机器学习/深度学习 算法 PyTorch
目标检测实战(五): 使用YOLOv5-7.0版本对图像进行目标检测完整版(从自定义数据集到测试验证的完整流程)
本文详细介绍了使用YOLOv5-7.0版本进行目标检测的完整流程,包括算法介绍、环境搭建、数据集准备、模型训练、验证、测试以及评价指标。YOLOv5以其高精度、快速度和模型小尺寸在计算机视觉领域受到广泛应用。
1287 0
目标检测实战(五): 使用YOLOv5-7.0版本对图像进行目标检测完整版(从自定义数据集到测试验证的完整流程)
|
3月前
|
机器学习/深度学习 监控 计算机视觉
目标检测实战(八): 使用YOLOv7完成对图像的目标检测任务(从数据准备到训练测试部署的完整流程)
本文介绍了如何使用YOLOv7进行目标检测,包括环境搭建、数据集准备、模型训练、验证、测试以及常见错误的解决方法。YOLOv7以其高效性能和准确率在目标检测领域受到关注,适用于自动驾驶、安防监控等场景。文中提供了源码和论文链接,以及详细的步骤说明,适合深度学习实践者参考。
711 0
目标检测实战(八): 使用YOLOv7完成对图像的目标检测任务(从数据准备到训练测试部署的完整流程)
|
3月前
|
机器学习/深度学习 JSON 数据可视化
YOLO11-pose关键点检测:训练实战篇 | 自己数据集从labelme标注到生成yolo格式的关键点数据以及训练教程
本文介绍了如何将个人数据集转换为YOLO11-pose所需的数据格式,并详细讲解了手部关键点检测的训练过程。内容涵盖数据集标注、格式转换、配置文件修改及训练参数设置,最终展示了训练结果和预测效果。适用于需要进行关键点检测的研究人员和开发者。
582 0
|
7月前
|
计算机视觉
【超详细】【YOLOV8使用说明】一套框架解决CV的5大任务:目标检测、分割、姿势估计、跟踪和分类任务【含源码】(2)
【超详细】【YOLOV8使用说明】一套框架解决CV的5大任务:目标检测、分割、姿势估计、跟踪和分类任务【含源码】
|
7月前
|
测试技术 计算机视觉
【YOLOv8性能对比试验】YOLOv8n/s/m/l/x不同模型尺寸大小的实验结果对比及结论参考
【YOLOv8性能对比试验】YOLOv8n/s/m/l/x不同模型尺寸大小的实验结果对比及结论参考
|
8月前
|
机器学习/深度学习 计算机视觉
【YOLOv8改进】MPDIoU:有效和准确的边界框损失回归函数 (论文笔记+引入代码)
YOLO目标检测专栏介绍了YOLO的有效改进和实战案例,包括卷积、主干网络、注意力机制和检测头的创新。提出了一种新的边界框回归损失函数MPDIoU,它基于最小点距离,能更好地处理不同宽高比的预测框,包含重叠、中心点距离和尺寸偏差的全面考虑。MPDIoU损失函数在YOLACT和YOLOv7等模型上的实验显示了优于现有损失函数的性能。此外,还介绍了WIoU_Scale类用于计算加权IoU,以及bbox_iou函数实现不同IoU变体的计算。详细实现和配置可在相应链接中查阅。
|
机器学习/深度学习 人工智能 算法
目标检测模型设计准则 | YOLOv7参考的ELAN模型解读,YOLO系列模型思想的设计源头
目标检测模型设计准则 | YOLOv7参考的ELAN模型解读,YOLO系列模型思想的设计源头
941 0
目标检测模型设计准则 | YOLOv7参考的ELAN模型解读,YOLO系列模型思想的设计源头

热门文章

最新文章