[发明专利]一种目标计数方法及装置

权利要求书

1.一种目标计数方法，其特征在于，所述方法包括：

将图像输入计数模型的编码器，得到若干尺寸不同的编码特征，并根据各编码特征确定所述图像对应的融合编码特征；

将所述融合编码特征分别输入所述计数模型的解码器的各卷积模块，针对每个卷积模块，通过该卷积模块，对所述融合编码特征进行卷积，得到该卷积模块对应的解码特征；其中，各卷积模块中卷积核对应的感受野不同；

根据各卷积模块对应的解码特征，确定融合解码特征，根据所述融合解码特征，确定所述图像的目标计数结果。

2.如权利要求1所述的方法，其特征在于，所述尺寸包括：分辨率以及通道数；

将图像输入计数模型的编码器，得到若干尺寸不同的编码特征，并根据各编码特征确定所述图像对应的融合编码特征，具体包括：

将图像输入计数模型的编码器，得到若干尺寸不同的编码特征，并将分辨率最大的编码特征作为基准特征；

针对所述基准特征外的每个编码特征，对该编码特征进行上采样，得到与所述基准特征分辨率相同的特征，作为该编码特征对应的分辨率还原特征；

对所述分辨率还原特征进行常规卷积，得到与所述基准特征通道数相同的特征，作为该编码特征对应的与所述基准特征尺寸相同的还原特征；

将各编码特征对应的还原特征与所述基准特征进行叠加，得到所述图像对应的融合编码特征。

3.如权利要求1所述的方法，其特征在于，所述解码器的卷积模块包括混合卷积模块以及单一卷积模块；

将所述融合编码特征分别输入所述计数模型的解码器的各卷积模块，针对每个卷积模块，通过该卷积模块，对所述融合编码特征进行卷积，得到该卷积模块对应的解码特征，具体包括：

将所述融合编码特征分别输入所述计数模型的解码器的各混合卷积模块，并将所述融合编码特征输入所述解码器的单一卷积模块；

针对每个混合卷积模块，通过该混合卷积模块，对所述融合编码特征进行常规卷积，得到中间特征，对所述中间特征进行膨胀卷积，得到该混合卷积模块对应的解码特征；其中，对所述融合编码特征进行常规卷积的卷积核的大小与对所述中间特征进行膨胀卷积的膨胀率正相关；

通过所述单一卷积模块，对所述融合编码特征进行常规卷积，得到所述单一卷积模块对应的解码特征。

4.如权利要求3所述的方法，其特征在于，根据各卷积模块对应的解码特征，确定融合解码特征，具体包括：

对各混合卷积模块对应的解码特征进行拼接，得到聚合特征；

将所述聚合特征与所述单一卷积模块对应的解码特征进行叠加，得到融合解码特征。

5.如权利要求1所述的方法，其特征在于，根据所述融合解码特征，确定所述图像的目标计数结果，具体包括：

对所述融合解码特征进行常规卷积，得到所述图像对应的密度图；

根据所述密度图，确定所述图像中目标的数量，作为所述图像的目标计数结果。

6.如权利要求1所述的方法，其特征在于，所述计数模型通过以下方法训练得到：

将所述图像作为训练样本，确定所述图像中包含的目标数量，作为所述训练样本的标签；

将所述训练样本输入待训练的所述计数模型，得到所述训练样本的目标计数结果；

根据所述训练样本的目标计数结果与标签间的差异，对所述计数模型进行训练。

7.如权利要求5所述的方法，其特征在于，所述计数模型通过以下方法训练得到：

将所述图像作为训练样本，确定所述图像的密度图，作为所述训练样本的标签；

将所述训练样本输入待训练的所述计数模型，输出所述训练样本的密度图；

根据所述训练样本的标签与所述计数模型输出的密度图间的差异，对所述计数模型进行训练。