[发明专利]高速铁路车载视频去雾霾方法和装置

说明书

本发明提供了一种高速铁路车载视频去雾霾方法和装置，该方法包括：获取雾霾图像和不成对清晰图像；利用循环生成对抗网络，构建去雾霾网络；根据雾霾图像和不成对清晰图像，对去雾霾网络进行不成对训练，确定去雾霾图像；根据去雾霾图像，对完成不成对训练的去雾霾网络进行有监督训练，确定训练好的去雾霾网络；根据训练好的去雾霾网络，去除高速铁路车载视频中含有雾霾的图像，确定无雾霾的高速铁路车载视频。本发明通过对去雾霾网络进行不成对训练和有监督训练，可以提升对真实采集的雾霾图像的去雾霾效果。

技术领域

本发明涉及图像处理技术领域，尤其涉及一种高速铁路车载视频去雾霾方法和装置。

背景技术

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

高速铁路车载视频常用于运行环境安全检查任务，但视频画面质量易受沿线天气影响，如雾霾、沙尘暴等，非常不利于自动化的运行环境安全检查。因此，需要去除视频画面中的雾霾，提升图像质量。现有的图像去雾霾方法主要针对合成的雾霾图像，有对应的清晰图像，可以进行有监督的训练。

但是这些合成雾霾图像训练的模型应用到真实雾霾图像上时，去雾霾效果比较差。然而，实际中的真实雾霾图像通常无对应的清晰图像，无法进行有监督的训练。

因此，如何提供一种新的方案，其能够解决上述技术问题是本领域亟待解决的技术难题。

发明内容

本发明实施例提供一种高速铁路车载视频去雾霾方法，通过对去雾霾网络进行不成对训练和有监督训练，可以提升对真实采集的雾霾图像的去雾霾效果，该方法包括：

获取雾霾图像和不成对清晰图像；

利用循环生成对抗网络，构建去雾霾网络；

根据雾霾图像和不成对清晰图像，对去雾霾网络进行不成对训练，确定去雾霾图像；

根据去雾霾图像，对完成不成对训练的去雾霾网络进行有监督训练，确定训练好的去雾霾网络；

根据训练好的去雾霾网络，去除高速铁路车载视频中含有雾霾的图像，确定无雾霾的高速铁路车载视频。

本发明实施例还提供一种高速铁路车载视频去雾霾装置，包括：

图像获取模块，用于获取雾霾图像和不成对清晰图像；

去雾霾网络构建模块，用于利用循环生成对抗网络，构建去雾霾网络；

不成对训练模块，用于根据雾霾图像和不成对清晰图像，对去雾霾网络进行不成对训练，确定去雾霾图像；

有监督训练模块，用于根据去雾霾图像，对完成不成对训练的去雾霾网络进行有监督训练，确定训练好的去雾霾网络；

高速铁路车载视频去雾霾模块，用于根据训练好的去雾霾网络，去除高速铁路车载视频中含有雾霾的图像，确定无雾霾的高速铁路车载视频。

本发明实施例还提供一种计算机设备，包括存储器、处理器及存储在存储器上并可在处理器上运行的计算机程序，所述处理器执行所述计算机程序时实现上述一种高速铁路车载视频去雾霾方法。

本发明实施例还提供一种计算机可读存储介质，所述计算机可读存储介质存储有计算机程序，所述计算机程序被处理器执行时实现上述一种高速铁路车载视频去雾霾方法。

本发明实施例还提供一种计算机程序产品，所述计算机程序产品包括计算机程序，所述计算机程序被处理器执行时实现上述一种高速铁路车载视频去雾霾方法。