基于多尺度金字塔Transformer的人群计数方法

张少乐，雷涛，王营博，周强，薛明园，赵伟强

（1.陕西科技大学 电气与控制工程学院, 陕西 西安 710021; 2.陕西科技大学 电子信息与人工智能学院, 陕西西安 710021; 3.陕西科技大学 陕西省人工智能联合实验室, 陕西 西安 710021; 4.中电科西北集团有限公司西安分公司, 陕西 西安 710065）

随着城市化进程的不断推进和人口数量的持续增长，城市公共场所的人口密度也不断提高，而这些人口密集场所的管理和安全问题成为城市管理者和公共安全机构的头号难题[1]。为解决这些问题，人群计数应运而生。

人群计数是计算机视觉领域中一个重要的研究方向，其主要应用于公共场所人群数量的实时监测和分析，从而实现对人流的预测、规划和管理。这项技术可以广泛应用于人口密集的公共场所，如商场、机场、火车站、地铁站等。人群计数可以实现对人员数量、流量和分布的实时监测和分析，帮助城市管理部门进行人流预测和规划，减少拥堵和安全隐患，提高城市管理效率[2]。人群计数的另一个重要应用领域是应急救援。在自然灾害和人为事故发生时，人群计数可以快速计算出现场人员的数量和位置，协助应急救援部门对灾害现场进行快速响应和救援。在冠状病毒（COVID-19）疫情期间，人群计数也被广泛应用于公共场所的人员数量管控，帮助政府和公共场所管理者控制人员流动，减少病毒的传播。

然而，人群计数技术在实际应用中仍然面临很多挑战，其中最为艰巨的挑战为：1)计数算法对复杂场景的适应性低，在人群密集的公共场所中，人员数量众多、行为复杂，场景复杂多变；2)场景中的透视畸变、光照变化、复杂的人群形态等问题也会对计数结果产生影响，高鲁棒性的计数算法才能适应复杂多变的场景环境；3)图像特征利用率低，不同尺度的特征语义信息不同，为了尽可能挖掘当前图像蕴含的细节信息，需要增强融合图像中不同尺度的特征。

为了更好地应对密集人群计数任务中图像场景中背景复杂、目标尺度变化较大导致人群计数精度较低的问题，本文提出一种基于多尺度金字塔Transformer的人群计数方法(multi-scale pyramid Transformer network, MSPT-Net)。针对人群场景背景复杂的问题，设计一个新颖的Transformer结构，该结构通过深度可分离自注意力模块同时提取局部与全局特征信息，增强网络特征表达能力。针对目标尺度变化大、遮挡等导致图像特征利用率低，不同尺度的特征语义信息不同的问题，设计特征金字塔模块提取浅层和深层地特征，设计多尺度感受野的回归头模块，通过多尺度的空洞卷积和金字塔平均池化增强特征捕获能力，并能够有效地融合不同尺度的特征进而预测密度图。在训练过程中，使用深度监督，对网络中间的每个阶段添加了额外监督损失，让网络更好的收敛。同时，针对于人群计数任务中人头位置标注繁琐问题，本文使用弱监督学习策略，与目前主流的人群计数方法相比较在计数方面接近最先进的水平。

1 相关工作

早期的人群计数方法通常使用基于检测的方法[3]，如滑动窗口和头部检测等。这些方法虽然简单易行，但往往受到高密度、目标重叠和背景干扰等因素的影响，导致计数结果不准确[4]。为了解决这些问题，一些基于回归的方法被提出，如Chen等[5]提出的累积属性支持向量回归机(cumulative attribute for support vactor regression,CA-SVR)，该方法的主要思想是学习一种特征到人群数量的映射，通过建立群体数与人群密度图之间的回归模型，可以很容易地获得总体计数结果。虽然基于回归的方法在整体上提升了计数性能，但是它们忽略了图像中的空间信息，只得到一个计数结果，缺乏可靠性和可解释性。

