基于分割注意力与线性变换的轻量化目标检测

张艳,孙晶雪,孙叶美,刘树东,王传启

Lightweight object detection based on split attention and linear transformation

Yan ZHANG,Jing-xue SUN,Ye-mei SUN,Shu-dong LIU,Chuan-qi WANG

表 1 基于YOLOv5L改进的模型在Pascal VOC数据集上的消融实验结果

Tab.1 Ablation results on Pascal VOC dataset based on YOLOv5L model

模型 PSA 特征融合模块1 特征融合模块2 特征融合模块3 mAP/% rR/% tI/ms nr/106 GFLOPS 原始模型 — — — — 81.2 75 3.6 44.53 114.4 模型1 √ — — — 84.8 78.5 21.1 44.39 113.9 模型2 √ √ — — 85.6 82.3 9.6 36.47 107.4 模型3 √ — √ — 86.8 82 8.4 36.47 107.4 本研究模型 √ — — √ 85.7 81.8 8.1 28.49 100.6