数据和源码见文末

1.任务概述

数据集使用的是东北大学收集的一个钢材缺陷检测数据集，需要检测出钢材表面的6种划痕。同时，数据集格式是VOC格式，需要进行转化，上传的源码中的数据集是经过转换格式的版本。

2.数据与标签配置方法

在数据集目录下，train文件夹下有训练集数据及YOLO标签，valid文件夹下是验证集数据及YOLO标签。data.yaml是数据的配置文件，需要在训练和测试时进行指定。基于YOLOV5的钢材缺陷检测

train目录

基于YOLOV5的钢材缺陷检测

valid目录

基于YOLOV5的钢材缺陷检测

data.yaml是数据的配置文件，里面指定了训练集和验证集数据的目录，这里是使用的相对路径，如果容易报错的话，可以改为绝对路径。nc是类别数，names是类别名称。

基于YOLOV5的钢材缺陷检测3.标签转换脚本

原始标注文件格式是vo

声明：本站所有文章，如无特殊说明或标注，均为本站原创发布。任何个人或组织，在未征得本站同意时，禁止复制、盗用、采集、发布本站内容到任何网站、书籍等各类媒体平台。如若本站内容侵犯了原著者的合法权益，可联系我们进行处理。