[发明专利]目标区域确定模型的训练方法、装置和计算机可读存储介质在审
申请号: | 201910485054.4 | 申请日: | 2019-06-05 |
公开(公告)号: | CN110287954A | 公开(公告)日: | 2019-09-27 |
发明(设计)人: | 朱延东;周恺卉;王长虎 | 申请(专利权)人: | 北京字节跳动网络技术有限公司 |
主分类号: | G06K9/32 | 分类号: | G06K9/32;G06K9/46 |
代理公司: | 北京竹辰知识产权代理事务所(普通合伙) 11706 | 代理人: | 陈龙 |
地址: | 100041 北京市石景山区*** | 国省代码: | 北京;11 |
权利要求书: | 查看更多 | 说明书: | 查看更多 |
摘要: | |||
搜索关键词: | 目标区域 关键点 训练样本集合 计算机可读存储介质 卷积神经网络 目标区域识别 电子设备 收敛条件 样本图像 预测 准确率 | ||
1.一种目标区域确定模型的训练方法,其特征在于,包括:
获取训练样本集合;其中,所述训练样本集合由多个标记了目标区域关键点的样本图像组成;
确定卷积神经网络的参数;
将所述训练样本集合输入所述卷积神经网络,得到每个样本图像的预测关键点;
根据每个样本图像的预测关键点和真实关键点确定预测误差;
若所述预测误差大于预设误差,则根据所述预设误差重新确定所述卷积神经网络的参数,继续重复训练过程,直至所述预测误差小于或等于所述预设误差,结束训练过程,得到目标区域确定模型。
2.根据权利要求1所述的方法,其特征在于,所述目标区域为车牌区域。
3.根据权利要求2所述的方法,其特征在于,所述关键点为所述车牌区域四个顶角对应的点。
4.一种目标区域确定方法,其特征在于,包括:
获取待识别图像;
将所述待识别图像输入采用权利要求1-3任一项所述的目标区域确定模型的训练方法进行训练得到的目标区域确定模型;
通过所述目标区域确定模型预测得到多个关键点;
根据所述多个关键点确定目标区域。
5.根据权利要求4所述的方法,其特征在于,所述根据所述多个关键点确定目标区域,包括:
从所述多个关键点中筛选出预设个数关键点;
根据所述预设个数关键点确定所述目标区域。
6.根据权利要求5所述的方法,其特征在于,所述从所述多个关键点中筛选出预设个数关键点,包括:
通过目标区域识别模型对所述待识别图像进行识别,得到预测目标区域;
从所述多个关键点中筛选出位于所述预测目标区域内的预设个数关键点。
7.根据权利要求4所述的方法,其特征在于,所述根据所述多个关键点确定目标区域,包括:
从所述多个关键点中选取预设个数关键点进行连线,得到多个闭合区域;
根据预设条件从所述多个闭合区域中筛选出一个闭合区域作为所述目标区域。
8.根据权利要求4-7任一项所述的方法,其特征在于,所述目标区域为车牌区域。
9.一种目标区域确定模型的训练装置,其特征在于,包括:
样本获取模块,用于获取训练样本集合;其中,所述训练样本集合由多个标记了目标区域关键点的样本图像组成;
目标区域确定模型目标区域确定模型
参数确定模块,用于确定卷积神经网络的参数;
关键点预测模块,用于将所述训练样本集合输入所述卷积神经网络,得到每个样本图像的预测关键点;
误差确定模块,用于根据每个样本图像的预测关键点和真实关键点确定预测误差;
参数调整模块,用于若所述预测误差大于预设误差,则根据所述预设误差重新确定所述卷积神经网络的参数,继续重复训练过程,直至所述预测误差小于或等于所述预设误差,结束训练过程,得到目标区域确定模型。
10.一种目标区域确定装置,其特征在于,包括:
图像获取模块,用于获取待识别图像;
图像输入模块,用于将所述待识别图像输入采用权利要求1-3任一项所述的目标区域确定模型的训练方法进行训练得到的目标区域确定模型;
预测模块,用于通过所述目标区域确定模型预测得到多个关键点;
区域确定模块,用于根据所述多个关键点确定目标区域。
11.一种电子设备,包括:
存储器,用于存储非暂时性计算机可读指令;以及
处理器,用于运行所述计算机可读指令,使得所述处理器执行时实现根据权利要求1-3中任意一项所述的目标区域确定模型的训练方法。
该专利技术资料仅供研究查看技术是否侵权等信息,商用须获得专利权人授权。该专利全部权利属于北京字节跳动网络技术有限公司,未经北京字节跳动网络技术有限公司许可,擅自商用是侵权行为。如果您想购买此专利、获得商业授权和技术合作,请联系【客服】
本文链接:http://www.vipzhuanli.com/pat/books/201910485054.4/1.html,转载请声明来源钻瓜专利网。