随着计算机视觉技术的发展，基于深度学习的人群计数方法逐渐成为主流。其中，基于密度图的方法是目前最为流行和有效的方法之一，该方法通过将输入图像利用几何自适应高斯核估计图像中每个人头的大小并转换为密度图，然后通过回归密度图中的像素数量估计人群数量[6]，可以在复杂场景下获得更精确的计数结果，如图1所示。

图1 基于密度图回归的方法Fig.1 Methods based on density map regression

伴随着卷积神经网络在计算机视觉领域中迅速发展，使得许多新的人群计数方法[7-13]被提出。其中，针对多尺度变化问题，Zhang等[7]提出多列卷积神经网络(multi-column convolutional neural network, MCNN)，使用大小不同的卷积核提取人群的多尺度信息。针对多列结构难以训练、结构冗余、输入图片需要根据密度分级等原因，Li等[8]提出空洞卷积神经网络模型(congested scene recognition network, CSRNet)，首次将空洞卷积用于人群计数，通过保持分辨率的同时扩大感受野，保留了更多图像细节信息。然而，这些方法不能有效处理复杂场景和多尺度问题，导致人群计数的准确性和稳定性较低。为了解决这些问题，研究人员又提出许多改进的方法，如多尺度卷积神经网络[9]、注意力机制[10]、可形变卷积网络[11]、生成对抗网络[12]等。此外，还有学者[13-14]研究了基于密度图的后处理技术，如弱监督学习[13]、多任务学习[14]等，为准确估计人群数量奠定了基础。

随着计算机视觉技术的不断发展，研究人员发现，使用简单的卷积神经网络模型难以处理复杂场景下的密集人群计数。因为在这些复杂场景下，人群的位置相互接近，互相遮挡，使得人数估计变得非常困难。因此，如何处理密集人群计数问题成为研究人员关注的重点。注意力机制[15]的本质是从关注全部到关注重点，具有参数少、速度快和效果好等优点。在人群计数方法中引入注意力机制使得人员遮挡、人群分布不均匀等问题得到解决。Liu等[16]提出一种融合注意力机制的可变形卷积网络(attention-injective deformable convolutional network for crowd, ADCrowdNet)，通过引入注意力机制强调了人群区域，并使用可变形卷积保证了在高度拥挤场景中密度图的准确性。最新研究[17-18]侧重于设计不同的注意力机制，以关注全局背景下尺度和密度的变化。同时，一些研究通过优化新的图像增强和损失函数提高计数的准确性[19-20]。然而，这些方法通常需要足够的数据和经验，而且网络结构设计较为复杂，没有显著的改进效果。

近年来，Transformer[21]已成为深度学习领域的一个热门模型，在各种任务中都取得了非常优秀的效果。Transformer是一种基于自注意力机制的神经网络，可以处理输入序列中的全局依赖关系。值得关注的是，视觉自注意力模型(vision Transformer, ViT)[22]的出现引起了计算机视觉领域的广泛关注，并在人群计数任务中显示出良好的结果。一种基于Transformer的弱监督人群计数模型(weakly-supervised crowd counting with Transformer, TransCrowd)[13]被提出，这是第1个基于Transformer的人群计数研究，它从基于Transformer序列计数的角度重新构造了人群计数问题，通过使用Transformer的自注意机制来有效地提取语义人群信息。实验结果表明，该模型在密集人群计数任务中取得了非常好的性能。虽然Transformer在提取上下文信息具有强大的优势，但是在局部信息获取上仍然不足，所以如何获取图像的局部信息及全局信息进而提高计数精度仍是人群计数任务的研究重点。综上所述，Transformer在人群计数领域的应用已经得到了广泛的研究和应用，并且不断有新的方法和改进被提出，为人群计数的研究和实践带来了新的思路和可能性。

2 多尺度金字塔Transformer网络

本文提出的多尺度金字塔Transformer网络结构如图2所示，网络整体包含块嵌入(patch embedding)、Transformer模块、特征金字塔模块(feature pyramid module, FPM)、多尺度感受野的回归头模块4个模块。

