★ FEATURED ARTICLE 口罩佩戴检测这几年几乎成了深度学习毕设里的“默认选项”。它效果好、成本低、可解释性也强目标检测本身又是一个足够成熟的领域把模型迁移到“行人有没有戴口罩”这个具体场景时难度曲线刚好卡在本科生能顺利完工、研究生也能写出深度分析的位置。我身边不少同学就是靠类似题目完成了从调包侠到能看懂训练曲线的蜕变所以如果你正在纠结选题这个方向确实值得认真考虑。先说清楚这个项目到底在做什么。它的本质不是图像分类而是目标检测输入一张图片或一帧视频模型需要先把画面里每个行人用边界框框出来然后判断框内的人是“戴了口罩”还是“没戴口罩”。再加上行人属于小目标居多、背景复杂、光线变化大的场景实际做下来要踩的坑并不少但也正因为有这些坑它才足够撑起一整个毕业设计的体量。这篇文章我会从整体设计思路讲起再到数据准备、模型选型、训练调参、推理部署最后整理一份我实测下来的排查技巧。全程用我在这个项目里踩过的坑和验证过的方法说话你照着复现能少走至少两周弯路。1. 项目整体设计与思路拆解1.1 为什么说这是个“小而全”的毕设场景一个合格的毕业设计不能只是“把模型跑通”它还得有数据、有实验、有对比、有结论。口罩佩戴检测恰好把这些问题全占齐了你需要采集或收集数据需要做标注和预处理需要对比不同模型的性能需要解释检测结果的准确率和召回率最后还需要做一个可视化或部署展示。这整个链路走完基本就把计算机视觉方向的核心技能都过了一遍。我曾经见过有同学把题目改成“交通场景行人安全装备检测”数据里不仅有口罩还有安全帽、反光背心甚至施工隔离服。这个改法其实很聪明——核心方法完全不变但应用场景更具体做实验时能讲的故事更多。如果你时间充裕也可以在口罩检测的基础上加一个“未戴口罩”的行人跟踪计数功能让项目从单纯的检测升级成带有预警性质的系统这在答辩时会明显加分。不过要注意扩展功能的前提是基础检测已经足够稳定。如果一开始就把时间浪费在漂亮的界面上最后模型精度一塌糊涂答辩老师随便问两句就露馅了。基础模型和数据实验永远应该排在第一位。1.2 技术路线选型为什么我最终选了 YOLO很多新手第一次看到这个题目第一反应是“这不就是个二分类吗”。真去做就会发现如果只做图像分类摄像头画面里同时出现五个行人时你根本不知道谁戴了谁没戴分类网络只能输出整体概率但检测任务需要的是每个个体的空间位置和类别。硬要用分类网络做得先跑一个人体检测器再把每个人的头像裁出来二次分类流程复杂且误差会累积。所以主流做法还是直接用端到端的目标检测模型。我在选型时认真比较过三套方案两阶段的 Faster R-CNN、单阶段的 SSD、以及 YOLO 系列。Faster R-CNN 精度确实强但推理速度慢训练也重。SSD 速度快但小目标检测能力偏弱而行人佩戴口罩检测里“远处的小脸”恰恰是高频场景。对比下来YOLO 系列在两阶段精度和单阶段速度之间找到了一个很舒服的平衡点尤其是 YOLOv5 和 YOLOv8开箱即用文档清晰复现成本极低。还有一个很现实的原因YOLO 生态里预训练权重丰富可以基于 COCO 上的通用权重微调。毕竟口罩检测的数据集规模很难和 COCO 这种上百万张的通用数据集相比没有预训练权重从零训练效果会差好几个档次。使用 YOLOv8n 或 YOLOv8s 这样的小模型结合预训练权重在消费级显卡上训练一次时间成本也很可控对毕设来说是特别稳的选择。1.3 数据集获取与标注策略多数人在这里翻车数据是这个项目的地基。公开数据集确实有但量级和质量参差不齐。我使用的训练集主要来自两部分一是开源的口罩检测数据集里面覆盖了室内外、不同肤色、不同遮挡程度的人脸二是我自己模拟实际监控场景补充的一批照片比如商场入口、校门口、地铁通道这些地方的行人画面。两条腿走路正负样本的比例会更接近真实场景。拿到原始图片后最耗时的是标注。标注工具有不少Windows 上用 LabelImg 最顺手macOS 上可以用 Label Studio两者都支持直接导出 YOLO 格式的 txt 标签文件。标注前必须先定好类别规范我建议标准场景下用两类就够了“with_mask”和“without_mask”。如果你想把“口罩戴在下巴上”这种情况也纳入可以加一个类似“mask_worn_improperly”的类别但要保证每个类别样本量足够否则模型会严重偏向样本多的类别。提示标注规范一定要在项目开始时定死比如“口罩完全遮住口鼻才算戴好”“遮挡超过 40% 的样本不参与训练”“侧脸只露出一半时怎么标”。这些规则如果定得晚返工量会非常痛苦。2. 核心细节解析与实操要点2.1 数据预处理不能只知道增强开关在哪YOLO 训练代码里默认会做一些数据增强比如随机翻转、色彩抖动、Mosaic 拼图这些都很有效但默认参数不一定适合你的数据。比如口罩检测涉及大量肤色和背景色HSV 色彩增强如果调得太激烈会把皮肤色调拉到完全失真模型在真实摄像头画面上反而会误检。我调试下来的经验是把 hsv_h、hsv_s、hsv_v 控制在默认范围的 70% 左右把翻转概率设为 0.5打开 Mosaic 和 MixUp。特别是 Mosaic它会一次性把四张训练图拼在一起让小目标被迫在更小的区域出现这对提升远距离行人检测能力帮助非常大几乎是不用白不用的增强手段。另外一个容易被忽略的预处理点是图片分辨率。YOLOv8 默认训练尺寸是 640x640但实际监控画面的行人占比通常很小如果你发现远距离漏检严重可以在显存允许的情况下把 imgsz 调成 960 甚至 1280。分辨率上去了小目标检测效果会立竿见影地提升代价仅仅是训练时间变长。我在实验对比里从 640 提到 960 后远距离行人的召回率大约提升了 6 个点这在毕设里是很值得写进结论的数据。2.2 训练参数怎么定三步快速决定学习率、批大小和轮次很多同学拿到代码第一件事就是把 epoch 拉满到 300然后直接把训练丢进去跑两天最后效果很差才回头排查。我的经验是先做小规模验证再决定正式训练参数步骤简单但非常有效。第一步先拿约 2000 张样本、训练 20 个 epoch 跑一轮盯着训练 loss 曲线看。正常的 loss 应该在前几个 epoch 快速下降之后缓慢收敛。如果 loss 完全不动大概率是学习率太低或数据有问题。第二步根据 loss 曲线调整学习率。YOLO 默认的优化器自带自动学习率调整但手动判断的方法很简单loss 震荡剧烈就降低学习率loss 下降太慢就适当提高初始学习率。我的实践中batch size 为 16 时初始学习率 0.01 对 YOLOv8s 来说有点过激降到 0.005 后曲线稳定很多。如果你的显存小只能开 batch size 8学习率也应相应下调否则每一步梯度噪声过大收敛会很不稳定。第三步正式训练时轮次不需要太夸张。口罩检测这个任务本身不算难在数据量 5000 到 10000 张的规模下100 到 150 个 epoch 已经足够。再多轮次不仅浪费时间还容易过拟合到训练集里的特定灯光和角度。关键是训练过程中保留最后一轮之外的最好权重我通常会在第 80 和第 120 轮附近观察验证集 mAP选最高点的权重用于推理。2.3 评估指标解读mAP 高不代表一切毕设答辩里被问得最多的问题就是“你的模型到底准不准”。很多同学只丢出一个 mAP50这其实不够严谨。你需要同时报告精确率 Precision、召回率 Recall 和 mAP50-95这几个指标从不同侧面衡量模型。精确率回答的是“模型认为没戴口罩的人里有多少是真的没戴”召回率回答的是“画面里所有没戴口罩的人模型到底找回了多少”。对这个场景来说安全性项目往往召回率更重要——漏检一个没戴口罩的人比误报一个戴了口罩的人问题更严重。因此我在调参时会下意识地把置信度阈值调低一点比如从默认的 0.25 降到 0.15再配合 NMS 去重让模型更倾向于“多检”而不是“漏检”。mAP50-95 则是一个更苛刻的指标它计算了多个 IoU 阈值下的平均精度能反映边界框定位的精细程度。对毕设来说把 mAP50 作为主指标mAP50-95 作为辅指标是合理的呈现方式。指标它到底在说什么这个项目里的用法Precision检测出的目标有多少是准确的误报多不多Recall真实目标有多少被找出来了漏检多不多mAP50IoU 阈值 0.5 时的平均精度主评估标准mAP50-95多个 IoU 阈值的综合评价检验框定位精度FPS每秒处理帧数判断能不能实时部署3. 实操过程与核心环节实现3.1 开发环境搭建与算力准备这个项目的环境搭建其实没什么玄学。PyTorch 加上 YOLO 相关的库就够了。我不建议新手在 Linux 和 Windows 之间反复横跳用你熟悉的系统就行。我的主力环境是 Windows 11 PyTorch 2.x CUDA 12.1显卡是消费级的 8GB 显存训练 YOLOv8s 完全没压力。安装主库就一条命令pip install ultralytics这条命令会自带 torch 和 torchvision但要注意版本兼容性尤其是 CUDA 版本。安装完跑一句python -c import torch; print(torch.cuda.is_available())输出 True 才代表 GPU 可用。如果公司或实验室没有 GPU 资源也可以租云服务器但数据上传和断点续传要提前规划好别等训练到一半才发现磁盘不够。数据目录结构建议直接按 YOLO 约定来组织省去后续改路径的麻烦dataset/ ├── images/ │ ├── train/ │ └── val/ ├── labels/ │ ├── train/ │ └── val/ └── mask.yaml3.2 数据集划分与 YOLO 配置文件编写数据划分要注意一个细节同一个场景拍摄的连续视频帧不能一部分进训练集、一部分进验证集否则模型相当于见过答案再考试验证指标的参考价值会大打折扣。我遇到过一次这样的情况监控视频里同一个人在不同帧反复出现结果验证集 mAP 高达 0.98换到完全陌生的视频就掉到 0.8 以下。正确做法是按“场景”划分数据比如商场的画面只进训练集地铁站的画面只进验证集。配置文件非常简单把类别和路径填进去就行path: D:/mask_dataset train: images/train val: images/val nc: 2 names: 0: with_mask 1: without_mask这里有一点要注意路径建议用绝对路径尤其是你在多台机器之间拷贝项目时相对路径经常因为目录层级不一致而报错。3.3 训练命令与监控策略一切就绪后训练命令简洁明了yolo detect train datamask.yaml modelyolov8s.pt epochs120 imgsz640 batch16 device0modelyolov8s.pt表示加载官方预训练权重这是保证收敛速度和最终效果的关键。训练日志里除了记录每个 epoch 的 loss还会生成 confusion matrix、F1 曲线、PR 曲线这些文件训练结束后全在runs/detect/train/目录下。我要特别强调监控 loss 曲线。很多人训练跑完才打开结果看一眼这是不对的。训练前 20 个 epoch 时就应该每 10 个 epoch 打开一次 loss 曲线图看看是平滑下降还是剧烈波动。如果波动到看不出下降趋势立刻停止训练调低学习率或检查数据是否出了乱子不要指望它“再多跑几个 epoch 就好了”。损失震荡通常是优化器步长过大或标签噪声太高拖着不处理只会浪费时间。3.4 推理与可视化效果实现训练完拿到best.pt权重接下来的推理才是能让你兴奋起来的环节。调用预测接口特别简单from ultralytics import YOLO model YOLO(runs/detect/train/weights/best.pt) results model.predict(test.jpg, conf0.25, saveTrue, projectoutput, namedemo)这条代码会在 output 目录下生成带检测框的可视化图片。想要对视频检测直接传入视频文件路径就行模型会逐帧推理。但毕设答辩时的实时演示更常用摄像头画面需要写一个实时流处理脚本import cv2 from ultralytics import YOLO model YOLO(best.pt) cap cv2.VideoCapture(0) while cap.isOpened(): ret, frame cap.read() if not ret: break results model(frame, conf0.25, imgsz640, verboseFalse)[0] annotated_frame results.plot() cv2.imshow(Mask Detection, annotated_frame) if cv2.waitKey(1) 0xFF ord(q): break cap.release() cv2.destroyAllWindows()实际演示时有个小技巧如果能控制现场光线尽量选择亮度充足的机位。摄像头画面里的低光噪点会让模型误检率直线上升我自己的出街演示里最翻车的案例就是傍晚采光不足的室内连续把书包误识别成了未戴口罩的人。4. 常见问题与排查技巧实录4.1 高频问题速查表一表对照定位故障这个项目在实操中反复出现的问题其实很集中我把它们整理成了一张故障速查表。你在训练或推理过程里遇到诡异情况可以按这张表快速定位而不是毫无方向地反复重跑训练。现象常见原因我的处理办法训练 loss 一直不降学习率过高或标签文件错位先跑 10 个 epoch把学习率降到当前一半再试验证 mAP 高但实际效果差数据划分不随机同源视频进验证集按场景切分数据集不要按单帧随机切分远距离行人全部漏检输入分辨率太低将 imgsz 从 640 提到 960观察召回率变化摄像头推理很卡模型太大或笔记本 CPU 推理换用 YOLOv8n或用 TensorRT 做加速头盔、书包被误检成人脸负样本中干扰物体与正样本布局雷同专门收集一批干扰样本标注为背景类别并加入训练夜间场景效果差训练集中夜间照片太少使用数据增强模拟低光或补充夜间数据4.2 训练 loss 不降的排查套路loss 不降是所有训练任务里最让人抓狂的问题口罩检测也不例外。我第一次跑通这个项目时loss 在 0.7 左右横盘了很久无论怎么加轮次都不动。后来逐个排查代码才发现是数据集里有一个子目录的标签全部是空的等于模型在反复学习“这些图里什么都没有”梯度方向被带偏了。所以遇到 loss 不降第一步检查标签文件数量和图片数量是否一一对应。第二步打开几个 label 文件看看坐标是否分布合理有没有图片只有 0 标签但没有 1 标签这类数据倾斜会直接破坏训练平衡。第三步再考虑调学习率。多数人和我一样第一反应是调参其实数据问题比参数问题常见得多。还有一个隐蔽的坑YOLO 标签归一化后的坐标值必须在 0 到 1 之间。如果你的标注工具导出了像素坐标或格式不对训练时模型会把大量候选框过滤掉loss 看似正常但 mAP 极低。我做完一次新数据集后习惯写一个小脚本检查标签边界值确保没有越界和 NaN。4.3 小目标漏检和人群遮挡怎么改善行人场景里最典型的困难是两个远处的人脸只有几十个像素以及前后遮挡导致脸部大面积不可见。针对小目标问题除了前面说到的提高输入分辨率还有一种思路是使用多尺度训练。YOLO 训练时默认会从输入尺寸上下浮动采样适当增强多尺度策略能让模型见惯不同尺寸的目标泛化能力更好。但代价同样是训练时间变长需要平衡。遮挡问题相对更难处理。我尝试过在标注阶段把严重遮挡样本直接丢掉结果模型对部分遮挡完全免疫失败。后来反过来把所有能看出口罩状态但遮挡比例不一的行人都保留并且统一标注规范模型被迫学会从眉毛、发际线、眼镜框等局部特征推断口罩状态。测试下来对戴眼镜、戴口罩且脸被围巾半遮的行人检测稳定性明显提升。数据标注的粒度最终会直接体现为模型能力的上限。4.4 答辩展示时怎么选图和讲数据这个项目的答辩展示有两个关键加分点一是要有一张能证明“远距离检测不拉胯”的图二是要有一张能证明“密集人群不漏检”的图。哪怕你的整体 mAP 不是特别惊艳只要这两类场景表现好答辩老师就会认为你真的理解了这个应用的价值。很多同学只会挑最简单的特写图片展示那里全是人脸的正面大图模型效果当然好但反而暴露不了真实问题。我在写毕设报告时还额外加了一段多帧稳定性分析在同一段视频里统计同一个目标在连续 30 帧内检测框的坐标波动。如果模型偶尔漏检几帧、下一帧又出现说明跟踪不够稳定但这也正好可以作为下一阶段改进方向来讨论。相比空泛地说“准确率达到 95%”具体分析为什么在某些帧出现跳变能显著拉高答辩的技术深度。5. 最后再分享一个经验疫情后口罩检测的毕设热度降了一些但它的技术框架完全没有贬值安全帽检测、反光衣识别、甚至工装穿戴合规检查都是它的直接变体。你在这次项目里积累的数据处理流程、YOLO 微调方法、指标分析习惯换一个数据集就能快速复用到完全不同的领域这是我的亲身体会。如果再让我重新做一次这个项目我会在第一天就把“场景定义”写得更清楚明确监控距离是 3 米还是 10 米光照环境是室内还是室外这些条件直接决定训练数据要拍什么、分辨率要设多大。项目做到中途才发现场景假设错了返工成本会非常高。最后给你两个可以立马上手的实操建议第一训练时不要追求“一步到位”先用小模型 YOLOv8n 跑通全流程再逐步换成 s 或 m 规格这样能快速暴露数据和流程问题第二所有实验结果都用固定随机种子保存一份方便对比复现答辩前也更容易找回当初跑出最佳指标的配置。这套流程走完你拿到的不仅是一个能用的检测系统更是一套可以迁移到其他目标检测项目上的完整方法论。
阅读完成 · 觉得有帮助？
