一种基于多任务深度学习的行人属性识别方法技术

本发明专利技术提供一种基于多任务深度学习的行人属性识别方法，利用行人检测算法在监控视频检测出多姿态的行人样本图；对所检测出的行人样本图进行重命名，并进行行人属性手工标注，划分训练集样本和测试集样本；对训练集样本进行训练，得到多任务行人属性识别模型，针对任务的需求设计多任务行人属性识别网络；利用多任务行人属性识别模型对测试集样本每个图片输出其属于所有属性中所有类别的置信度。设计了一个多任务的深度网络，在一个网络中学习所有的行人属性，该方法不仅解决了资源的占用问题，同时由于在同一个网络中进行特征共享，大大提升了模型的识别精度和速度。

【技术实现步骤摘要】
一种基于多任务深度学习的行人属性识别方法
本专利技术属于计算机视觉与模式识别领域，更具体地涉及一种基于多任务深度学习的行人属性识别方法。
技术介绍
近年来，监控摄像头在城市中得到大量普及，对人们生活起到了保驾护航的作用。如今全球每天监控录像产生数据量就达到上千PB，随着高清4K技术的推广和普及，该数据量还将会成倍增长，视频监控也将步入更高清的大数据井喷时代。如何从海量的数据中进行有用的信息检索？是现代安防重点解决的问题之一。在以往数据规模较小，通常人为从监控视频中找到符合特征的行人。如今随着监控数据的海量增长，人力资源成本的提高，仅依靠人力筛选已变得不切实际；另一方面，刑事案件往往对检索具有时效性要求。所以如何准确对视频中行人进行自动化描述和识别，对实现行人的快速检索具有十分重要的意义。行人属性的识别是实现视频中行人检索的前提，目前学术界就行人属性识别也做了相关研究，大致可以分为基于传统特征算法的行人属性识别、基于深度学习的行人属性识别两种，具体方案如下：1、基于传统特征算法的行人属性识别，这种方法需要进行特征模板的选取设计，如上衣颜色，利用上衣区域设计颜色特征匹配模板，并进行颜色分类，再在相同颜色中来进行特征匹配。该方法容易受到光照和环境的影响。虽然在资源占用方面具有一定优势，但是其受到环境因素的影响比较大。2、基于单一任务深度学习的行人属性识别方法，该方法也需要进行特征选取，如性别、年龄、发型等，就其中一种特征训练一个深度网络模型，再利用网络模型的深度特征向量之间的余弦距离进行特征匹配。由于行人间存在较多相似性，而性别、年龄、发型都无法对是否是同一个人进行唯一认证，所以通常需要多组特征组合，而多个特征模型则会消耗更多的时间和硬件显存资源，从而大大增加了产品的成本。精度上有一定的优势，但是在多组属性加载模型对速度和显存都会带来很大负担，不利于产品化中使用。
技术实现思路
针对现有技术的缺点，如精度差、速度慢和显存资源占用大等弊端，本专利技术的目的在于提出一种基于多任务深度学习的行人属性识别方法，本专利技术欲解决的技术问题是：一方面利用深度学习来学习更加鲁棒的特征；另一方面，考虑到行人属性间的特征存在着相关性，设计了一个多任务的深度网络，在一个网络中学习所有的行人属性，该方法不仅解决了资源的占用问题，同时由于在同一个网络中进行特征共享，大大提升了模型的识别精度和速度。为解决上述技术问题：本专利技术提供了一种基于多任务深度学习的行人属性识别方法，其特征在于：包括如下步骤：S1：获取多个点位、场景的监控视频，利用行人检测算法在监控视频检测出多姿态的行人样本图；S2：对所检测出的所述行人样本图进行重命名，并进行行人属性手工标注，划分训练集样本和测试集样本；S3：针对任务的需求设计多任务行人属性识别网络；S4:将所述步骤S2收集的所述训练集样本放入所述步骤S3所设计的所述多任务行人属性识别网络进行训练，得到多任务行人属性识别模型；S5:对于所述测试集样本，利用所述步骤S4得到的所述多任务行人属性识别模型对其中每个图片输出其属于所有属性中所有类别的置信度。进一步地，在所述步骤S5中，还包括按照类别置信度大小排序，得到所述测试集样本图片在所有属性中所属的类别。进一步地，在所述步骤S2中，还包括对所述行人样本图中相同姿态下的行人样本进行筛选和过滤，对剩余行人样本进行人工标注，给出该剩余行人样本每个图片所有类别的标签。进一步地，所述类别至少包括性别、年龄、发型、是否戴口罩、是否戴帽子、是否背包、是否拎东西、上衣类型、下衣类型。进一步地，对标注好的样本进行划分，随机选取其中20％的样本作为所述测试集样本，剩余的80％为所述训练集样本。进一步地，所述多任务行人属性识别网络为多层深度神经网络结构。进一步地，所述多层深度神经网络结构的输入层对应着多个行人属性输入数据向量，输出层每个节点对应着每个行人属性所述类别的预测。进一步地，在预测过程中，学习属性间的相关性、属性的局部性及属性的全局性。进一步地，所述属性间的相关性包括下衣属性，所述属性的局部性包括头部区域属性，所述属性的全局性包括全身区域属性。进一步地，所述下衣属性为裙子，所述头部区域属性包括头发颜色、是否戴口罩，所述全身区域属性包括是否拎东西、是否背包。与现有技术相比，本专利技术至少具有下述的有益效果或优点：本专利技术提供的这种基于多任务深度学习的行人属性识别方法，一方面利用深度学习来学习更加鲁棒的特征；另一方面，考虑到行人属性间的特征存在着相关性，如性别和下衣类型，这两种属性间下衣属性为裙子的样本性别也基本为女性，设计了一个多任务的深度网络，在一个网络中学习所有的行人属性，该方法不仅解决了资源的占用问题，同时由于在同一个网络中进行特征共享，大大提升了模型的识别精度和速度。附图说明以下将结合附图对本专利技术做进一步详细说明；图1为本专利技术多任务深度学习的行人属性识别方法流程图。图2为本专利技术实施例多任务网络结构示例图。具体实施方式下面将结合本专利技术实施例中的附图，对本专利技术实施例中的技术方案进行清楚、完整地描述，显然，所描述的实施例是本专利技术一部分实施例，而不是全部的实施例。基于本专利技术中的实施例，本领域普通技术人员在没有做出创造性劳动前提下所获得的所有其他实施例，都属于本专利技术保护的范围。本专利技术目的在于提出一种基于多任务深度学习的行人属性识别方法，本专利技术欲解决的技术问题是：一方面利用深度学习来学习更加鲁棒的特征；另一方面，考虑到行人属性间的特征存在着相关性，设计了一个多任务的深度网络，在一个网络中学习所有的行人属性，该方法不仅解决了资源的占用问题，同时由于在同一个网络中进行特征共享，大大提升了模型的识别精度和速度。为解决上述技术问题：如图1所示的基于多任务深度学习的行人属性识别方法流程图，包括如下步骤：S1：获取多个点位、场景的监控视频，利用行人检测算法在监控视频检测出多姿态的行人样本图；尽可能保证样本的多样性；S2：对所检测出的所述行人样本图进行重命名，并进行行人属性手工标注，划分训练集样本和测试集样本；S3：针对任务的需求设计多任务行人属性识别网络；S4:将所述步骤S2收集的所述训练集样本放入所述步骤S3所设计的所述多任务行人属性识别网络进行训练，得到多任务行人属性识别模型；S5:对于所述测试集样本，利用所述步骤S4得到的所述多任务行人属性识别模型对其中每个图片输出其属于所有属性中所有类别的置信度。该方法还包括构建行人属性样本的过程和多任务行人属性识别网络的设计。具体如下：在构建行人属性样本的过程中，考虑到行人检测算法在监控视频中获取行人图片存在一定的冗余，首先对相同姿态下的行人样本进行筛选和过滤；接着，对剩下的行人样本进行人工标注，给出每个图片所有类别中的标签，标注的类别有如下9种，分布为：性别(男/女)、年龄(少年/青年/中年/老年)、发型(长发/短发)、戴口罩(是/否)、戴帽子(是/否)、背包(是/否)、拎东西(是/否)、上衣类型(长袖/短袖)、下衣类型(裙子/长裤/短裤)；最后，对标注好的样本进行划分，随机选取其中20％的样本作为测试集，剩下80％为训练集。在多任务行人属性识别网络的设计过程中，在多任务网络设计上，其所述的多任务网络结构为多层深度神经网络结构，输入层对应着多个属性输入数据本文档来自技高网...

