模型训练方法、装置、人脸筛选方法及电子设备.pdf

上传人:刘** 文档编号:10304542 上传时间:2021-06-17 格式:PDF 页数:14 大小:553.57KB
收藏 版权申诉 举报 下载
模型训练方法、装置、人脸筛选方法及电子设备.pdf_第1页
第1页 / 共14页
模型训练方法、装置、人脸筛选方法及电子设备.pdf_第2页
第2页 / 共14页
模型训练方法、装置、人脸筛选方法及电子设备.pdf_第3页
第3页 / 共14页
文档描述:

《模型训练方法、装置、人脸筛选方法及电子设备.pdf》由会员分享,可在线阅读,更多相关《模型训练方法、装置、人脸筛选方法及电子设备.pdf(14页完成版)》请在专利查询网上搜索。

1、(19)中华人民共和国国家知识产权局 (12)发明专利申请 (10)申请公布号 (43)申请公布日 (21)申请号 202010623482.1 (22)申请日 2020.06.30 (71)申请人 重庆紫光华山智安科技有限公司 地址 400700 重庆市北碚区云汉大道117号 附386号 (72)发明人 刘畅 (74)专利代理机构 北京超凡宏宇专利代理事务 所(特殊普通合伙) 11463 代理人 张磊 (51)Int.Cl. G06K 9/00(2006.01) G06K 9/46(2006.01) G06K 9/62(2006.01) G06N 3/04(2006.01) G06N 3/0。

2、8(2006.01) (54)发明名称 模型训练方法、 装置、 人脸筛选方法及电子 设备 (57)摘要 本申请供一种模型训练方法、 装置、 人脸筛 选方法及电子设备。 通过该模型训练方法, 由于 在训练时所使用了人脸关键点图像以及人脸置 信度图像, 因此, 图像中所携带的人脸关键点以 及人脸置信度可以用于 “指导” 神经网络模型进 行特征提取。 同时, 保持基于人脸关键点图像所 获得的待优化的神经网络结构中部分网络层的 参数不变, 通过人脸置信度图像对该待优化的神 经网络模型进行训练, 调整其中其他网络层的参 数, 使得训练好的神经网络能够兼顾基于人脸关 键点图像训练出的特征提取方式, 以及基。

3、于人脸 置信度图像训练处的特征提取方式, 进而提高对 人脸图像的分辨精度。 权利要求书2页 说明书9页 附图2页 CN 111753793 A 2020.10.09 CN 111753793 A 1.一种模型训练方法, 其特征在于, 应用于电子设备, 所述电子设备配置有未训练好的 神经网络模型, 所述方法包括: 获取样本图像, 所述样本图像包括人脸关键点图像以及人脸置信度图像; 通过所述人脸关键点图像训练所述神经网络模型进行人脸图像筛选, 获得待优化的神 经网络模型; 保持所述待优化的神经网络模型中第一网络层的参数, 通过所述人脸置信度图像再次 训练所述待优化的神经网络模型进行人脸图像筛选, 。

4、获得训练好的神经网络模型。 2.根据权利要求1所述的模型训练方法, 其特征在于, 所述电子设备还配置有教师网络 模型, 所述教师网络模型包括所述待优化的神经网络模型中的第二网络层的网络结构, 其 中, 所述第二网络层属于所述第一网络层的一部分; 所述通过所述人脸置信度图像再次训练所述待优化的神经网络模型进行人脸图像筛 选, 获得最终训练好的神经网络模型的之前, 还包括: 保持所述第二网络层的参数; 通过所述人脸置信度图像训练所述教师网络模型进行人脸图像筛选, 获得训练好的教 师网络模型; 所述通过所述人脸置信度图像再次训练所述待优化的神经网络模型进行人脸图像筛 选, 获得最终训练好的神经网络模。

5、型的步骤, 包括: 结合所述训练好的教师网络模型, 通过所述人脸置信度图像再次训练所述待优化的神 经网络模型进行人脸图像筛选, 获得所述训练好的神经网络模型。 3.根据权利要求2所述的模型训练方法, 其特征在于, 所述教师网络模型中的网络层数 大于所述待优化的神经网络模型中的网络层数。 4.根据权利要求2所述的模型训练方法, 其特征在于, 所述结合所述训练好的教师网络 模型, 通过所述人脸置信度图像再次训练所述待优化的神经网络模型进行人脸图像筛选, 获得所述训练好的神经网络模型的步骤, 包括: 基于所述神经网络模型的第一损失函数, 获得通过所述人脸置信度图像训练所述待优 化的神经网络模型时的第。

