【技术实现步骤摘要】
本专利技术涉及图像识别，尤其涉及一种铁路客运站旅客属性识别方法及装置。
技术介绍
1、随着铁路线网的不断完善，列车出行已逐渐成为人们的首选方式。客运站作为铁路与旅客接口的重要地点，对站内旅客进行全方位实时监控是保障旅客候车安全、提高旅客出行体验的重要手段。
2、然而，站内面积庞大、结构复杂、候车旅客多，极大增加了工作人员服务难度。通过图像识别对旅客进行监控是一种可行的方式，但现有图像识别方法难以对旅客的多个属性进行全面精准识别，从而无法实现对旅客的有效监控，进而无法保证旅客的候车安全。
技术实现思路
1、针对现有技术存在的问题，本专利技术实施例提供一种铁路客运站旅客属性识别方法及装置。
2、本专利技术提供一种铁路客运站旅客属性识别方法，包括：
3、获取铁路客运站中的旅客图像；
4、将所述旅客图像输入至预先训练好的旅客属性识别模型，得到所述旅客图像中旅客的多个预设属性的属性值；
5、其中，所述旅客属性识别模型包括依次连接的特征识别模块和属性空间记忆模块，所述属性空间记忆模块连接有每一个所述预设属性对应的第一分类器，所述属性空间记忆模块中存储有每一个所述预设属性对应的空间注意力图，所述空间注意力图是在所述旅客属性识别模型的训练过程中生成并更新得到的；
6、所述特征识别模块用于生成所述旅客图像的第一特征图；
7、所述属性空间记忆模块用于对所述旅客图像的第一特征图以及各所述预设属性的空间注意力图进行融合，得
8、所述第一分类器用于基于所述目标特征图确定所述旅客图像中各所述预设属性的属性值。
9、根据本专利技术提供的铁路客运站旅客属性识别方法，所述空间注意力图是通过如下方法得到的：
10、在所述旅客属性识别模型的当前次迭代训练中，基于所述预设属性对应的第一分类器的当前权重以及各样本图像的第一特征图，生成所述预设属性对应的多个第一空间注意力图，所述第一空间注意力图与所述样本图像一一对应；
11、基于各所述样本图像对应的标签以及各所述第一空间注意力图的特征值，对各所述第一空间注意力图进行筛选以及融合处理，得到所述预设属性对应的第二空间注意力图；
12、基于所述预设属性对应的第二空间注意力图和当前空间注意力图，生成所述预设属性在所述当前次迭代训练中的目标空间注意力图。
13、根据本专利技术提供的铁路客运站旅客属性识别方法，所述基于所述预设属性对应的第一分类器的当前权重以及各样本图像的第一特征图，生成所述预设属性对应的多个第一空间注意力图，包括：
14、获取所述预设属性对应的第一分类器的当前权重与各所述样本图像中目标样本图像的第一特征图的乘积；
15、基于所述预设属性对应的第一分类器的当前权重与所述目标样本图像的第一特征图的乘积，生成所述预设属性对应的第一空间注意力图。
16、根据本专利技术提供的铁路客运站旅客属性识别方法，所述基于各所述样本图像对应的标签以及各所述第一空间注意力图的特征值，对各所述第一空间注意力图进行筛选以及融合处理，得到所述预设属性对应的第二空间注意力图，包括：
17、基于各所述样本图像对应的标签以及各所述第一空间注意力图的特征值，对各所述第一空间注意力图进行筛选，得到多个所述第一空间注意力图中的候选空间注意力图；
18、对各所述候选空间注意力图进行池化处理，得到所述预设属性对应的第二空间注意力图。
19、根据本专利技术提供的铁路客运站旅客属性识别方法，所述基于所述预设属性对应的第二空间注意力图和当前空间注意力图，生成所述预设属性在所述当前次迭代训练中的目标空间注意力图，包括：
20、对所述预设属性对应的第二空间注意力图以及所述预设属性对应的当前空间注意力图进行加权池化处理；
21、基于所述加权池化处理的结果，生成所述预设属性在所述当前次迭代训练中的目标空间注意力图。
22、根据本专利技术提供的铁路客运站旅客属性识别方法，所述特征识别模块包括依次连接的第一子识别模块和第二子识别模块，所述第二子识别模块与所述属性空间记忆模块连接；所述旅客属性识别模型还包括依次连接的掩码对比学习模块和第三子识别模块，所述第三子识别模块连接有每一个所述预设属性对应的第二分类器，所述掩码对比学习模块与所述第一子识别模块连接；所述掩码对比学习模块用于基于所述第一子识别模块输出的第二特征图生成随机掩码特征图；
23、所述旅客属性识别模型是通过如下方法训练得到的：
24、基于所述第二分类器对样本图像的预测结果，以及所述第一分类器对所述样本图像的预测结果，对所述旅客属性识别模型进行迭代训练；
25、其中，对所述旅客属性识别模型进行迭代训练的过程中，所述第二子识别模块与所述第三子识别模块之间参数共享，以及，所述第一分类器与所述第二分类器之间参数共享。
26、根据本专利技术提供的铁路客运站旅客属性识别方法，所述基于所述第二分类器对样本图像的预测结果以及所述第一分类器对所述样本图像的预测结果，对所述旅客属性识别模型进行迭代训练，包括：
27、基于所述第二分类器对所述样本图像的预测结果以及所述第一分类器对所述样本图像的预测结果，确定所述旅客属性识别模型的对比损失，基于所述对比损失，对所述旅客属性识别模型进行迭代训练。
28、根据本专利技术提供的铁路客运站旅客属性识别方法，所述得到所述旅客图像中旅客的多个预设属性的属性值之后，还包括：
29、基于所述旅客图像中旅客的多个所述预设属性的属性值，确定目标旅客；
30、基于所述目标旅客的各所述预设属性中目标属性的属性值，生成预警信息，并发送至所述目标旅客对应的区域的预设终端；其中，所述目标旅客对应的区域为所述目标旅客对应的所述旅客图像的采集区域。
31、本专利技术还提供一种铁路客运站旅客属性识别装置，包括：
32、数据获取模块，用于获取铁路客运站中的旅客图像；
33、数据处理模块，用于将所述旅客图像输入至预先训练好的旅客属性识别模型，得到所述旅客图像中旅客的多个预设属性的属性值；
34、其中，所述旅客属性识别模型包括依次连接的特征识别模块和属性空间记忆模块，所述属性空间记忆模块连接有每一个所述预设属性对应的第一分类器，所述属性空间记忆模块中存储有每一个所述预设属性对应的空间注意力图，所述空间注意力图是在所述旅客属性识别模型的训练过程中生成并更新得到的；
35、所述特征识别模块用于生成所述旅客图像的第一特征图；
36、所述属性空间记忆模块用于对所述旅客图像的第一特征图以及各所述预设属性的空间注意力图进行融合，得到所述旅客图像的目标特征图；
37、所述第一分类器用于基于所述目标特征图确定所述旅客图像中各所述预设属性的属性值。
38、本专利技术还提供一种电子设备，包括存储器、处理器及本文档来自技高网...
【技术保护点】
1.一种铁路客运站旅客属性识别方法，其特征在于，包括：
2.根据权利要求1所述的铁路客运站旅客属性识别方法，其特征在于，所述空间注意力图是通过如下方法得到的：
3.根据权利要求2所述的铁路客运站旅客属性识别方法，其特征在于，所述基于所述预设属性对应的第一分类器的当前权重以及各样本图像的第一特征图，生成所述预设属性对应的多个第一空间注意力图，包括：
4.根据权利要求2所述的铁路客运站旅客属性识别方法，其特征在于，所述基于各所述样本图像对应的标签以及各所述第一空间注意力图的特征值，对各所述第一空间注意力图进行筛选以及融合处理，得到所述预设属性对应的第二空间注意力图，包括：
5.根据权利要求2所述的铁路客运站旅客属性识别方法，其特征在于，所述基于所述预设属性对应的第二空间注意力图和当前空间注意力图，生成所述预设属性在所述当前次迭代训练中的目标空间注意力图，包括：
6.根据权利要求1所述的铁路客运站旅客属性识别方法，其特征在于，所述特征识别模块包括依次连接的第一子识别模块和第二子识别模块，所述第二子识别模块与所述属性空间记忆模块
7.根据权利要求6所述的铁路客运站旅客属性识别方法，其特征在于，所述基于所述第二分类器对样本图像的预测结果以及所述第一分类器对所述样本图像的预测结果，对所述旅客属性识别模型进行迭代训练，包括：
8.根据权利要求1至7任一项所述的铁路客运站旅客属性识别方法，其特征在于，所述得到所述旅客图像中旅客的多个预设属性的属性值之后，还包括：
9.一种铁路客运站旅客属性识别装置，其特征在于，包括：
10.一种电子设备，包括存储器、处理器及存储在所述存储器上并可在所述处理器上运行的计算机程序，其特征在于，所述处理器执行所述程序时实现如权利要求1至8任一项所述的铁路客运站旅客属性识别方法。
...【技术特征摘要】
1.一种铁路客运站旅客属性识别方法，其特征在于，包括：
2.根据权利要求1所述的铁路客运站旅客属性识别方法，其特征在于，所述空间注意力图是通过如下方法得到的：
3.根据权利要求2所述的铁路客运站旅客属性识别方法，其特征在于，所述基于所述预设属性对应的第一分类器的当前权重以及各样本图像的第一特征图，生成所述预设属性对应的多个第一空间注意力图，包括：
4.根据权利要求2所述的铁路客运站旅客属性识别方法，其特征在于，所述基于各所述样本图像对应的标签以及各所述第一空间注意力图的特征值，对各所述第一空间注意力图进行筛选以及融合处理，得到所述预设属性对应的第二空间注意力图，包括：
5.根据权利要求2所述的铁路客运站旅客属性识别方法，其特征在于，所述基于所述预设属性对应的第二空间注意力图和当前空间注意力图，生成所述预设属性在所述当前次迭代训练中的目标空间注意力图，包括：
6.根据权利要求1所述的铁路客运站旅客属性识别方法，其特征在于，所述特征识别模块包括依次连接的第一子识别模块和第二子...
【专利技术属性】
技术研发人员：李超，宁斐，刘硕研，刘玉鑫，李樊，杨恩泽，刘祎然，梁博，刘卫平，李志跃，张旭易，张煜山，丛红萍，
申请(专利权)人：中国国家铁路集团有限公司，
类型：发明
国别省市：
还没有人留言评论。发表了对其他浏览者有用的留言会获得科技券。