基于注意力机制的多任务目标计数系统设计

作     者：李永慧 LI Yonghui

作者机构：山西科技学院光机电工程学院山西晋城048000 

出 版 物：《电视技术》 (Video Engineering)

年 卷 期：2024年第48卷第7期

页      面：47-52页

学科分类：08[工学] 0835[工学-软件工程] 081202[工学-计算机软件与理论] 0812[工学-计算机科学与技术（可授工学、理学学位）] 

主　　题：目标计数 注意力机制 多任务学习 交叉特征金字塔网络 

摘      要：提出基于注意力机制的深层神经网络用于目标计数,其任务是为输入图像的目标进行精确数目统计。该网络模型同时引进多任务学习方法,多尺度融合得到密度特征图和注意力特征图进行目标计数。首先,使用交叉特征金字塔网络进行特征提取;其次,将提取的特征分别用于密度特征图及注意力特征图进行交叉融合;最后,通过多任务学习将两个输出特征图逐元素运算,得到精确的密度特征图。提出的网络模型在行人检测数据集(ShanghaiTech)与多类别的行为识别数据集(UCF_CC_50)上进行了训练与测试,实验结果表明,通过在各个分支引入注意力机制,可以有效提高整个模型预测结果的准确率。