6、一损失值; 基于所述训练好的教师网络模型的第二损失函数, 获得通过所述训练好的教师网络模 型筛选所述人脸置信度图像时的第二损失值; 根据所述第一损失值的第一权值以及所述第二损失值的第二权值, 获得所述第一损失 值以及所述第二损失值的加权求和结果, 其中, 所述第一权值大于所述第二权值; 判断所述加权求和结果是否小于损失阈值; 若所述加权求和结果小于所述损失阈值, 则获得所述训练好的神经网络模型; 若所述加权求和结果不小于损失阈值, 则根据所述加权求和结果调整所述待优化的神 经网络模型的权值, 再次从所述基于所述神经网络模型的第一损失函数, 获得通过所述人 脸置信度图像训练所述待优化的神经网络模。

7、型时的第一损失值的步骤执行。 5.根据权利要求2所述的模型训练方法, 其特征在于, 所述教师网络模型包括残差网 络。 6.一种人脸筛选方法, 其特征在于, 应用于电子设备, 所述电子设备配置有经权利要求 1-5任意一项所述的模型训练方法训练好的神经网络模型, 所述方法包括: 权利要求书 1/2 页 2 CN 111753793 A 2 获取待筛选人脸图像; 通过所述训练好的神经网络模型对所述待筛选人脸图像进行人脸图像筛选, 获得筛选 结果。 7.一种模型训练装置, 其特征在于, 应用于电子设备, 所述电子设备配置有未训练好的 神经网络模型, 所述模型训练装置还包括: 样本获取模块, 用于获取样。

8、本图像, 所述样本图像包括人脸关键点图像以及人脸置信 度图像; 第一训练模块, 用于通过所述人脸关键点图像训练所述神经网络模型进行人脸图像筛 选, 获得待优化的神经网络模型; 第二训练模块, 用于保持所述待优化的神经网络模型中第一网络层的参数, 通过所述 人脸置信度图像再次训练所述待优化的神经网络模型进行人脸图像筛选, 获得训练好的神 经网络模型。 8.根据权利要求7所述的模型训练装置, 其特征在于, 所述电子设备还配置有教师网络 模型, 所述教师网络模型包括所述待优化的神经网络模型中的第二网络层的网络结构, 其 中, 所述第二网络层属于所述第一网络层的一部分, 所述第二训练模块的训练步骤, 。

9、包括: 保持所述待优化的神经网络模型中第一网络层的参数; 通过所述人脸置信度图像训练所述教师网络模型进行人脸图像筛选, 获得训练好的教 师网络模型; 所述通过所述人脸置信度图像再次训练所述待优化的神经网络模型进行人脸图像筛 选, 获得最终训练好的神经网络模型的步骤, 包括: 结合所述训练好的教师网络模型, 通过所述人脸置信度图像再次训练所述待优化的神 经网络模型进行人脸图像筛选, 获得所述训练好的神经网络模型。 9.一种电子设备, 其特征在于, 所述电子设备包括处理器以及存储器, 所述存储器存储 有能够被所述处理器执行的机器可执行指令, 所述机器可执行指令被所述处理器执行时, 实现如权利要求1。

10、-6任意一项所述的模型训练方法。 10.一种存储介质, 其特征在于, 所述存储介质存储有计算机程序, 所述计算机程序被 处理器执行时, 实现如权利要求1-6任意一项所述的模型训练方法。 权利要求书 2/2 页 3 CN 111753793 A 3 模型训练方法、 装置、 人脸筛选方法及电子设备 技术领域 0001 本申请涉及数据处理领域, 具体而言, 涉及一种模型训练方法、 装置、 人脸筛选方 法及电子设备。 背景技术 0002 随着深度学习技术在计算机视觉领域取得巨大成功, 应用于监控视频的人脸识别 技术也取得了长足的进步和发展。 在实际的业务中, 需要将视频图像中出现的人脸进行抓 拍, 将。

11、抓拍的人脸图像用于后续的其他应用(例如, 目标人物的识别、 目标人物跟踪等)。 为 了尽可能的将视频图像中的人脸抓拍全, 往往会将用于触发抓拍动作的相关阈值配置得较 小。 由于用于触发抓拍动作的相关阈值配置得较小, 会使得抓拍的人脸图像中会出现一些 疑似人脸的非人脸图像。 该非人脸图像会对后续的其他处理造成一定的影响。 发明内容 0003 为了克服现有技术中的至少一个不足, 本申请实施例的目的之一在于提供一种模 型训练方法, 应用于电子设备, 所述电子设备配置有未训练好的神经网络模型, 所述方法包 括: 0004 获取样本图像, 所述样本图像包括人脸关键点图像以及人脸置信度图像; 0005 通。

