YOLOv8实现热轧带钢表面缺陷目标检测(附代码和数据集)
数据集中images文件存放的是jpg格式的缺陷图片,annatation中存放的是缺陷图片对应的标注为xml格式文件,包括缺陷的类型、位置和大小等信息。在自己创建的data文件夹中,创建data.yaml文件,存放自己的训练集和测试集地址,检测项目类别数,检测项目类别:(对于本数据集共有6中类别,因此nc:6。)(种类名称放到names:)使用YOLOv8进行缺陷检测,需要的数据集格式为txt。(n、s、m、l、x五种不同规模的预训练模型选择一个即可),放到ultralytics-main目录下即可。
原创
2023-12-21 17:47:04 ·
2488 阅读 ·
1 评论