【技术保护点】
1.一种基于多任务深度学习的行人属性识别方法，其特征在于：包括如下步骤：S1：获取多个点位、场景的监控视频，利用行人检测算法在监控视频检测出多姿态的行人样本图；S2：对S1检测出的所述行人样本图进行重命名，并进行行人属性手工标注，划分训练集样本和测试集样本；S3：针对任务的需求设计多任务行人属性识别网络；S4:将所述步骤S2收集的所述训练集样本放入所述步骤S3所设计的所述多任务行人属性识别网络进行训练，得到多任务行人属性识别模型；S5:对于所述测试集样本，利用所述步骤S4得到的所述多任务行人属性识别模型对其中每个图片输出其属于所有属性中所有类别的置信度。

【技术特征摘要】
1.一种基于多任务深度学习的行人属性识别方法，其特征在于：包括如下步骤：S1：获取多个点位、场景的监控视频，利用行人检测算法在监控视频检测出多姿态的行人样本图；S2：对S1检测出的所述行人样本图进行重命名，并进行行人属性手工标注，划分训练集样本和测试集样本；S3：针对任务的需求设计多任务行人属性识别网络；S4:将所述步骤S2收集的所述训练集样本放入所述步骤S3所设计的所述多任务行人属性识别网络进行训练，得到多任务行人属性识别模型；S5:对于所述测试集样本，利用所述步骤S4得到的所述多任务行人属性识别模型对其中每个图片输出其属于所有属性中所有类别的置信度。2.根据权利要求1所述的方法，其特征在于：在所述步骤S5中，还包括按照类别置信度大小排序，得到所述测试集样本图片在所有属性中所属的类别。3.根据权利要求1所述的方法，其特征在于：在所述步骤S2中，还包括对所述行人样本图中相同姿态下的行人样本进行筛选和过滤，对剩余行人样本进行人工标注，给出该剩余行人样本每个图片所有类别的标签。4.根据权利要求...

【专利技术属性】
技术研发人员：毛亮，林焕凯，朱婷婷，黄仝宇，汪刚，宋一兵，侯玉清，刘双广，
申请(专利权)人：高新兴科技集团股份有限公司，
类型：发明
国别省市：广东,44