基于改进YOLOv5的道路目标检测

  • 打印
  • 收藏
收藏成功


打开文本图片集

摘要:为提高自动驾驶中的道路目标检测精度,设计了一种基于YOLOv5的道路目标检测模型。该模型在YOLOv5s的网络模型基础上,将原始的初始锚框聚类算法改为K-means+[KG-*4]+算法来减小随机带来的聚类误差;并在Backbone中SPP模块之前引入SENet注意力机制,以增强道路目标重要特征并抑制一般特征,达到提高检测网络对道路目标的检测能力。(剩余10550字)

monitor
客服机器人