基于传统算法和YOLO的手机壳缺陷检测

基于传统算法与YOLO的手机壳缺陷检测系统

一、立项背景与目标
手机壳生产环节中，划痕、气泡、毛边、色差等缺陷直接影响产品良率与用户体验。传统人工检测效率低、误检率高，难以满足大规模产线需求。本项目融合传统图像处理算法与YOLOv8目标检测模型，构建高精度、高效率的自动化缺陷检测系统，实现手机壳表面缺陷的实时识别、定位与分类，目标将检测准确率提升至98%以上，检测速度控制在0.5秒/件以内，显著降低人工成本与漏检率。

二、软件功能与核心模块
图像采集模块：支持工业相机实时采集手机壳多角度图像，自动触发拍照与图像预处理（灰度化、去噪、增强对比度）。
传统算法模块：基于OpenCV实现边缘检测、形态学操作、模板匹配等算法，用于快速筛查明显缺陷（如毛边、缺角）。
YOLOv8缺陷识别模块：训练专用缺陷数据集，实现对划痕、气泡、凹陷、印刷偏移等复杂缺陷的精准识别与定位，支持多类别输出。
结果融合与决策模块：综合传统算法与YOLO结果，通过置信度加权与规则引擎输出最终判定，支持缺陷类型、位置、严重等级标注。
人机交互界面：提供实时检测画面、缺陷热力图、统计报表、历史数据查询、报警推送等功能，支持参数配置与模型更新。

三、业务流程与功能路径
图像输入：工业相机采集手机壳图像 → 图像预处理 → 输入检测系统。
双路并行检测：
传统算法路径：边缘检测 → 形态学处理 → 缺陷初筛。
YOLO路径：图像输入 → YOLOv8推理 → 缺陷定位与分类。
结果融合：双路结果比对 → 置信度加权 → 最终判定（合格/不合格+缺陷类型）。
输出与反馈：显示缺陷位置与类型 → 生成检测报告 → 触发分拣机构 → 数据存入数据库 → 支持历史追溯与模型迭代优化。

系统采用“传统算法+深度学习”双轨架构，传统模块用OpenCV实现边缘/形态学处理，负责快速筛查；YOLO模块用PyTorch训练，负责复杂缺陷识别。我负责YOLO模块，训练数据集含5000+标注图像，最终模型mAP@0.5达96.3%，单图推理耗时0.12秒。难点在于小目标划痕漏检，通过增加数据增强（随机裁剪、旋转）与调整Anchor尺寸解决，漏检率从12%降至3.5%。