一种基于双通道循环生成对抗网络的红外视频彩色化方法

摘要

本发明涉及一种基于双通道循环生成对抗网络的红外视频彩色化方法，包括以下步骤：采集夜视红外视频数据和彩色视频数据；将视频数据集抽帧处理后放入数据库，构建训练集和测试集；构建双通道循环生成对抗深度学习网络，通过循环训练的方法提高生成网络G、生成网络F、判别网络DX和判别网络DY的能力；实时获取红外视频中的红外图像，将该红外图像与上一帧红外图像彩色化的结果输入经过训练后的生成网络G，由生成网络G对红外图像序列进行彩色化处理。本发明增强了夜视红外视频的彩色化效果，提高了彩色化视频的可观察性，无需人工干预。