基于改进YOLOv5的安全帽佩戴检测算法_张锦

针对现有安全帽佩戴检测干扰性强、检测精度低等问题,提出一种基于改进 YOLOv5的安全帽检测新算法。首先,针对安全帽尺寸不一的问题,使用K-Means++算法重新设计先验框尺寸并将其匹配到相应的特征层;其次,在特征提取网络中引入多光谱通道注意力模块,使网络能够自主学习每个通道的权重,增强特征间的信息传播,从而加强网络对前景和背景的辨别能力;最后,在训练迭代过程中随机输入不同尺寸的图像,以此增强算法的泛化能力。

  • 2023-02-06
  • 收藏0
  • 阅读682
  • 下载0
  • 9页
  • pdf
  • 3.81M

评价

评分 :
   *