图2 多尺度金字塔Transformer网络结构Fig.2 Inter-scale pyramid Transformer network architecture

输入图片经过块嵌入模块分割成固定大小的图像块。输出被展平为一维向量序列输入Transformer模块提取全局特征。每个阶段的一维序列都会被重塑为二维的特征图并经过特征金字塔模块上采样到相同的分辨率，以方便之后的相加操作。三分支的特征图都通过一个多尺度空洞卷积模块(multi-scale atrous convolution, MAC)增强特征并回归出中间密度图，为之后的深度监督做准备，这3个分支的损失函数分配较小的权重。相加之后的特征图经过一个金字塔平均池化模块(pyramid average pooling, PAP)，回归出最终的密度图，该分支的损失函数位置分配较大的权重。

2.1 金字塔Transformer骨干网络

目前在人群计数任务中，为解决人群场景背景复杂问题，大多数工作还是基于卷积神经网络，但由于卷积神经网络卷积核的感受野固定，其难以捕获全局上下文信息。Transformer是一种基于自注意力机制的神经网络，可以处理输入序列中的全局依赖关系，但其缺少对局部信息提取的能力。基于此，本文将Transformer块堆叠设计成局部注意力和全局注意力交替使用的形式，使其同时拥有局部和全局的感受野进而捕获局部关系和进行全局上下文建模。同时，为了充分利用目标多尺度信息，设计了多层的特征金字塔结构，利用不同阶段的特征图进行浅层和深层特征的信息交互，增强网络的表达能力。

2.1.1 特征图预处理

在人群计数任务中，图像通常是高维数据，具有很大的空间结构，同时也具有丰富的语义信息。为了对图像进行有效地处理，需要将这些高维的数据通过块嵌入模块转换为低维的向量表示，以便能够在网络中进行处理和分析。输入特征图经过块嵌入模块，其将输入图像在进入Transformer模块时转换为一维向量。假设输入特征图大小为x∈RH×W×3，其中H、W、3分别代表高度、宽度和通道数。通过块嵌入层将图片裁剪为HW/K2个尺寸相同的块，每个块的大小为K×K×3，本文将这个二维序列展平为一维向量i∈RN×D，其中N=HW/K2，D=K×K×3。块嵌入模块可以将输入的图像尺寸统一化，并且在进行特征提取时可以避免过拟合，因为每个图像块的特征表示都是独立的。此外，块嵌入模块在进行特征提取时考虑了图像的局部信息，因此对于网络可以提供更好的性能。

2.1.2 Transformer模块

由于输入图像通常是高分辨率，这使得Transformer全局注意力策略的计算量会随着图像的分辨率成二次方增长，在密集预测任务中会呈现高计算复杂度，如给定h×w的输入，维数为C的Transformer计算复杂度为Ω(4hwC2+2(hw)2C)，由于h×w比较大，所以本文旨在将输入的平方降为线性的，从而大幅度降低计算量。因为金字塔视觉自注意力模型 (pyramidvisiontransformer,PVT)[23]能够很好地用于密集的像素级预测任务中，所以本文对基于PVT的Transformer模块的全局注意力策略进行了优化改进，提出深度可分离卷积自注意力(depthwise separable self-attention,DSSA)模块，该模块能提取局部与全局特征信息，增强了网络特征表达能力，同时降低了网络的复杂度，如图3所示。从自注意力机制的效率和感受野角度出发，将Transformer块堆叠设计成局部注意力和全局注意力交替使用的形式，拥有局部和全局的感受野可以捕捉短距离和长距离的关系。

图3 深度可分离卷积自注意力模块Fig.3 Depth-separable convolution self-attention module

