AG-YOLOv5检测算法

端侧低延迟高精度

通过轻量化骨干与注意力机制,在边缘计算设备上实现检测精度与速度的平衡。

Yanqiu, Wang · 金尚泰 · 殷辰堃 · Zejin, Wu

2023

技术优势

推理更快

用轻量化模块替换骨干网络中的标准卷积和 CSP 模块,减少计算量。

精度更高

在骨干网络末端和颈部添加注意力机制,增强关键特征提取。

损失收敛更稳

采用 WIoU 替代 CIoU 作为损失函数,并采用 HardSwish 激活函数,综合提升模型性能。

应用场景