12、过所述人脸关键点图像训练所述神经网络模型进行人脸图像筛选, 获得待优化 的神经网络模型; 0006 保持所述待优化的神经网络模型中第一网络层的参数, 通过所述人脸置信度图像 再次训练所述待优化的神经网络模型进行人脸图像筛选, 获得训练好的神经网络模型。 0007 可选地, 所述电子设备还配置有教师网络模型, 所述教师网络模型包括所述待优 化的神经网络模型中的第二网络层的网络结构, 其中, 所述第二网络层属于所述第一网络 层的一部分; 0008 所述通过所述人脸置信度图像再次训练所述待优化的神经网络模型进行人脸图 像筛选, 获得最终训练好的神经网络模型的之前, 还包括: 0009 保持所述第二网。

13、络层的参数; 0010 通过所述人脸置信度图像训练所述教师网络模型进行人脸图像筛选, 获得训练好 的教师网络模型; 0011 所述通过所述人脸置信度图像再次训练所述待优化的神经网络模型进行人脸图 像筛选, 获得最终训练好的神经网络模型的步骤, 包括: 0012 结合所述训练好的教师网络模型, 通过所述人脸置信度图像再次训练所述待优化 的神经网络模型进行人脸图像筛选, 获得所述训练好的神经网络模型。 0013 可选地, 所述教师网络模型中的网络层数大于所述待优化的神经网络模型中的网 络层数。 0014 可选地, 所述结合所述训练好的教师网络模型, 通过所述人脸置信度图像再次训 说明书 1/9 页。

14、 4 CN 111753793 A 4 练所述待优化的神经网络模型进行人脸图像筛选, 获得所述训练好的神经网络模型的步 骤, 包括: 0015 基于所述神经网络模型的第一损失函数, 获得通过所述人脸置信度图像训练所述 待优化的神经网络模型时的第一损失值; 0016 基于所述训练好的教师网络模型的第二损失函数, 获得通过所述训练好的教师网 络模型筛选所述人脸置信度图像时的第二损失值; 0017 根据所述第一损失值的第一权值以及所述第二损失值的第二权值, 获得所述第一 损失值以及所述第二损失值的加权求和结果, 其中, 所述第一权值大于所述第二权值; 0018 判断所述加权求和结果是否小于损失阈值;。

15、 0019 若所述加权求和结果小于所述损失阈值, 则获得所述训练好的神经网络模型; 0020 若所述加权求和结果不小于损失阈值, 则根据所述加权求和结果调整所述待优化 的神经网络模型的权值, 再次从所述基于所述神经网络模型的第一损失函数, 获得通过所 述人脸置信度图像训练所述待优化的神经网络模型时的第一损失值的步骤执行。 0021 可选地, 所述教师网络模型包括残差网络。 0022 本申请实施例的目的之二在于提供一种人脸筛选方法, 应用于电子设备, 所述电 子设备配置有经模型训练方法训练好的神经网络模型, 所述方法包括: 0023 获取待筛选人脸图像; 0024 通过所述训练好的神经网络模型对。

16、所述待筛选人脸图像进行人脸图像筛选, 获得 筛选结果。 0025 本申请实施例的目的之三在于提供一种模型训练装置, 应用于电子设备, 所述电 子设备配置有未训练好的神经网络模型, 所述模型训练装置还包括: 0026 样本获取模块, 用于获取样本图像, 所述样本图像包括人脸关键点图像以及人脸 置信度图像; 0027 第一训练模块, 用于通过所述人脸关键点图像训练所述神经网络模型进行人脸图 像筛选, 获得待优化的神经网络模型; 0028 第二训练模块, 用于保持所述待优化的神经网络模型中第一网络层的参数, 通过 所述人脸置信度图像再次训练所述待优化的神经网络模型进行人脸图像筛选, 获得训练好 的神。

17、经网络模型。 0029 可选地, 所述电子设备还配置有教师网络模型, 所述教师网络模型包括所述待优 化的神经网络模型中的第二网络层的网络结构, 其中, 所述第二网络层属于所述第一网络 层的一部分, 所述第二训练模块的训练步骤, 包括: 0030 保持所述待优化的神经网络模型中第一网络层的参数; 0031 通过所述人脸置信度图像训练所述教师网络模型进行人脸图像筛选, 获得训练好 的教师网络模型; 0032 所述通过所述人脸置信度图像再次训练所述待优化的神经网络模型进行人脸图 像筛选, 获得最终训练好的神经网络模型的步骤, 包括: 0033 结合所述训练好的教师网络模型, 通过所述人脸置信度图像再。

