Automatic video colorization based on contrastive learning and optical flow
Shuyu Xiao,Yongfang Wang,Yihan Wang
DOI: https://doi.org/10.1007/s11042-023-17883-5
IF: 2.577
2024-01-03
Multimedia Tools and Applications
Abstract:Video colorization is the process of assigning reasonable color values to gray videos. Most of the existing video colorization methods use an image colorization algorithm directly regardless of correlation between frames in videos. Besides, these methods need reference, but it is difficult to obtain the guidance of users or reference images in practical applications. To overcome this limitation, we propose an automatic video colorization based on contrastive learning and optical flow. Contrastive learning optimizes the temporal correlation loss between the feature vectors of the gray video frame patches and the generated color video frame patches during training, which ensures the consistency of image content at the patch level. The optical flow module maintains the temporal consistency of colored videos by minimizing temporal correlation loss. In particular, the perceptual loss is introduced to improve the subjective colorization effect. Experiments show that the proposed algorithm outperforms the existing methods in terms of subjective quality. To reproduce the results, we release the source code at: https://github.com/orange-416/CLAC-GAN.
computer science, information systems, theory & methods,engineering, electrical & electronic, software engineering