基于神经网络的安全帽检测方法和装置

    公开(公告)号:CN112307984A

    公开(公告)日:2021-02-02

    申请号:CN202011204292.2

    申请日:2020-11-02

    摘要: 本发明提供基于神经网络的安全帽检测方法,建立第一神经网络,所述第一神经网络的基础网络为YOLOv3网络,利用第一神经网络对目标图片中的安全帽进行检测;YOLOv3网络中的骨干网络模块中增加了可变卷积处理和双注意力机制处理,并且利用上采样拼接的方式对部分特征图的尺寸进一步变小。通过本发明的方法,能够更加准确地检测出安全帽。