公开(公告)号：CN117975534A

公开(公告)日：2024-05-03

申请号：CN202410176684.4

申请日：2024-02-08

Applicant: 福建信息职业技术学院

Inventor： 林炳炎 ,  侯麦笛 ,  吴邦辉 ,  林洁朗

IPC: G06V40/16 ,  G06V10/25 ,  G06V10/764 ,  G06V10/766 ,  G06V10/80 ,  G06V10/82 ,  G06N3/045

Abstract: 本发明涉及一种基于改进YOLOv8的口罩检测装置及方法，属于图像处理技术领域，包括步骤A：自建包含人脸和口罩的图像数据集，并进行口罩位置和口罩类别的标注；步骤B：采用Mosaic数据增强增加各个层次大小数据集数量；并将对应的口罩位置标注按照相同的方式进行组合和调整；步骤C：对YOLOv8模型进行改进，在颈部网络采用Slim‑Neck结构，在检测层采用DyHead检测头，构建YOLOv8‑SLIM‑DYHEAD模型；步骤D：利用数据增强后的数据集对YOLOv8‑SLIM‑DYHEAD模型进行训练，通过最小化损失函数优化模型参数并获得最优YOLOv8‑SLIM‑DYHEAD模型；步骤E：将待检测的数据输入到训练好的YOLOv8‑SLIM‑DYHEAD模型，对口罩佩戴情况进行检测；本发明利用构建的YOLOv8‑SLIM‑DYHEAD模型进行口罩检测，在保持精度的同时降低模型的复杂度。