公开(公告)号：CN118942121A

公开(公告)日：2024-11-12

申请号：CN202411121492.X

申请日：2024-08-15

Applicant: 重庆邮电大学

Inventor： 朱红军 ,  谢雨欣 ,  韦庆杰 ,  钱鹰 ,  王奕琀 ,  刘苏 ,  刘歆

IPC: G06V40/10 ,  G06V20/52 ,  G06V10/44 ,  G06V10/80 ,  G06V10/764 ,  G06V10/82 ,  G06N3/0464 ,  G06N3/048 ,  G06N3/08

Abstract: 本发明涉及一种针对小目标的安全帽佩戴检测系统及方法，属于计算机视觉技术领域，包括采集建筑工地现场图像并进行数据标注，构建训练数据集和测试数据集并进行预处理；基于预设YOLOX模型构建针对小目标的安全帽佩戴检测网络，所述安全帽佩戴检测网络用于安全帽佩戴检测；将训练集图像和测试集图像输入到所述安全帽佩戴检测网络中进行训练，并采用WIoU v3作为边界框回归损失函数调整所述安全帽佩戴检测网络的参数；将待测建筑工地现场图像输入到训练好的安全帽佩戴检测网络，输出得到所述待测建筑工地现场图像的实时检测结果。本发明可以提高安全帽佩戴检测的精度，以实现对场景中安全帽佩戴情况的识别和检测。