基于局部信息编码特征金字塔的轻量多类目标计数网络OA
Lightweight Multi-class Target Counting Network Based on Feature Pyramid with Local Information Encoding
针对现有目标计数算法抗背景杂波能力差以及对高度遮挡、尺度变化大的目标计数准确率低的问题,提出一种基于局部信息编码特征金字塔的多类目标计数网络模型.所提算法利用卷积神经网络中特征图的冗余性来搭建轻量级的骨干网络,同时添加带有局部信息编码机制的特征金字塔模块学习目标的局部特征,最后使用卷积层组成的回归头和分类头分别进行目标的数量预测和位置预测.为实现多类目标计数任务,将已有人群计数数据集ShanghaiTech和车辆计数数据集CARPK(Car Parking)的训练集混合并对之训练;为与已有方法进行对比,分别在这两类目标数据集的测试集上进行测试,并以平均绝对误差和均方误差作为计数评价指标.实验结果证明所提出算法能进行多类目标计数且在目标计数上的表现优于其他方法.
Addressing the limitations of existing object counting algorithms,which struggle with background clutter and exhibit low accuracy when dealing with heavily occluded or significantly varying object scales,we propose a novel lightweight multi-class ob-ject counting network based on feature pyramid with local information encoding(FPLE-MOCN).This model leverages the redun-dancy of feature maps in convolutional neural networks to construct an efficient and rapid lightweight backbone network.Additional-ly,a feature pyramid module with a local information encoding mechanism is introduced to capture the local features of targets.Fi-nally,regression and classification heads composed of convolutional layers are employed for predicting the number and the location of objects at the same time.To achieve multi-class object counting,we combine the training sets of the existing crowd counting data-set(ShanghaiTech)and the vehicle counting dataset(CARPK)for training.For comparison with existing methods,we evaluate our model on the test sets of both datasets separately and use both mean absolute error and mean squared error as evaluation metrics for counting.Experimental results demonstrate that FPLE-MOCN can perform multi-class object counting and outperforms other meth-ods in terms of counting accuracy.
魏祥一;张莉
苏州大学 计算机科学与技术学院,江苏 苏州 215006苏州大学 计算机科学与技术学院,江苏 苏州 215006
信息技术与安全科学
多类目标计数局部特征编码卷积神经网络轻量级骨干网络
multi-class object countinglocal feature codingconvolutional neural networkslightweight backbone network
《山西大学学报(自然科学版)》 2026 (1)
100-107,8
江苏省高校自然科学研究基金资助项目(19KJA550002)
评论