[发明专利]跟踪模型的训练方法、装置、介质和计算设备

说明书

本发明的实施方式提供了一种跟踪模型的训练方法、装置、介质和计算设备。该方法包括：基于对训练数据集进行预训练得到的网络参数构建回归模型；基于所述回归模型对跟踪模型的参数进行更新，以实现对所述跟踪模型的训练。本发明能够基于预训练得到的网络参数构建回归模型，并基于回归模型对跟踪模型进行训练，即能够利用网络参数等先验知识构建的回归模型对跟踪模型进行训练，从而在训练过程中基于先验知识快速的实现对跟踪模型进行训练的训练目标，极大地缩短了训练时间，提升了跟踪模型的训练效率。

技术领域

本发明的实施方式涉及深度学习技术领域，更具体地，本发明的实施方式涉及一种跟踪模型的训练方法、装置、介质和计算设备。

背景技术

本部分旨在为权利要求书中陈述的本发明的实施方式提供背景或上下文。此处的描述不因为包括在本部分中就承认是现有技术。

近年来，随着人工智能技术的快速发展，人工智能技术在教育领域也有了广泛应用，为提升教学质量与效果，帮助学生更高效的学习书法，并发扬中华传统书法文化魅力，出现了基于人工智能技术对书法学习时笔尖的运动轨迹进行跟踪的模型。目前，通常需要对跟踪模型进行训练，才能将跟踪模型投入实际的应用中。然而，在对跟踪模型进行训练的过程中模型需要寻找全局最优解，在寻找全局最优解的过程中会限于到低学习率区域，从而导致对跟踪模型的训练需要耗费大量的时间，降低了跟踪模型的训练效率。

发明内容

在本上下文中，本发明的实施方式期望提供一种跟踪模型的训练方法、装置、介质和计算设备。

在本发明实施方式的第一方面中，提供了一种跟踪模型的训练方法，包括：

基于对训练数据集进行预训练得到的网络参数构建回归模型；

基于所述回归模型对跟踪模型的参数进行更新，以实现对所述跟踪模型的训练。

在本实施方式的一个实施例中，基于对训练数据集进行预训练得到的网络参数构建回归模型，包括：

对输入的训练数据集进行预训练，得到网络参数和所述网络参数对应的矩阵数据集；

基于所述矩阵数据集构建回归模型。

在本实施方式的一个实施例中，对输入的训练数据集进行预训练，得到网络参数和所述网络参数对应的矩阵数据集，包括：

基于输入的训练数据集对跟踪模型进行预训练，得到所述跟踪模型的网络参数；

从所述网络参数中获取权值矩阵、偏置矩阵、权值梯度矩阵以及偏置梯度矩阵；

基于所述权值矩阵和所述权值梯度矩阵，生成权值梯度矩阵数据集；

基于所述偏置矩阵和所述偏置梯度矩阵，生成偏置梯度矩阵数据集；

将所述权值梯度矩阵数据集和所述偏置梯度矩阵数据集确定为矩阵数据集。

在本实施方式的一个实施例中，基于所述矩阵数据集构建回归模型，包括：

确定所述权值梯度矩阵数据集对应的第一卷积层，并构建所述第一卷积层对应的权值回归模型；

确定所述偏置梯度矩阵数据集对应的第二卷积层，并构建所述第二卷积层对应的偏置回归模型。

在本实施方式的一个实施例中，对所述跟踪模型进行训练的算法为梯度下降算法和改进算法，其中：

获取对所述跟踪模型进行的最新两次连续训练的训练正确率；

基于所述最新两次连续训练的训练正确率，计算正确率增幅；

当所述正确率增幅小于预设阈值时，获取当前算法；

当所述当前算法为所述梯度下降算法时，将所述当前算法更新为所述改进算法；