带标注的轴承划痕检测数据集,识别率83.0%,1180张图,支持yolo,coco json,voc xml,文末有模型训练代码

带标注的轴承划痕检测数据集,识别率83.0%,1180张图,支持yolo,coco json,voc xml,文末有模型训练代码

模型训练指标参数:

数据集拆分

总图数:1180 张图数
训练集

769 张图

验证集

236 张图

测试集

175 张图

预处理

自动定向:应用

调整大小:拉伸到640x640

增强

数据集标签:

'scratch'

标签解释

scratch-》划痕

数据集图片和标注信息示例:

数据集下载:

yolo26:https://download.csdn.net/download/pbymw8iwm/92897476

yolo v12:https://download.csdn.net/download/pbymw8iwm/92897479

yolo v11:https://download.csdn.net/download/pbymw8iwm/92897480

yolo v9:https://download.csdn.net/download/pbymw8iwm/92897477

yolo v8:https://download.csdn.net/download/pbymw8iwm/92897478

yolo v7:https://download.csdn.net/download/pbymw8iwm/92897481

coco json:https://download.csdn.net/download/pbymw8iwm/92897482

pascal voc xml: https://download.csdn.net/download/pbymw8iwm/92897483

YOLO模型训练

下载数据集之后解压到当前文件夹,然后将 我的仓库 https://gitcode.com/pbymw8iwm/YOLOProject里的训练模型脚本复制到文件夹下,假设你使用的是yolov8来训练你就用 python train_yolov8.py

​​

模型验证测试情况:

验证测试代码:

python 复制代码
#需要安装pip install ultralytics
from ultralytics import YOLO
import cv2
 
# 加载训练好的 YOLO .pt 模型
model = YOLO('best.pt')  # 替换为你实际的 .pt 模型文件路径
 
# 定义要测试的图片路径
image_path = './image.jpg'  # 替换为你实际的图片文件路径
 
# 使用模型对图片进行预测
results = model(image_path)
 
# 获取预测结果
for result in results:
    # 获取绘制了检测框的图片
    annotated_image = result.plot()
 
    # 显示图片
    cv2.imshow("YOLOv Inference", annotated_image)
 
    # 等待按键退出
    cv2.waitKey(0)
 
    # 关闭所有 OpenCV 窗口
    cv2.destroyAllWindows()
相关推荐
YOLO数据集集合2 分钟前
无人机航拍桥梁巡检数据集 | 桥梁结构缺陷检测 深度学习目标检测数据10338期
深度学习·yolo·目标检测·计算机视觉·无人机
前网易架构师-高司机13 分钟前
带标注的薄荷病叶数据集,识别率98.8%,3533张图,支持yolo,coco json,voc xml,文末有模型训练代码
yolo·数据集·缺陷·薄荷·叶子·风干·变质
爱睡懒觉的焦糖玛奇朵1 小时前
【视觉检测之人员奔跑检测算法开发思路】
人工智能·python·深度学习·算法·yolo·视觉检测
叫我:松哥1 小时前
基于深度学习的辣椒叶片病害识别系统设计实现,融合CBAM注意力机制的改进ResNet-50模型和YOLO检测,准确率达96%
图像处理·人工智能·深度学习·yolo·flask·bootstrap·注意力机制
爱睡懒觉的焦糖玛奇朵2 小时前
【从视频到数据集:焦糖玛奇朵的魔法工具Dataset Cleaner】
人工智能·python·学习·算法·yolo·音视频
YOLO数据集集合16 小时前
无人机航拍街道巡检数据集 | 空中视角车辆检测、交通流量统计、违停识别、智能交通YOLO数据集10399期
深度学习·yolo·目标检测·无人机
stsdddd18 小时前
YOLO系列目标检测数据集大全【第二十期】
yolo·目标检测·目标跟踪
西西弗Sisyphus18 小时前
YOLO26 自定义损失函数 重写 init_criterion 方法 损失类不继承基类
pytorch·python·yolo·yolo11·yolo26
stsdddd1 天前
YOLO系列目标检测数据集大全【第二十一期】
yolo·目标检测·目标跟踪
张飞飞飞飞飞1 天前
目标检测-根据YOLO格式标签统计目标尺寸分布
人工智能·yolo·目标检测