18、次训练所述待优化 的神经网络模型进行人脸图像筛选, 获得所述训练好的神经网络模型。 0034 本申请实施例的目的之四在于提供一种电子设备, 所述电子设备包括处理器以及 说明书 2/9 页 5 CN 111753793 A 5 存储器, 所述存储器存储有能够被所述处理器执行的机器可执行指令, 所述机器可执行被 所述处理器执行时, 实现所述的模型训练方法。 0035 本申请实施例的目的之五在于提供.一种存储介质, 所述存储介质存储有计算机 程序, 所述计算机程序被处理器执行时, 实现所述的模型训练方法。 0036 相对于现有技术而言, 本申请具有以下有益效果: 0037 本申请实施例提供的模型训练。

19、方法、 装置、 人脸筛选方法及电子设备。 通过该模型 训练方法, 由于在训练时所使用了人脸关键点图像以及人脸置信度图像, 因此, 图像中所携 带的人脸关键点以及人脸置信度可以用于 “指导” 神经网络模型进行特征提取。 同时, 保持 基于人脸关键点图像所获得的待优化的神经网络结构中部分网络层的参数不变, 通过人脸 置信度图像对该待优化的神经网络模型进行训练, 调整其中其他网络层的参数, 使得训练 好的神经网络能够兼顾基于人脸关键点图像训练出的特征提取方式, 以及基于人脸置信度 图像训练处的特征提取方式, 进而提高对人脸图像的分辨精度。 附图说明 0038 为了更清楚地说明本申请实施例的技术方案,。

20、 下面将对实施例中所需要使用的附 图作简单地介绍, 应当理解, 以下附图仅示出了本申请的某些实施例, 因此不应被看作是对 范围的限定, 对于本领域普通技术人员来讲, 在不付出创造性劳动的前提下, 还可以根据这 些附图获得其他相关的附图。 0039 图1为本申请实施例提供的电子设备的硬件结构示意图; 0040 图2为本申请实施例提供的模型训练方法的步骤流程图; 0041 图3为本申请实施例提供的网络模型的结构示意图; 0042 图4为本申请实施例提供的人脸筛选方法的步骤流程图; 0043 图5为本申请实施例提供的模型训练装置的结构示意图。 0044 图标: 110-模型训练装置; 120-存储器。

21、; 130-处理器; 700-待优化的神经网络模型; 800-教师网络模型; 1101-样本获取模块; 1102-第一训练模块; 1103-第二训练模块。 具体实施方式 0045 为使本申请实施例的目的、 技术方案和优点更加清楚, 下面将结合本申请实施例 中的附图, 对本申请实施例中的技术方案进行清楚、 完整地描述, 显然, 所描述的实施例是 本申请一部分实施例, 而不是全部的实施例。 通常在此处附图中描述和示出的本申请实施 例的组件可以以各种不同的配置来布置和设计。 0046 因此, 以下对在附图中提供的本申请的实施例的详细描述并非旨在限制要求保护 的本申请的范围, 而是仅仅表示本申请的选定。

22、实施例。 基于本申请中的实施例, 本领域普通 技术人员在没有作出创造性劳动前提下所获得的所有其他实施例, 都属于本申请保护的范 围。 0047 应注意到: 相似的标号和字母在下面的附图中表示类似项, 因此, 一旦某一项在一 个附图中被定义, 则在随后的附图中不需要对其进行进一步定义和解释。 0048 在本申请的描述中, 需要说明的是, 术语 “第一” 、“第二” 、“第三” 等仅用于区分描 述, 而不能理解为指示或暗示相对重要性。 说明书 3/9 页 6 CN 111753793 A 6 0049 如背景技术所介绍的, 随着深度学习技术在计算机视觉领域取得巨大成功, 应用 于监控视频的人脸识别。

23、技术也取得了长足的进步和发展。 在实际的业务中, 需要将视频图 像中出现的人脸进行抓拍, 将抓拍的人脸图像用于后续的其他应用(例如, 目标人物的识 别、 目标人物跟踪等)。 为了尽可能的将图像中的人脸抓拍全, 往往会将用于触发抓拍动作 的相关阈值配置得较小, 使得抓拍的人脸图像中会出现一些疑似人脸的非人脸图像。 该非 人脸图像会对后续的其他处理造成一定的影响。 0050 鉴于此, 可以通过深度神经网络模型将人脸检测与人脸筛选相结合。 该方案中, 通 过深度神经网络在图像中输出人脸检测框的同时, 针对每个人脸检测框输出对应是置信度 分数。 该置信度分数用于指示该人脸检测框位置的图像属于人脸的概率。

