带标注的轴承划痕检测数据集,识别率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()
相关推荐
qq_25294131688 小时前
日本道路病害检测数据集 | 道路病害 裂缝检测 坑洼识别 市政巡检9287期
人工智能·深度学习·yolo·目标检测·计算机视觉·道路检测
qq_252941316810 小时前
伊蚊检测数据集 | 伊蚊检测 病媒生物 公共卫生 目标检测9304期
人工智能·yolo·目标检测·计算机视觉·蚊子·尹蚊
qq_252941316818 小时前
无人机视角公路两侧滑坡检测数据集 | 公路巡检 滑坡检测 地质灾害 无人机航拍9302期
yolo·目标检测·目标跟踪·无人机·地质灾害·无人机视角·滑坡检测
向哆哆1 天前
羊行为目标检测数据集:4类别、2,000张图像 | 目标检测
人工智能·yolo·目标检测·目标跟踪
燃烧的枫1 天前
Windows 下 YOLO11 Pose 的 TensorRT C++ 部署
windows·yolo·tensorrt·1024程序员节·yolo11-pose
磁场转动100万匹2 天前
道路巡检系统检测案例:YOLOv5 + Flask + PyQt 全栈实现
python·yolo
alwaysrun2 天前
YOLOv8n INT8量化实战
yolo·模型量化
城南皮卡丘2 天前
苹果果实成熟度检测数据集 4195张+YOLO格式
深度学习·yolo·目标检测·计算机视觉·数据集
微尘hjx3 天前
【数据集 11】面向城市道路的坑洞检测数据集(含8760张,已按比例划分训练、验证、测试集,支持YOLO全系列训练)
人工智能·yolo·目标检测·数据集·道路坑洞·路坑数据集·路坑
William一直在路上4 天前
把轻量化 YOLO 模型部署到你的手机上
yolo·智能手机