带标注的轴承划痕检测数据集,识别率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()
相关推荐
CoderIsArt4 小时前
印花缺陷检测方案二C# + Halcon 引擎 + YOLO 融合实现
yolo
梦之美丽生活11 小时前
【本地部署及docker部署】yolov8_detect
yolo·docker·容器
YOLO数据集集合1 天前
基准标记目标检测数据集 |基准标记 视觉定位 相机标定 目标检测 YOLO格式 深度学习数据集 计算机9038期
数码相机·yolo·目标检测·视觉定位·基准标定
YOLO数据集集合1 天前
排球动作识别数据集 | 排球动作 体育分析 动作识别9041期
深度学习·数据集·动作识别·排球动作识别·运动动作识别·体育分析·排球动作
YOLO数据集集合1 天前
隧道病害检测数据集 | 隧道检测 裂缝识别 渗水监测 剥落检测9037期
人工智能·深度学习·yolo·隧道·裂缝检测·裂缝识别·隧道病害
深瞳智检1 天前
深瞳智检(小程序)正式上线!【免费/快捷/方便】获取YOLO检测数据集
人工智能·yolo·目标检测·小程序·数据集·cv数据收集
a1117761 天前
SOP行为监控系统 开源项目yolo
yolo
jay神2 天前
2026年YOLO还有哪些创新点可以做?
python·yolo·毕业设计·科研·课程设计·创新
毕竟是shy哥2 天前
计算YOLO数据集中每个类的目标数
算法·yolo·机器学习
仙女修炼史2 天前
gradcam在yolo中的应用
人工智能·python·yolo