24、。 另外, 还可以通过 对待筛选图像进行人脸关键点以及人脸角度等信息的提取, 基于提取的人脸关键点以及人 脸角度等信息判断待筛选图像是否属于人脸图像。 0051 上述方案中, 存在应用灵活性欠佳、 稳定性较差、 计算复杂度高等问题, 而且, 实际 使用效果容易受光照、 角度、 遮挡、 人物表情等因素影响。 本申请实施例提供一种模型训练 方法, 应用于电子设备, 其中, 电子设备配置有未训练好的神经网络模型。 该模型训练方法 用于对未训练好的神经网络模型进行训练, 将训练好的神经网络模型单独用于对抓拍的待 筛选人脸图像进行人脸图像筛选。 0052 其中, 针对该电子设备, 请参照图1, 为本申请。

25、实施例提供的该电子设备的结构示 意图。 该电子设备包括模型训练装置110、 存储器120以及处理器130。 0053 所述存储器120、 处理器130以及其他各元件相互之间直接或间接地电性连接, 以 实现数据的传输或交互。 例如, 这些元件相互之间可通过一条或多条通讯总线或信号线实 现电性连接。 所述模型训练装置110包括至少一个可以软件或固件(firmware)的形式存储 于所述存储器120中或固化在所述电子设备的操作系统(operating system, OS)中的软件 功能模块。 所述处理器130用于执行所述存储器120中存储的可执行模块, 例如所述模型训 练装置110所包括的软件功能。

26、模块及计算机程序等。 0054 其中, 所述存储器120可以是, 但不限于, 随机存取存储器(Random Access Memory, RAM), 只读存储器(Read Only Memory, ROM), 可编程只读存储器(Programmable Read-Only Memory, PROM), 可擦除只读存储器(Erasable Programmable Read-Only Memory, EPROM), 电可擦除只读存储器(Electric Erasable Programmable Read-Only Memory, EEPROM)等。 其中, 存储器120用于存储程序, 所述处理。

27、器130在接收到执行指令后, 执行所述程序。 0055 针对该模型训练方法, 请参照图2, 为本申请实施例提供的该模型训练方法的步骤 流程图, 以下将对方法包括各个步骤进行详细阐述。 0056 步骤S100, 获取样本图像, 样本图像包括人脸关键点图像以及人脸置信度图像。 0057 本申请实施例中, 在训练神经网络模型时通过训练样本所携带的人脸关键点 “告 知” 该神经网络模型, 如何检测提取人脸图像中的各器官的特征信息, 以及人脸图像中各器 官的相对位置信息。 同时, 通过人脸置信度图像 “告知” 该神经网络模型, 哪些样本图像具有 典型的人脸特征信息。 0058 因此, 本申请实施例中, 。

28、通过人脸关键点图像以及人脸置信度图像中所携带的关 键点信息以及置信度信息 “指导” 该神经网络模型进行特征提取。 说明书 4/9 页 7 CN 111753793 A 7 0059 步骤S200, 通过人脸关键点图像训练神经网络模型进行人脸图像筛选, 获得待优 化的神经网络模型。 0060 步骤S300, 保持待优化的神经网络模型中第一网络层的参数, 通过人脸置信度图 像再次训练待优化的神经网络模型进行人脸图像筛选, 获得训练好的神经网络模型。 0061 应理解的是, 通过人脸关键点图像训练出的待优化的神经网络模型能够提取足够 好的人脸特征。 同时, 考虑到神经网络模型所提取的浅层特征具有很大。

29、的相似性, 因此, 该 电子设备可以保持该待优化的神经网络模型中浅层网络层的参数不变, 通过人脸置信度图 像再次训练该待优化的神经网络模型, 使得训练好的神经网络能够兼顾基于人脸关键点图 像训练出的特征提取方式, 以及基于人脸置信度图像训练处的特征提取方式。 0062 在一种可能的示例中, 该待优化的神经网络模型包括9个卷积层以及2个连接层。 由于该待优化的神经网络模型中的前几层卷积层所提取的特征为浅层特征, 因此, 可以保 持第1层至第4层的参数不变, 通过人脸置信度图像对待优化的神经网络模型进行训练, 调 整剩余网络层的参数。 0063 值得说明的是, 具体保持第几层的网络参数不变, 可以。

30、根据网络层的具体数量以 及具体的训练需求进行适应性调整, 本申请实施例不做过多的限定。 0064 通过该模型训练方法, 由于在训练时所使用了人脸关键点图像以及人脸置信度图 像, 因此, 图像中所携带的人脸关键点以及人脸置信度可以用于 “指导” 神经网络模型进行 特征提取。 同时, 保持基于人脸关键点图像所获得的待优化的神经网络结构中部分网络层 的参数不变, 通过人脸置信度图像对该待优化的神经网络模型进行训练, 调整其中其他网 络层的参数, 使得训练好的神经网络能够兼顾基于人脸关键点图像训练出的特征提取方 式, 以及基于人脸置信度图像训练处的特征提取方式。 使得训练好的神经网络模型能够高 效、 。