受深度可分离卷积[24]中的分组设计的影响，首先将第l层Zl-1的输入一维向量进行重塑变成h×w的二维特征图，其次在局部窗口内对每个特征图进行空间分组，计算局部分组自注意力(locally-grouped multi-head self-attention, L-MSA)，最后计算全局下采样自注意力(global-downsampling multi-head self-attention, G-MSA)。具体而言，在LMSA阶段，二维特征图被平均划分为M×M大小的子窗口，仅在子窗口内部进行自注意力计算，计算量会大大减少，L-MSA的计算复杂度和时间复杂度为

VIT[22]中MSA模块的计算复杂度和时间复杂度为

式中：h×w为特征图的高度和宽度，C为特征图的深度，M为每个窗口的大小。

相比于VIT中的自注意力，计算复杂度和时间复杂度从输入的平方降为线性的，并且每个子窗口之间没有交互通信，以此获得局部特征信息。

在G-MSA阶段，从每个子窗口提取一个低维度特征作为各个窗口的表征，然后基于这个表征再与其他原始窗口的每个像素点进行注意力操作，相当于自注意力中键的作用，一直重复此过程直到所有窗口间都进行了注意力计算。在这个过程中通过下采样每个子窗口使其包含整个窗口的所有信息，同时用这个特征与每个窗口进行交互，能够实现全局交互。并且G-MSA过程中的K、V是在缩小特征的基础上计算的，但Q是全局的，因此注意力仍然可以恢复到全局，这种做法显著减少了计算量。计算复杂度和时间复杂度为

同时网络结构中也交叉了必要的多层感知器(multi-layer perceptron, MLP)，归一化层(layer normalization, LN)和残差连接等。如图1 Transforme模块所示，Zl-1、Zl、Zl′、Zl′′、Zl′′′分别表示每层的输出。

整个Transformer块可以表示为

该注意力机制是对图像特征的空间维度进行分组，分别计算各局部空间的自注意力，再利用全局自注意力机制对分组注意力结果进行融合，这种机制计算效率更好，性能更优，同时易于部署，便于实际应用。

2.1.3 特征金字塔模块

尽管Transformer能够提取全局特征，但是提取出的深层特征图在通过上采样后，仍会缺少很多细节信息。此外，深层次的特征虽然包含了丰富的语义信息，但是由于分辨率低通常较模糊，难以区分不同对象的边界，这使得人群计数网络很难准确获取人群的位置信息。相比之下，浅层的特征虽然语义信息较少，但由于分辨率高，可以准确地包含物体的位置信息[25]。为了充分利用深层的语义信息和浅层的细节信息，设计了一个特征金字塔模块，如图2中的特征金字塔模块(feature pyramid module, FPM)所示。它能够有效地融合深层和浅层的特征，增强多尺度信息交互，从而提高小目标检测的精度。具体而言，将所有阶段的特征图上采样到输入图像的1/8大小，方便与其他方法进行比较[8]。

本文设计的基于深度可分离自注意力的金字塔Transformer主干网络结构，不但能有效捕获图像的局部和全局信息，而且在不显著损失性能的情况下能够降低计算量。设计的特征金字塔结构实现了密集人群图像浅层细节特征和深层语义特征的高效融合，能够补充深层特征图在通过上采样后缺失的细节信息。

2.2 多尺度感受野的回归头

因为本文的Transformer骨干和特征金字塔模块已经捕获了足够的局部和全局信息，所以，只需要使用一个简单的回归头来回归精确的密度图。但针对人群计数场景目标尺度变化的问题，引入空洞卷积[26]不仅可以减少参数，而且还可以通过调整卷积核的空洞大小增加其感受野来实现不同尺度的特征提取，从而提高模型准确性。考虑到不同阶段语义信息的丢失，本文采用多尺度空洞卷积模块和金字塔平均池化模块进行特征增强，以进一步提高模型的性能。

2.2.1 多尺度空洞卷积模块

