摘要

本发明公开了一种基于深度神经网络的视频自动着色方法,包括以下步骤:获取原始彩色视频数据集,将彩色视频转化为黑白视频,得到用于网络训练的黑白视频帧序列和彩色视频帧序列;分别计算彩色视频帧序列和黑白视频帧序列中相邻两帧间的正向和反向光流;从数据集中选择相邻的三帧输入至特征提取网络中提取出特征信息图;在目标图像中计算出每个像素点的相邻相似区域;将特征信息图输入初步着色网络得到每一帧图像的多张初步着色图;输入三帧的初步着色图输入光流对齐模块,利用时序损失函数来对初步着色网络进行约束;将初步着色网络的输出和光流对齐模块的输出输入至加强着色网络得到最终输出,并利用L-1范数计算最终输出和真实值的误差。