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目录下


相关文章
|
机器学习/深度学习 人工智能 数据库
【表面缺陷检测】表面缺陷检测数据集汇总
本文收集整理了16个表面缺陷检测相关的数据集,并对每个数据集的特点进行了简单的介绍。
【表面缺陷检测】表面缺陷检测数据集汇总
|
7月前
|
机器学习/深度学习 边缘计算 算法
金属材料表面六种缺陷类型数据集:工业视觉检测的优质训练资源
本数据集包含1800张640×640金属表面缺陷图像,涵盖裂纹、夹杂、斑块、凹坑、氧化皮、划痕6类典型缺陷,已按7:2:1划分训练/验证/测试集,标注规范(YOLO格式),适配YOLO、Faster R-CNN等模型,助力工业视觉检测研究与落地。
|
机器学习/深度学习 并行计算 算法
YOLOv8改进 | 卷积篇 |手把手教你添加动态蛇形卷积(Dynamic Snake Convolution)
YOLOv8改进 | 卷积篇 |手把手教你添加动态蛇形卷积(Dynamic Snake Convolution)
1859 0
|
6月前
|
机器学习/深度学习 人工智能 算法
公路表面裂缝目标检测数据集(1500 张图片已划分、已标注)| AI训练适用于目标检测任务
本数据集含1500张真实道路图像,已划分train/val/test并精准标注裂缝(crack)边界框,覆盖沥青/水泥路面、多光照与多角度场景,适配YOLO、RT-DETR等主流目标检测模型,专为道路病害智能巡检与小目标检测研究设计。
|
10月前
|
机器学习/深度学习 人工智能 机器人
焊接情况检测数据集(千张图片已划分)| 面向工业质检的目标检测训练集
总结来看,本次分享的焊接情况检测数据集为工业智能化提供了坚实的基础。数据集涵盖了“良好焊缝”、“不良焊缝”和“缺陷”三大类别,采用了标准的 YOLO 标注格式,保证了在目标检测任务中能够高效、准确地训练模型。通过合理划分训练集、验证集和测试集,开发者可以充分利用数据进行模型优化与验证,从而在实际工业生产环境中实现对焊接表面缺陷的自动检测与监控。
焊接情况检测数据集(千张图片已划分)| 面向工业质检的目标检测训练集
|
Linux 开发工具
7种比较Linux中文本文件的最佳工具
7种比较Linux中文本文件的最佳工具
7种比较Linux中文本文件的最佳工具
|
机器学习/深度学习 边缘计算 算法
金属材料表面六种缺陷类型数据集 | 适用于YOLO等视觉检测模型(1800张图片已划分、已标注)
本数据集包含1800张金属表面缺陷图像,涵盖裂纹、夹杂、凹坑等6类缺陷,已标注并按train/val/test划分,支持YOLO、Faster R-CNN等模型训练,适用于工业质检与智能检测研究。
金属材料表面六种缺陷类型数据集 | 适用于YOLO等视觉检测模型(1800张图片已划分、已标注)
|
JSON 人工智能 数据格式
AI计算机视觉笔记二十六:YOLOV8自训练关键点检测
本文档详细记录了使用YOLOv8训练关键点检测模型的过程。首先通过清华源安装YOLOv8,并验证安装。接着通过示例权重文件与测试图片`bus.jpg`演示预测流程。为准备训练数据,文档介绍了如何使用`labelme`标注工具进行关键点标注,并提供了一个Python脚本`labelme2yolo.py`将标注结果从JSON格式转换为YOLO所需的TXT格式。随后,通过Jupyter Notebook可视化标注结果确保准确性。最后,文档展示了如何组织数据集目录结构,并提供了训练与测试代码示例,包括配置文件`smoke.yaml`及训练脚本`train.py`,帮助读者完成自定义模型的训练与评估。
7093 3
|
Kubernetes Cloud Native 调度
《分布式任务调度框架深度对比:Quartz/XXL-JOB/Elastic-Job/PowerJob选型指南》​
根据IDC预测,到2025年全球将有75%的企业任务调度系统需要重构以适应云原生架构。技术雷达监测:定期关注CNCF技术趋势报告渐进式改造:从非核心业务开始验证新框架人才储备:重点培养具备K8s Operator开发能力的调度专家评估现有系统的云原生适配度在测试环境部署PowerJob 4.3.3参与CNCF调度技术社区讨论制定6个月框架迁移路线图(注:本文数据来自各框架官方路线图、CNCF年度报告及笔者压力测试结果,转载请保留出处)
3266 0
|
机器学习/深度学习 人工智能 算法
基于YOLOv8的钢铁缺陷实时检测系统【训练和系统源码+Pyside6+数据集+包运行】
基于YOLOv8的钢铁缺陷实时检测系统,通过1800张图片训练,开发了带GUI界面的检测系统,支持图片、视频和摄像头实时检测,提高生产效率和产品质量。系统基于Python和Pyside6开发,具备模型权重导入、检测置信度调节等功能。项目代码、数据集可通过特定链接获取。
806 1
基于YOLOv8的钢铁缺陷实时检测系统【训练和系统源码+Pyside6+数据集+包运行】