由于人群计数场景复杂，实际图像存在人群密度明显变化及图像严重遮挡等问题。虽然改进的Transformer模块可以同时提取全局和局部信息，但是不同阶段的语义信息内容丰富程度也不同，通过特征金字塔模块上采样后对于人群密度和不同场景信息可能会丢失，为了加强网络对于不同尺度目标捕获的特征并且精确回归密度图，本文设计了一个多尺度空洞卷积模块，该模块通过并行堆叠具有不同膨胀率的空洞卷积层，增大了感受野，更适用于人群密度估计任务，如图4所示，它能够用于挖掘图像细节信息，加强对特征的提取。

图4 多尺度空洞卷积模块Fig.4 Multi-scale dilated convolutional module

具体而言，多尺度空洞卷积模块包含3列路径和1×1路径，每列由单个卷积层和空洞卷积层组成。本文将相应的卷积核和空洞率设置得尽可能小，目的是适应小规模对象的人群计数场景。每个卷积层之后是归一化(batch normalization,BN)层和ReLU激活函数。每列输出的特征图和快捷路径相加以利用它们的多尺度特征，使得多尺度空洞卷积模块更轻量化的同时获得更好的性能。

2.2.2 金字塔平均池化模块

在通过多尺度空洞卷积模块后，已经获得了充分的结构和空间信息。但在人群计数任务中，同一个场景中往往存在不同尺度的人头大小，此时需要应对人头尺度变化并减少解码部分的信息损失。为防止丢掉太多高维语义信息，为此设计了金字塔平均池化模块，用于融合不同尺度的特征图。该模块通过不同层级输出不同尺度的特征图，以保持全局特征的权重，并通过1×1卷积核处理每个金字塔层级后的输出。通过双线性插值直接对低维特征图进行上采样，使其与原始特征图尺度相同。将不同层级的特征图拼接为最终的金字塔，平均池化全局特征。通过一层1×1卷积生成最终的密度图。该模块能够应对人头尺度变化，帮助提高人群计数的准确性，如图5所示，它可以满足不同尺度的输入图像，在不丢失信息的情况下可以提取每个空间位置的特征。

图5 金字塔平均池化模块Fig.5 Pyramid average pool module

本文采用多尺度空洞卷积模块和金字塔平均池化模块进行特征增强并作为回归头，是因为其简单且高效。在特征图相同情况下，空洞卷积可以得到更大的感受野，更大的感受野可以提高在密集任务中小物体识别分割的效果，从而获得更加密集的数据。金字塔平均池化可以提取不同尺寸的空间特征信息，提升网络对于空间布局和人头尺度变化的鲁棒性。

2.3 深度监督

2.3.1 全监督损失函数

本文设计一种基于基于分布匹配(distribution matching, DM-Count)的损失函数[20]。在人群计数过程中，使用高斯平滑每一个注释点会损害泛化性能，因此采用分布匹配方法进行人群计数。DM-Count使用计数损失、最优传输损失和总变量损失3个项来制定损失函数。

人群计数的目标是使||||尽可能接近1||Z||1，计数损失被定义为它们之间的绝对差：

式中：Z和分别代表真值和预测密度图，||·||1代表L1范数。

最优传输是指将一种概率分布转化为另一种概率分布的最优成本。本文使用最优传输来度量预测密度图和地面真实密度图之间的相似性，因为最优传输损失可使模型具有强大的拟合能力。最优传输成本也可以通过双重公式计算为[20]

式中：µ和ν分别为定义在向量空间上的2组点x和y上的2个概率度量。

最优传输损失LOT定义为

式中：W为式(9)代表的最优传输成本，α*、β*为式(9)的解。

DM-Count中的总变量损失使用了原始的地面真相的头部注释，不够平滑，无法建立一个强有力的人物形象。特别是在一些稀疏的场景中，人群的规模更大，用像素表示一个人是不合理的。为了解决这个问题，采用均方误差损失（即L2损失）正则化预测与真值之间的差距。L2损失为

式中N为样本数。

因此，通过计数损失L1、最优传输损失LOT及均方误差损失L2的加权和制定的损失函数LM为

