公开(公告)号：CN117974986A

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

申请号：CN202410209117.4

申请日：2024-02-26

Applicant: 中国石油大学(华东)

Inventor： 高阳 ,  齐玉娟 ,  舒搏锋 ,  王子杰

IPC: G06V10/25 ,  G06V40/10 ,  G06V10/82 ,  G06N3/0464 ,  G06N3/08

Abstract: 本发明提供了一种基于改进的YOLOv5及注意机制的安全帽检测方法，属于图像处理技术领域。所述方法具体包括以下步骤：由实际工地场景构建施工现场场景下的安全帽数据集；改进基于YOLOv5安全帽检测模型，在Neck网络中使用BiFPN结构代替FPN结构，以提高网络对不同尺度目标的检测能力；添加通道先验卷积注意力CPCA模块，优先分配注意力给小目标同时提升对安全帽佩戴检测精度；将YOLOv5的损失函数替换为一种能够在不同的检测任务中聚焦不同的回归样本的Focaler‑IoU损失函数，以提高回归结果的精确度，本发明能够准确地检测人员安全帽的佩戴情况，提升了对小目标的检测精度，减少漏检误检的情况。