基于YOLOv11的工业轴承缺陷检测 产线实时质检系统
基于YOLOv11的工业产线实时质检系统
本项目聚焦于在制造产线中部署YOLOv11（You Only Look Once第11版）实时目标检测系统，旨在实现产品缺陷自动识别、合格/不良品分类及智能计数。该系统可直接集成到生产流程中，实现质量控制的自动化升级。
目录结构
- 项目概述
- 数据集说明
- 技术方案
- 性能指标
- 使用指南
- 演示视频
- 致谢
项目概述
制造业亟需高效的质检系统来保障产品一致性并减少损耗。本项目采用最先进的YOLOv11目标检测模型，主要实现三大功能：
- 实时检测产线上的产品
- 自动分类为「合格品」或「不良品」
- 双类别产品自动计数
该系统以高速度、高精度和可扩展性为核心设计目标，专为工业场景优化。
数据集说明
采用Kaggle开源工业数据集，包含多工况下的产品图像，每张图片均标注边界框并分类为：
- 不良品（Damaged/خراب）：存在可见缺陷
- 合格品（OK/سالم）：符合质量标准
关键参数
- 数据集来源：[工业铸件产品真实数据集]
- 图像数量：700张
- 标注类型：边界框+双标签（Damaged/OK）
技术方案
模型架构
采用YOLO11l与YOLO11s双版本，该架构具有以下优势：
- 单阶段检测实现毫秒级推理
- 支持多类别分类（本项目中为双类别）
训练流程
- 数据预处理
- 统一缩放至512×512分辨率
- 按8:1:1划分训练集/验证集/测试集
- 模型训练
- 基于预训练权重进行迁移学习
- 定制化微调
- 性能评估
- 核心指标：精确率、召回率、mAP、F1分数
- 测试集mAP@0.5达95.10%
部署方案
通过Python+OpenCV在产线模拟环境中实现实时推理。
性能指标
| 指标 | 数值 |
|---|---|
| 精确率 | 95.55% |
| 召回率 | 95.65% |
| mAP@0.5 | 95.10% |
| 损失值 | 0.17 |
效果演示
点击下载测试视频示例：[下载链接]
使用指南
环境配置
- 通过yml文件快速创建环境
- 基础要求：
- Python 3.10.16
- 安装依赖库：
pipinstalltorch torchvision opencv-python ultralytics基于YOLOv11的工业轴承缺陷检测 产线实时质检系统