式中：λ1和λ2为损失系数，在DM-Count[20]中分别设置为0.01和0.1。

为了解决训练过程中的梯度下降和梯度消失的问题，本文设计了一个辅助损失函数增强训练[27]。由于不同深度的卷积层对最终损失的贡献不同，一般来说，网络越深，感知范围越广，提取到的特征表达能力越强，因此较深层的特征图输出在某种程度上比浅层的输出更重要。根据这个原则及相关实验的支持，将辅助监督损失均设置为均方误差损失(L2损失)，每个阶段的辅助损失如图2所示，并将不同深度的辅助损失函数的权重设置为可学习的ni,i∈(1,2,3), 所以最终的损失函数为

2.3.2 弱监督损失函数

为了训练方便，本文对3个多尺度分支和多尺度融合网络设计了4个smoothL1损失，因为不同图像中人群的数量变化很大，并且L1、L2损失单独对离散值很敏感，存在梯度爆炸问题。因为网络的不同分支在检测不同尺寸的头部时将具有不同的精度，所以损失函数中使用任务特定的可学习权重ωi,i∈(1,2,3,4)，本文的弱监督损失函数定义为

3 实验结果与分析

为了验证本文提出方法的有效性，设计了一系列实验进行探讨分析，验证所提出方法的有效性和优越性。在本节中，首先说明实验细节、数据集和评估指标，其次展示所提出的MSPT-Net在3个数据集上的实验结果以及可视化结果，最后在ShanghaiTech数据集[7]进行消融研究。

3.1 实验细节

本实验在训练的过程中主要在NVIDIA Ge-Force RTX 3 090 24 GB、Python3.7和PyTorch 1.7的服务器上实现，骨干采用的是基于PVT改进模型。ST_PartB和UCF-QNRF的裁剪大小均为512，ST_PartA和NWPU裁剪大小均为256。本文使用了AdamW优化算法并且训练批次大小为8，初始学习率设为1×10-5。

3.2 数据集

ShanghaiTech数据集包含1 198张图片，共标记了330 165个人头数量。数据集分为Part A和Part B 2部分。Part A包含482张图像，图像目标较为密集，其中训练集300张图像，测试集182张图像，图像平均分辨率为589×868。Part B共包含716张图像，图像目标较为稀疏，其中训练集400张图像，测试集316张图像，图像平均分辨率为768×1 024。

UCF-QNRF数据集[28]共包括1 535张人群图像，其中训练集1 201张图像，测试集334张图像。就注释数量而言，UCF-QNRF是迄今为止最大的数据集，可用于训练和评估大规模人群密集计数模型。该数据集的图像是从不同网站收集的，这些图像在场景和图像大小方面都具有很大的多样性。其中训练集1 201张图像，测试集334张图像，图像平均分辨率为2 013×2 902。

NWPU数据集[29]是目前最大的人群计数数据集，由5 109张高分辨率图像和213万多条注释组成。图像中的人数范围从0～20 033。此外，在这个数据集中引入了负样本，它指的是没有人或具有与人群场景相似纹理的图像。其中训练集3 109张图像，验证集500张图像，测试集1 500张图像，图像平均分辨率为2 311×3 383。

3.3 评价指标

采用平均绝对值误差(mean absolute error, MAE)和均方根误差(root mean square error, MSE)用于评价模型性能，即

式中：N为来自测试集的样本数，Yi和分别为第i个测试图像中的预测数量和真实数量。通过对模型输出的人群密度图求和获得预测计数。对于MAE和MSE，如果值越小，则测试样本越接近真实人数。

3.4 实验结果对比

为了评估所提出模型的有效性，在3个公共的数据集上进行了实验验证，包括Shanghai-Tech[7]、UCF-QNRF[28]和NWPU[29]数据集。此外，在全监督方法上也与CSRNet[8]、DM-Count[20]、SUA-Fully[30]、MFP-Net[18]、TransCrowd[13]、DLMPNet[31]、SC2Net[32]、FIDTM[33]等先进的网络模型进行比较。同时，本文在弱监督方法上也与CCSLL[34]、MATT[35]、TransCrowd[13]、MPS[36]进行比较。具体结果如表1所示，其中√表示选择方式，—表示未选择。