31、准确地对抓拍的人脸图像进行筛选。 例如, 遮挡人脸、 大角度人脸或者模糊人脸等难例 人脸。 0065 进一步地, 由于神经网络模型的计算量与网络层数成正相关, 但考虑到网络层数 越多, 能够获得更好的拟合效果, 因此, 本申请实施例中, 为了提高神经网络模型计算效率 同时, 保证该神经网络模型的拟合效果。 0066 请参照图3, 该电子设备还配置有用于监督该待优化的神经网络模型700的教师网 络模型800。 该教师网络模型800包括待优化的神经网络模型700中第二网络层的网络结构, 即该教师网络模型800与待优化的神经网络模型700共享该第二网络层的结构以及参数, 其 中, 该第二网络层属于第。

32、一网络层的一部分。 0067 应理解的是, 该教师网络模型800所包括的网络层数大于该待优化的神经网络模 型700所包括的网络层数, 如此, 以确保该教师网络模型800相较于该待优化的神经网络模 型700, 具有更好的拟合效果。 0068 然而, 由于教师网络模型800的网络层数与待优化的神经网络模型700所包括的网 络层数相差较大时, 会使得教师网络模型800与待优化的神经网络模型700之间在模型结构 上存在较大的差异, 进而导致教师网络模型800与待优化的神经网络模型700之间出现信息 鸿沟, 使得教师网络模型800并不能对待优化的神经网络模型700起到监督效果, 因此, 教师 网络模型8。

33、00所包括的网络层数不能远大于待优化的神经网络模型700所包括的网络层数。 0069 经研究发现, 该教师网络模型800的网络层数是待优化的神经网络模型700的网络 说明书 5/9 页 8 CN 111753793 A 8 层数的1.5-2倍时, 教师网络模型800能够对待优化的神经网络模型700起到较好的监督效 果。 0070 同时, 由于教师网络模型800所包括的网络层数大于该待优化的神经网络模型700 所包括的网络层数, 为了避免因为网络层数的增加所带来的退化问题, 即随着网络层数增 加, 但教师网络模型800在训练集上的准确率却饱和甚至下降。 该教师网络模型800包括一 定数量的残差网。

34、络, 用以克服退化问题。 0071 由于该教师网络模型800与待优化的神经网络模型700共享第二网络层的结构以 及参数, 为了避免人脸置信度图像影响浅层网络的权值参数, 在本申请实施例中, 该电子设 备保持该第二网络层的参数不变, 通过人脸置信度图像训练该教师网络模型进行人脸图像 筛选, 获得训练好的教师网络模型800。 0072 其中, 值得说明的是, 由于教师网络模型800与待优化的神经网络模型700共享第 二网络层的结构以及参数, 在相同的样本图像的情况下, 教师网络模型800的拟合能力会强 于待优化的神经网络模型700。 为了待优化的神经网络模型700的鲁棒性能最大程度的逼近 教师网络。

35、模型, 经研究发现, 教师网络模型800保持参数不变的网络层数越少, 待优化的神 经网络模型700保持参数不变的层数越多, 会使得待优化的神经网络模型700更容易拟合样 本图像。 即第二网络层属于第一网络层的一部分时, 会使得待优化的神经网络模型700更容 易拟合样本图像。 0073 其中, 在确定第二网络层的具体数量时, 研究发现, 第二网络层的数量选取教师网 络模型800中网络层数的0.2-0.25倍时, 可以使得教师网络模型800能够获得较好的浅层特 征以及较好的鲁棒性能。 0074 进一步地, 该电子设备将训练好的教师网络模型800在筛选人脸置信度图像时的 输出结果作为参考, 通过人脸。

36、置信度图像再次训练待优化的神经网络模型700进行人脸图 像筛选, 获得训练好的神经网络模型。 0075 其中, 通过人脸置信度图像再次训练待优化的神经网络模型700进行人脸图像筛 选时, 作为一种可能的实施方式, 该电子设备保持待优化的神经网络模型700中第二网络层 的权值, 基于神经网络模型的第一损失函数, 获得通过人脸置信度图像训练待优化的神经 网络模型700时的第一损失值; 基于训练好的教师网络模型的第二损失函数, 获得通过训练 好的教师网络模型筛选人脸置信度图像时的第二损失值。 0076 其中, 该第一损失函数可以是欧式距离损失函数, 第二损失函数可以是sigmoid损 失函数。 00。

