[发明专利]基于视频增强的目标跟踪方法、装置、设备及存储介质

说明书

本申请提出一种基于视频增强的目标跟踪方法、装置、设备及存储介质，该方法包括：获取待增强视频数据；通过预先训练的低光图像增强网络对待增强视频数据进行增强处理，得到增强后的视频数据；通过预设目标跟踪网络对增强后的视频数据进行目标跟踪处理，得到待增强视频数据对应的目标跟踪视频序列。本申请构建并训练了低光图像增强网络，通过低光图像增强网络来对待增强视频数据进行增强，提高了待增强视频数据中每个视频帧中的对比度及色度，减少了每个视频帧中的噪声，使得待增强视频数据中的细节更加清晰，便于识别待跟踪目标。在此视频增强基础上对戴增强视频数据进行目标跟踪，大大提高了目标跟踪的准确率。

技术领域

本申请属于视频处理技术领域，具体涉及一种基于视频增强的目标跟踪方法、装置、设备及存储介质。

背景技术

拍摄图像或视频时常受到环境的限制，使得拍摄的图像或视频存在亮度不足、对比度低、噪声严重等缺点。如夜晚的监控录像因受光线严重不足的限制，录像都有着极度黑暗、细节模糊以及噪点严重的问题，难以辨别清楚录像中的目标类别，对目标检测与跟踪有着相当大的阻碍。因此需要对这种图像或视频进行增强处理。

当前，相关技术中提供了一些基于视频增强的目标跟踪方法，例如多尺度Retinex(图像去雾算法)低光增强技术，经由该技术增强后，虽然获得了高亮度图像，但其对比度、色度以及纹理细节却遭到了一定破坏，并且底层噪声也伴随着亮度一起被放大，最终得到的视频图像效果并不能满足人眼视觉效果，仍然难以辨别清楚视频图像中的目标，难以实现目标检测与跟踪的任务。

发明内容

本申请提出一种基于视频增强的目标跟踪方法、装置、设备及存储介质，通过预先训练的低光图像增强网络来对待增强视频数据进行增强，提高了待增强视频数据中每个视频帧中的对比度及色度，减少了每个视频帧中的噪声，使得待增强视频数据中的细节更加清晰，便于识别待跟踪目标。在此视频增强基础上对戴增强视频数据进行目标跟踪，大大提高了目标跟踪的准确率。

本申请第一方面实施例提出了一种基于视频增强的目标跟踪方法，包括：

获取待增强视频数据；

通过预先训练的低光图像增强网络对所述待增强视频数据进行增强处理，得到增强后的视频数据；

通过预设目标跟踪网络对所述增强后的视频数据进行目标跟踪处理，得到所述待增强视频数据对应的目标跟踪视频序列。

在本申请的一些实施例中，所述通过预先训练的低光图像增强网络对所述待增强视频数据进行增强处理，得到增强后的视频数据之前，还包括：

构建低光图像增强网络的网络结构；

获取训练集，所述训练集中包括夜间视频图像；

根据所述训练集对构建的所述低光图像增强网络进行训练，得到训练好的低光图像增强网络。

在本申请的一些实施例中，所述构建低光图像增强网络的网络结构，包括：

将第一卷积层和激活层串联，得到特征提取模块；

将第二卷积层、第三卷积层、第四卷积层、第五卷积层、第六卷积层和第七卷积层依次串联，得到图像增强模块；

将预设数目个所述特征提取模块依次串联；

将每个所述特征提取模块分别连接一个所述图像增强模块；

将每个所述图像增加模块均与全连接层连接，得到低光图像增强网络的网络结构。

在本申请的一些实施例中，所述根据所述训练集对构建的所述低光图像增强网络进行训练，得到训练好的低光图像增强网络，包括：

从所述训练集中获取所述夜间视频图像；