ColorUNet: A convolutional classification approach to colorization
This paper tackles the challenge of colorizing grayscale images. We take a deep convolutional neural network approach, and choose to take the angle of classification, working on a finite set of possible colors. Similarly to a recent paper, we implement a loss and a prediction function that favor realistic, colorful images rather than "true" ones. We show that a rather lightweight architecture inspired by the U-Net, and trained on a reasonable amount of pictures of landscapes, achieves satisfactory results on this specific subset of pictures. We show that data augmentation significantly improves the performance and robustness of the model, and provide visual analysis of the prediction confidence. We show an application of our model, extending the task to video colorization. We suggest a way to smooth color predictions across frames, without the need to train a recurrent network designed for sequential inputs.
Code (0)
등록된 구현이 없습니다.
Tasks
ClassificationColorizationData AugmentationGeneral ClassificationMethods 이 논문이 사용한 방법론
Similar Papers 제목 키워드 기반
Automatic Image Colorization with Convolutional Neural Networks and Generative Adversarial Networks
Image colorization, the task of adding colors to grayscale images, has been the focus of significant research efforts in computer vision in recent years for its various application areas such as color restoration and aut…
Image ColorizationAutomatic Video Colorization using 3D Conditional Generative Adversarial Networks
In this work, we present a method for automatic colorization of grayscale videos. The core of the method is a Generative Adversarial Network that is trained and tested on sequences of frames in a sliding window manner. N…
ColorizationGenerative Adversarial NetworkLet there be color!: joint end-to-end learning of global and local image priors for automatic image colorization with simultaneous classification
We present a novel technique to automatically colorize grayscale images that combines both global priors and local image features. Based on Convolutional Neural Networks, our deep network features a fusion layer that all…
ColorizationImage ColorizationScene ClassificationDetecting Colorized Images via Convolutional Neural Networks: Toward High Accuracy and Good Generalization
Image colorization achieves more and more realistic results with the increasing computation power of recent deep learning techniques. It becomes more difficult to identify the fake colorized images by human eyes. In this…
ColorizationGeneral ClassificationImage ColorizationSaliency Detection With Fully Convolutional Neural Network
Saliency detection is an important task in image processing as it can solve many problems and it usually is the first step in for other processes. Convolutional neural networks have been proved to be very effective on se…
ColorizationSaliency Detection