37、77 该电子设备根据第一损失值的第一权值以及第二损失值的第二权值, 获得第一损 失值以及第二损失值的加权求和结果, 其中, 第一权值大于第二权值; 判断加权求和结果是 否小于损失阈值; 若加权求和结果小于损失阈值, 则获得训练好的神经网络模型。 0078 若加权求和结果不小于损失阈值, 该电子设备则根据加权求和结果调整待优化的 神经网络模型700的权值, 再次对输入的人脸置信度图像进行筛选, 直到该加权求和结果小 于该损失阈值。 0079 由于通过人脸置信度图像进行训练的主要目标是待优化的神经网络模型, 因此, 第一权值大于第二权值时, 能够取的比较好的训练结果。 0080 本申请实施例还提供。

38、一种人脸筛选方法, 应用于电子设备, 该电子设备配置有经 说明书 6/9 页 9 CN 111753793 A 9 上述模型训练方法训练好的神经网络模型。 请参照图4, 为本申请实施例提供的该人脸筛选 方法的步骤流程图。 0081 步骤S500, 获取待筛选人脸图像。 0082 其中, 该待筛选人脸图像可以是图像采集装置抓拍的人脸图像。 为了确保将图像 中的人脸抓拍全, 会将用于触发抓拍动作的相关阈值配置得较小。 由于用于触发抓拍动作 的相关阈值配置得较小, 会使得抓拍的人脸图像中会出现一些疑似人脸的非人脸图像。 例 如, 头盔等球状物体的图像。 0083 步骤S600, 通过训练好的神经网络。

39、模型对待筛选人脸图像进行人脸图像筛选, 获 得筛选结果。 0084 该电子设备通过训练好的神经网络模型对待筛选人脸图像处理后, 输出一概率 值, 将该概率值与概率阈值进行比较, 若超过该概率阈值, 则将待筛选人脸图像确定为真正 的人脸图像, 并对其进行输出。 若未超过该预设阈值, 则该待筛选人脸图像为非人脸图像。 0085 其中, 该概率阈值可以是0.5。 0086 本申请实施例还提供给一种模型训练装置110, 应用于电子设备。 该电子设备配置 有未训练好的神经网络模型, 模型训练装置110包括至少一个可以软件形式存储于存储器 120中的功能模块。 请参照图5, 从功能上划分, 模型训练装置1。

40、10可以包括: 0087 样本获取模块1101, 用于获取样本图像, 样本图像包括人脸关键点图像以及人脸 置信度图像。 0088 在本申请实施例中, 该样本获取模块1101用于执行图2中的步骤S100, 关于该样本 获取模块1101的详细描述可以参考步骤S100的详细描述。 0089 第一训练模块1102, 用于通过人脸关键点图像训练神经网络模型进行人脸图像筛 选, 获得待优化的神经网络模型700。 0090 在本申请实施例中, 该第一训练模块1102用于执行图2中的步骤S200, 关于该第一 训练模块1102的详细描述可以参考步骤S200的详细描述。 0091 第二训练模块1103, 用于保。

41、持待优化的神经网络模型700中第一网络层的参数, 通 过人脸置信度图像再次训练待优化的神经网络模型700进行人脸图像筛选, 获得训练好的 神经网络模型。 0092 在本申请实施例中, 该第二训练模块1103用于执行图2中的步骤S300, 关于该第二 训练模块1103的详细描述可以参考步骤S300的详细描述。 0093 其中, 该电子设备还配置有教师网络模型800。 该教师网络模型800包括待优化的 神经网络模型700中的第二网络层的网络结构, 其中, 第二网络层属于第一网络层的一部 分。 其中, 第二训练模块1103的训练步骤, 包括: 0094 保持待优化的神经网络模型700中第一网络层的参。

42、数; 0095 通过人脸置信度图像训练教师网络模型进行人脸图像筛选, 获得训练好的教师网 络模型; 0096 通过人脸置信度图像再次训练待优化的神经网络模型700进行人脸图像筛选, 获 得最终训练好的神经网络模型的步骤, 包括: 0097 结合训练好的教师网络模型, 通过人脸置信度图像再次训练待优化的神经网络模 型700进行人脸图像筛选, 获得训练好的神经网络模型。 说明书 7/9 页 10 CN 111753793 A 10 0098 本申请实施例还提供一种电子设备。 该电子设备包括处理器130以及存储器120。 该存储器120存储有能够被处理器130执行的机器可执行指令。 该机器可执行被处。

43、理器130 执行时, 实现上述模型训练方法。 0099 其中, 该电子设备可以是, 但不限于, 智能手机、 个人电脑(personal computer, PC)、 平板电脑、 个人数字助理(personal digital assistant, PDA)、 移动上网设备(mobile Internet device, MID)、 服务器以及图像采集设备等。 0100 本申请实施例还提供一种存储介质。 该存储介质存储有计算机程序, 计算机程序 被处理器130执行时, 实现上述模型训练方法。 0101 综上所述, 本申请实施例提供的模型训练方法、 装置、 人脸筛选方法及电子设备。 通过该模型训练。