表1 不同网络在Shanghai Tech、UCF-QNRF、NWPU数据集上的对比结果Table 1 Comparison results of different networks on Shanghai Tech, UCF-QNRF, NWPU datasets

在ShanghaiTech数据集上将本文方法与经典方法进行比较。在Part A部分，本文全监督方法与经典方法CSRNet相比，MAE和MSE分别降低了15.1和26.7，本文全监督方法与最新的方法FIDTM相比，MAE和MSE分别降低了3.9和15.1。在Part B部分，本文全监督方法与CSRNet相比，MAE和MSE分别降低了3.6和4.9，本文全监督方法与最新的方法FIDTM相比，MSE降低了0.7。本文弱监督方法甚至超越了一些全监督方法，本方法在Part A中获得MAE为65.5、MSE为69.9的优秀弱监督成绩，相比较于基准TransCrowd有着大幅度的提升。在Part B中，相较于最新弱监督方法MPS，MAE和MSE分别降低了1.8和2.2。实验结果表明，本文方法相比于传统的卷积神经网络方法有着明显的优势。与主流弱监督方法相比，仍然有着很强的竞争力。

为了进一步验证本文所提模型的泛化能力，在UCF-QNRF、NWPU数据集上也进行了实验。在UCF-QNRF数据集上本文全监督方法与CSRNet相比，MAE和MSE分别降低了38.8和68.6，本文方法与最新方法FIDTM相比，MAE和MSE分别降低了6.5和14.1，本文方法再次优于其他方法。同时在弱监督方法上，本文方法相比于基线TransCrowd和最新弱监督方法MPS仍然有着大幅度的提升。这是因为本文设计的特征金字塔模块可以包含更多细节信息，有助于检测小物体。同时本文设计的多尺度空洞卷积模块与金字塔平均池化模块，可以更好地从Transformer中捕获多尺度特征和全局上下文信息来回归人群数量。

在NWPU数据集上全监督方法与CSRNet相比，MAE和MSE分别降低了116.7和172.7，本文方法与最新方法FIDTM相比，MAE降低了12.1。在弱监督方法上，本文方法有着更大的提升。实验结果表明，本文方法在处理大规模数据时表现出卓越的性能，同时相对于其他主流算法仍能够保持准确的计数结果和较高的鲁棒性。

3.5 模型轻量化分析

为了客观衡量所提出模型的轻量化，对模型参数(Params)和计算量(FLOPs)的定量对比进行了实验验证，为保证实验数据的合理性，本文在基于Transformer的人群计数框架上进行重点对比试验分析，同时，对CNN网络模型也进行了数据对比以作参考。其中，Params和FLOPs越小越有利于轻量检测，GFLOPs是在256×256的输入规模下计算的。实验结果如表2所示。

表2 不同模型轻量化指标对比Table 2 Comparison of lightweight indexes of different models

对表2中数据进行分析发现，对于MCNN、CSRnet、DM-Count等均未使用Transformer的CNN网络模型，参数量相对较低，但FLOPs相对较高。本文方法的参数量为26.93 MB，FLOPs为9.63 GB，基准方法TransCrowd参数量为86.8 MB，FLOPs为49.3 GB，在性能提升的同时，参数量以及计算量也有着显著的下降。

3.6 可视化

