【技术实现步骤摘要】
一种用于跨境行人重识别的行人检测模型及其应用方法
[0001]本申请涉及行人重识别领域,特别是涉及一种用于跨境行人重识别的行人检测模型及其应用方法。
技术介绍
[0002]跨镜追踪(Person Re
‑
Identification,简称ReID)技术是现在计算机视觉研究的热门方向,主要解决跨摄像头跨场景下行人的识别与检索。相较于传统的视频监控,ReID有一定的优势但是也存在一定技术挑战;ReID在实际应用场景下的数据非常复杂,由于不同摄像设备之间的差异,同时行人兼具刚性和柔性的特性,外观易受穿着、尺度、遮挡、姿态和视角等影响,所以对跨镜追踪(ReID)算法的要求也更高,实际应用中常常出现由于无正面照、服装更换、遮挡、图像分辨率低、光线差异以及室内室外场景变化而无法识别同一个行人的情况,这些都是行人重识别的干扰因素,这些干扰因素不仅影响模型识别准确度,也在极大程度上影响了模型的识别效率。
[0003]虽然诸多研究人员已经在行人重识别领域下取得一定的进展,但是依旧难以很好地满足实际应用需求,行人ReID作为一个实例级识别问题,在实际设计和使用中仍面临许多问题,主要体现在以下几个方面:
[0004]1.为了在复杂环境中获取很好的行人识别结果,需要在模型中密集连接线性增加通道数,而这样设计的大数量的通道数导致了较重的内存访问成本,使得模型的运算成本大且效率低下;
[0005]2.目前的模型在光照不佳(如暗光照射)的条件下的行人检测精度不高,然而实际的行人重识别场景下经常会出现环境光 ...
【技术保护点】
【技术特征摘要】
1.一种用于跨境行人重识别的行人检测模型的构建方法,其特征在于,包括以下步骤:获取标记有行人目标框的图像集作为训练样本;将训练样本输入到行人检测模型中进行训练直到满足训练条件,所述行人检测模型包括第一采样分支、第二采样分支以及三个检测网络组成,其中所述第一采样分支内包括交替出现的训练结构重参化卷积模块和多尺度表征模块和空间金字塔池化层组成,最后一个多尺度表征模块的输出输入到空间金字塔池化层中得到第一采样分支的第一采样特征;所述第二分支包括依次连接的卷积层、上采样层、第一结构重参化聚合卷积模块、特征堆叠层、第二结构重参化聚合卷积模块、上下文细节处理模块、第二特征堆叠层以及第三结构重参化聚合卷积模块,第一采样特征输入到第二分支的卷积层且经过卷积层处理的特征输入到第二特征堆叠层中进行特征堆叠,第一分支中位于中间位置的多尺度表征模块的输出特征输入到第一特征堆叠层中进行特征堆叠;三个检测网络分别为同第一采样分支的空间金字塔池化层连接的第一检测网络,同第一结构重参化聚合卷积模块连接的第二检测网络,以及同第三结构重参化聚合卷积模块连接的第三检测网络,其中每一检测网络包括依次连接的上下文细节处理模块以及检测头部。2.根据权利要求1所述的用于跨境行人重识别的行人检测模型的构建方法,其特征在于,标记有行人目标框的图像输入到第一采样分支中输出第一采样特征,第一采样特征输入到第一个检测网络中输出第一检测结果,同时第一采样特征输入到第二采样分支中进行采样处理;第一采样特征经过卷积层、上采样层以及第一结构重参化聚合卷积模块后得到得到第一聚合特征,第一聚合特征输入到第二检测网络中输出第二检测结果,且第一聚合特征和第一采样分支中的第二多尺度表征模块的输出特征在第一特征堆叠层中进行特征堆叠得到第一堆叠特征,第一堆叠特征继续进入第二结构重参化聚合卷积模块和上下文细节处理模块后输入第二特征堆叠层,且第一采样特征经过卷积层处理后的特征也一并输入到第二特征堆叠层中,第二特征堆叠层输出第二堆叠特征,第二堆叠特征经过第三结构重参化聚合卷积模块后输入到第三检测网络中得到第三检测结果。3.根据权利要求1所述的用于跨境行人重识别的行人检测模型的构建方法,其特征在于,每一训练结构重参化卷积模块包括两个特征分离通道,其中第一特征分离通道为恒等映射分支,第二特征分离通道包括并行的恒等映射分支、3*3卷积分支以及1*1卷积分支以及连接三个分支的SILU层,两个特征分离通道的输出特征经过特征堆叠后进行特征通道混洗输出当前训练结构重参化卷积模块的输出特征。4.根据权利要求3所述的用于跨境行人重识别的行人检测模型的构建方法,其特征在于,当行人检测模块完成训练后投入推理使用时,行人检测模块中的训练结构重参化卷积模块转换为推理结构重参化卷积模块,推理结构重参化卷积模块在训练结构重参化卷积模块上使用结构重参化,将训练结构重参化卷积模块的第二特征分离通道中并行的恒等映射分支、3*3卷积分支以及1*1卷积分支转换为唯一的3*3卷积分支。5.根据权利要求1所述的用于跨境行人重识别的行人检测模型的构建方法,其特征在于,多尺度表征模块包括1*1逐点卷积、同1*1逐点卷积连接的多层提取层、同多层提取层连接的特征堆叠层以及同特征堆叠层连接的特征混洗层,除了第一层的提取层外的提取层均包括依次连接的1*1卷积、k*k卷积和1*1卷积,其中输入多尺度表征模块内的特征经过1*1逐点卷积后输出不同通道数量的通道特征,不同通道特征分别输入到对应的提取层中,且
上一层提取层的输出特征同对应当前提取层的通道特征拼接后输入到当前提取层中进行提取,不同提取层的输出特征进入特征堆叠层中进行特征堆叠,特征堆叠...
【专利技术属性】
技术研发人员:张香伟,夏如意,毛云青,王德平,
申请(专利权)人:城云科技中国有限公司,
类型:发明
国别省市:
还没有人留言评论。发表了对其他浏览者有用的留言会获得科技券。