低算力高精度
面向扫地机器人实时检测的低算力网络,通过改进池化与注意力结构在有限计算资源下实现更高精度。
Lv, Yong · Zhou, Yu · Qing, Chen · 迟文政 · 孙立宁 · Lei, Yu
Engineering Applications of Artificial Intelligence 2024
改进的最大池化结构保留非最大响应,防止有用特征被丢弃,提升特征表达能力。
在骨干网络后加入 CBAM,同时关注通道和空间维度,强化关键特征,抑制噪声。
解耦头将分类和回归分支分开,避免共享特征带来的任务冲突,提高定位和分类精度。
优化分类损失方法,加大对困难样本的惩罚,提升检测性能。