[发明专利]图像信息预测方法、装置、计算机设备和存储介质

说明书

本申请涉及一种图像信息预测方法、装置、计算机设备和存储介质。方法包括：从历史视频流中提取多帧的历史图像；提取各历史图像中的历史关键点；获取各历史关键点对应的历史关键点位置；根据各历史关键点位置预测未来帧图像的预测关键点位置，其中未来帧图像是视频流中的历史图像的下一帧图像；计算各预测关键点位置对应的预测位置热力图；根据各预测位置热力图以及历史图像进行模型训练得到预测模型；从目标视频流中提取目标历史图像，将目标历史图像输入至预测模型中，以根据预测模型得到包含关键点信息的目标未来帧图像。采用本方法能够提高图像信息预测的准确率。

技术领域

本申请涉及计算机技术领域，特别是涉及一种图像信息预测方法、装置、计算机设备和存储介质。

背景技术

视频预测可以应用于很多场景，如自动驾驶中的行人、汽车等移动轨迹预测，气象要素(温度、湿度、风速等)的未来场预测以及雷达回波的临近预测等。

传统技术中是直接根据历史数据中的历史关键点预测未来数据中的未来关键点，由于关键点中包含的信息有限，使得在预测过程中可利用的信息较少，造成预测的准确度低下。

发明内容

基于此，有必要针对上述技术问题，提供一种能够预测准确度的图像信息预测方法、装置、计算机设备和存储介质。

一种图像信息预测方法，方法包括：

从历史视频流中提取多帧的历史图像；

提取各历史图像中的历史关键点；

获取各历史关键点对应的历史关键点位置；

根据各历史关键点位置预测未来帧图像的预测关键点位置，其中未来帧图像是视频流中的历史图像的下一帧图像；

计算各预测关键点位置对应的预测位置热力图；

根据各预测位置热力图以及历史图像进行模型训练得到预测模型；

从目标视频流中提取目标历史图像，将目标历史图像输入至预测模型中，以根据预测模型得到包含图像信息预测的目标未来帧图像。

在其中一个实施例中，计算各预测关键点位置对应的预测位置热力图，包括：

利用高斯函数将各预测关键点位置转换为预测位置热力图。

在其中一个实施例中，根据各预测位置热力图以及历史图像进行模型训练得到预测模型，包括：

获取各预测关键点位置对应的预缓存图像；

将各预测关键点位置对应的预测位置热力图存储至对应的预缓存图像中；

将存储有预测位置热力图的各预缓存图像进行拼接融合得到综合的预测位置热力图；

根据综合的预测位置热力图以及历史图像进行模型训练得到预测模型。

在其中一个实施例中，根据综合的预测位置热力图以及历史图像进行模型训练得到预测模型，包括：

从多帧的历史图像中选取与当前时间最接近的最新历史帧图像；

获取最新历史帧图像对应的未来帧图像，以及未来帧图像对应的综合的最新预测位置热力图；

将综合的最新预测位置热力图以及最新历史帧图像输入至机器学习模型，以对机器学习模型进行训练得到预测参数，根据预测参数得到初始预测模型；