为了验证本文方法的有效性，选择了5类有代表性的样本，图6给出了在NWPU数据集使用不同方法上预测密度图的比较结果。在每张图片的右下角，标注了预测密度图的计数结果。第1列是负样本，样本信息中无人物出现，其纹理信息与密集人群的纹理信息相似，由于CSRNet直接融合所提取的特征而不进行区分，导致如第3行第1列的图像所示的较差的预测结果。而本文提出的MSPT-Net使用金字塔Transformer去提取不同阶段的特征，进而感知背景与细节信息，可以抑制背景噪声，提高模型的泛化能力。第2、3列为光照存在巨大差异的样本，在光照不一的照明条件下，本文模型仍实现了精细的预测结果，这表明本文模型具有很强的鲁棒性。第4列为严重遮挡以及人头尺度不一的样本、第5列为高密集人群样本，很明显，MSPT-Net可以提供良好的预测结果，由于CSRNet是单列网络没有融合多尺度信息、TransCrowd只捕获全局信息丢失了局部信息导致预测图像人头丢失，而本文提出的方法使用深度可分离自注意力模块能有效获取图像的局部和全局信息，同时，多尺度感受野的回归头模块能有效增强捕获的特征信息，所以本文方法能够更好地解决人群计数面临的计数场景复杂、目标尺度变化较大等问题。这些可视化结果表明，本文所提出的方法能够在不同场景、不同人群密度下有效地学习人群图像和人群密度图之间的映射关系，表明本文方法具有很强的鲁棒性。

图6 NWPU数据集不同方法的部分可视化结果Fig.6 Partial visualization result plots of different methods for the NWPU dataset

3.7 消融实验

为了训练方便，将Part A中的图像随机裁剪成一些尺寸为256×256的小尺寸图像，Part B中的图像随机裁剪成尺寸为512×512的图像。为了验证所设计的特征金字塔模块（FPM）、多尺度空洞卷积模块（MAC）和金字塔平均池化模块（PAP）的有效性，本文在ShanghaiTech数据集进行了消融实验，实验结果如表3和表4所示。

表3 关于FPM不同阶段特征的消融实验Table 3 Ablation experiments on the characteristics of different stages of FPM

表4 关于MAC和PAP的消融实验Table 4 Ablation experiments on MAC and PAP

表3为使用不同阶段的特征评估特征金字塔模块的性能。不同阶段的特征有着不同的语义特征，F1、F2、F3分别为从浅到深3个阶段的特征。可以发现，当缺少其中任何一个阶段的特征时，性能都会下降，同时深层次的特征有着更多的语义特征，对人群计数也更为至关重要。聚合其他阶段的特征可以提供在深层阶段丢失的语义信息，从而提高网络对于不同密度场景的适应性和计数精度。结果表明，特征金字塔模块有助于融合图像深层和浅层语义特征信息，进而提高了人群计数的准确性。

表4为分别使用不同模块评估多尺度感受野的回归头中多尺度空洞卷积模块和金字塔平均池化模块的必要性。实验结果表明，在只有一种模块的情况下，模型精度会有所下降，进一步验证了多尺度空洞卷积模块和金字塔平均池化模块的必要性。同时，本文也对深度监督训练方式的辅助损失进行了验证。实验结果表明，每条支路的损失都对衡量模型起到正面作用，同时采用三分支不同权重的损失可提高网络性能，使其更好地收敛。结果如表5所示。

表5 不同权重损失的消融实验Table 5 Ablation experiments with different weight loss

4 结束语

本文提出一种简单但高性能的基于多尺度金字塔Transformer人群计数方法（MSPT-Net），首先MSPT-Net能够同时捕获全局上下文信息和局部语义信息。其次为了融合深层和浅层语义特征信息，本文在网络中嵌入了特征金字塔模块，并通过多尺度的空洞卷积模块和金字塔平均池化模块增强特征来预测密度图。在网络的每个中间阶段都引入了额外的监督损失，以稳定训练过程。MSPT-Net在多个公开数据集上表现出优异性能，其可以应用于人群计数领域，如人流量监控和城市规划等。未来，将继续改进我们的方法，以实现更轻量化的模型，使其可以迁移至移动设备中。同时使用无监督的学习策略，解决人群计数任务中人头位置标注繁琐等数据问题。