基于YOLOv5 的车辆目标检测算法轻量化改进

  • 打印
  • 收藏
收藏成功


打开文本图片集

摘要:针对传统路端车辆目标检测算法参数多、检测速度较慢等问题,提出了基于YOLOv5 的车辆目标检测算法轻量化改进。首先,选用轻量化EfficientnetV2 卷积神经网络对原骨干网络进行重构,同时在网络中引入GAM 注意力机制;其次,为平衡CIoU 损失和IoU 损失在损失函数中的权重,引入α-CIoU 损失代替原有的CIoU 损失;最后,使用soft-NMS 算法替换原有的NMS 非极大值抑制算法。(剩余2924字)

monitor
客服机器人