[发明专利]一种驾驶模型训练方法、装置和终端设备有效
申请号: | 201811639549.X | 申请日: | 2018-12-29 |
公开(公告)号: | CN109685214B | 公开(公告)日: | 2021-03-02 |
发明(设计)人: | 闫泳杉 | 申请(专利权)人: | 百度在线网络技术(北京)有限公司 |
主分类号: | G06N3/08 | 分类号: | G06N3/08 |
代理公司: | 北京银龙知识产权代理有限公司 11243 | 代理人: | 许静;黄灿 |
地址: | 100085 北京市*** | 国省代码: | 北京;11 |
权利要求书: | 查看更多 | 说明书: | 查看更多 |
摘要: | |||
搜索关键词: | 一种 驾驶 模型 训练 方法 装置 终端设备 | ||
1.一种驾驶模型训练方法,其特征在于,包括:
获取基础驾驶模型,其中,所述基础驾驶模型为使用第一场景的样本数据进行训练得到的驾驶模型;
使用第二场景的样本数据对所述基础驾驶模型进行训练,以得到用于在所述第二场景进行驾驶的目标驾驶模型;
其中,所述基础驾驶模型包括多个卷积块,所述第一场景和所述第二场景存在共同道路特征,所述使用第二场景的样本数据对所述基础驾驶模型进行训练,以得到用于在第二场景进行驾驶的目标驾驶模型,包括:
使用第二场景的样本数据对所述基础驾驶模型进行多次微调,其中,每次微调时冻结所述基础驾驶模型的部分卷积块,直到所述基础驾驶模型的损失值低于特定门限值的,以获得用于在所述第二场景进行驾驶的目标驾驶模型;
所述多次微调中每次微调所冻结的卷积块数量不同;或者,
所述多次微调中存在所冻结的卷积块数量相同,但存在不同的卷积块的至少两次微调。
2.如权利要求1所述的方法,其特征在于,所述使用第二场景的样本数据对所述基础驾驶模型进行训练,以得到用于在第二场景进行驾驶的目标驾驶模型之前,所述方法还包括:
将预设学习率调低,以得到用于所述目标驾驶模型训练的目标学习率,其中,所述预设学习率为所述基础驾驶模型训练过程中使用的学习率或者通用学习率;
所述使用第二场景的样本数据对所述基础驾驶模型进行训练,以得到用于在第二场景进行驾驶的目标驾驶模型,包括:
按照所述目标学习率,使用第二场景的样本数据对所述基础驾驶模型进行训练,以得到用于在所述第二场景进行驾驶的目标驾驶模型。
3.如权利要求1或2所述的方法,其特征在于,所述第一场景为开放道路,所述第二场景为园区。
4.一种驾驶模型训练装置,其特征在于,包括:
获取模块,用于获取基础驾驶模型,其中,所述基础驾驶模型为使用第一场景的样本数据进行训练得到的驾驶模型;
训练模块,用于使用第二场景的样本数据对所述基础驾驶模型进行训练,以得到用于在所述第二场景进行驾驶的目标驾驶模型;
其中,所述基础驾驶模型包括多个卷积块,所述第一场景和所述第二场景存在共同道路特征,所述训练模块用于使用第二场景的样本数据对所述基础驾驶模型进行多次微调,其中,每次微调时冻结所述基础驾驶模型的部分卷积块,直到所述基础驾驶模型的损失值低于特定门限值的,以获得用于在所述第二场景进行驾驶的目标驾驶模型;
所述多次微调中每次微调所冻结的卷积块数量不同;或者,
所述多次微调中存在所冻结的卷积块数量相同,但存在不同的卷积块的至少两次微调。
5.如权利要求4所述的装置,其特征在于,所述装置还包括:
调整模块,用于将预设学习率调低,以得到用于所述目标驾驶模型训练的目标学习率,其中,所述预设学习率为所述基础驾驶模型训练过程中使用的学习率或者通用学习率;
所述训练模块用于按照所述目标学习率,使用第二场景的样本数据对所述基础驾驶模型进行训练,以得到用于在第二场景进行驾驶的目标驾驶模型。
6.如权利要求4或5所述的装置,其特征在于,所述第一场景为开放道路,所述第二场景为园区。
7.一种终端设备,其特征在于,包括处理器、存储器及存储在所述存储器上并可在所述处理器上运行的计算机程序,所述计算机程序被所述处理器执行时实现如权利要求1至3中任一项所述的驾驶模型训练方法的步骤。
8.一种计算机可读存储介质,其特征在于,所述计算机可读存储介质上存储有计算机程序,所述计算机程序被处理器执行时实现如权利要求1至3中任一项所述的驾驶模型训练方法的步骤。
该专利技术资料仅供研究查看技术是否侵权等信息,商用须获得专利权人授权。该专利全部权利属于百度在线网络技术(北京)有限公司,未经百度在线网络技术(北京)有限公司许可,擅自商用是侵权行为。如果您想购买此专利、获得商业授权和技术合作,请联系【客服】
本文链接:http://www.vipzhuanli.com/pat/books/201811639549.X/1.html,转载请声明来源钻瓜专利网。