44、方法, 由于在训练时所使用了人脸关键点图像以及人脸置信度图像, 因此, 图像中所携带的人脸关键点以及人脸置信度可以用于 “指导” 神经网络模型进行特征提取。 同时, 保持基于人脸关键点图像所获得的待优化的神经网络结构中部分网络层的参数不 变, 通过人脸置信度图像对该待优化的神经网络模型进行训练, 调整其中其他网络层的参 数, 使得训练好的神经网络能够兼顾基于人脸关键点图像训练出的特征提取方式, 以及基 于人脸置信度图像训练处的特征提取方式, 进而提高对人脸图像的分辨精度。 0102 在本申请所提供的实施例中, 应该理解到, 所揭露的装置和方法, 也可以通过其它 的方式实现。 以上所描述的装置实。

45、施例仅仅是示意性的, 例如, 附图中的流程图和框图显示 了根据本申请的多个实施例的装置、 方法和计算机程序产品的可能实现的体系架构、 功能 和操作。 在这点上, 流程图或框图中的每个方框可以代表一个模块、 程序段或代码的一部 分, 所述模块、 程序段或代码的一部分包含一个或多个用于实现规定的逻辑功能的可执行 指令。 也应当注意, 在有些作为替换的实现方式中, 方框中所标注的功能也可以以不同于附 图中所标注的顺序发生。 例如, 两个连续的方框实际上可以基本并行地执行, 它们有时也可 以按相反的顺序执行, 这依所涉及的功能而定。 也要注意的是, 框图和/或流程图中的每个 方框、 以及框图和/或流程。

46、图中的方框的组合, 可以用执行规定的功能或动作的专用的基于 硬件的系统来实现, 或者可以用专用硬件与计算机指令的组合来实现。 0103 另外, 在本申请各个实施例中的各功能模块可以集成在一起形成一个独立的部 分, 也可以是各个模块单独存在, 也可以两个或两个以上模块集成形成一个独立的部分。 0104 所述功能如果以软件功能模块的形式实现并作为独立的产品销售或使用时, 可以 存储在一个计算机可读取存储介质中。 基于这样的理解, 本申请的技术方案本质上或者说 对现有技术做出贡献的部分或者该技术方案的部分可以以软件产品的形式体现出来, 该计 算机软件产品存储在一个存储介质中, 包括若干指令用以使得一。

47、台计算机设备(可以是个 人计算机, 服务器, 或者网络设备等)执行本申请各个实施例所述方法的全部或部分步骤。 而前述的存储介质包括: U盘、 移动硬盘、 只读存储器(ROM, Read-Only Memory)、 随机存取存 储器(RAM, Random Access Memory)、 磁碟或者光盘等各种可以存储程序代码的介质。 0105 需要说明的是, 在本文中, 诸如第一和第二等之类的关系术语仅仅用来将一个实 体或者操作与另一个实体或操作区分开来, 而不一定要求或者暗示这些实体或操作之间存 在任何这种实际的关系或者顺序。 而且, 术语 “包括” 、“包含” 或者其任何其他变体意在涵盖 非排。

48、他性的包含, 从而使得包括一系列要素的过程、 方法、 物品或者设备不仅包括那些要 素, 而且还包括没有明确列出的其他要素, 或者是还包括为这种过程、 方法、 物品或者设备 说明书 8/9 页 11 CN 111753793 A 11 所固有的要素。 在没有更多限制的情况下, 由语句 “包括一个” 限定的要素, 并不排除在 包括所述要素的过程、 方法、 物品或者设备中还存在另外的相同要素。 0106 以上所述, 仅为本申请的各种实施方式, 但本申请的保护范围并不局限于此, 任何 熟悉本技术领域的技术人员在本申请揭露的技术范围内, 可轻易想到变化或替换, 都应涵 盖在本申请的保护范围之内。 因此, 本申请的保护范围应所述以权利要求的保护范围为准。 说明书 9/9 页 12 CN 111753793 A 12 图1 图2 说明书附图 1/2 页 13 CN 111753793 A 13 图3 图4 图5 说明书附图 2/2 页 14 CN 111753793 A 14 。

展开阅读全文
内容关键字: 模型 训练 方法 装置 筛选 电子设备
关于本文
本文标题:模型训练方法、装置、人脸筛选方法及电子设备.pdf
链接地址:https://www.zhuanlichaxun.net/pdf/10304542.html
关于我们 - 网站声明 - 网站地图 - 资源地图 - 友情链接 - 网站客服 - 联系我们

copyright@ 2017-2018 zhuanlichaxun.net网站版权所有
经营许可证编号:粤ICP备2021068784号-1