一种安全帽检测方法及装置
    3.
    发明公开

    公开(公告)号:CN115761831A

    公开(公告)日:2023-03-07

    申请号:CN202211117881.6

    申请日:2022-09-14

    Abstract: 本申请实施例提供一种安全帽检测方法及装置,包括初始化多尺度生成对抗网络;将采集的安全帽正常图像缩小至预定尺寸,将缩小后的安全帽正常图像输入多尺度生成器,由多尺度生成器输出安全帽重构图像;将安全帽重构图像和采集的安全帽异常图像输入鉴别器,由鉴别器输出安全帽重构图像是否为异常图像;对多尺度生成器和鉴别器进行训练直至收敛,得到训练后的多尺度生成对抗网络;利用训练后的多尺度生成对抗网络生成若干安全帽异常图像;构造包括采集的安全帽正常图像、安全帽异常图像和生成的安全帽异常图像的样本集,利用样本集对机器学习模型进行训练,得到安全帽检测模型,该模型能够准确检测施工人员是否佩戴安全帽,以及安全帽是否佩戴规范。

Patent Agency Ranking