Image Colorization By Capsule Networks
In this paper, a simple topology of Capsule Network (CapsNet) is investigated for the problem of image colorization. The generative and segmentation capabilities of the original CapsNet topology, which is proposed for image classification problem, is leveraged for the colorization of the images by modifying the network as follows:1) The original CapsNet model is adapted to map the grayscale input to the output in the CIE Lab colorspace, 2) The feature detector part of the model is updated by using deeper feature layers inherited from VGG-19 pre-trained model with weights in order to transfer low-level image representation capability to this model, 3) The margin loss function is modified as Mean Squared Error (MSE) loss to minimize the image-to-imagemapping. The resulting CapsNet model is named as Colorizer Capsule Network (ColorCapsNet).The performance of the ColorCapsNet is evaluated on the DIV2K dataset and promising results are obtained to investigate Capsule Networks further for image colorization problem.
Code (0)
등록된 구현이 없습니다.
Tasks
Colorizationimage-classificationImage ClassificationImage ColorizationMethods 이 논문이 사용한 방법론
Similar Papers 제목 키워드 기반
Two Decades of Colorization and Decolorization for Images and Videos
Colorization is a computer-aided process, which aims to give color to a gray image or video. It can be used to enhance black-and-white images, including black-and-white photos, old-fashioned films, and scientific imaging…
ColorizationImage EnhancementImage SegmentationSemantic Segmentation+1Comicolorization: Semi-Automatic Manga Colorization
We developed "Comicolorization", a semi-automatic colorization system for manga images. Given a monochrome manga and reference images as inputs, our system generates a plausible color version of the manga. This is the fi…
ColorizationLearning Diverse Image Colorization
Colorization is an ambiguous problem, with multiple viable colorizations for a single grey-level image. However, previous methods only produce the single most probable colorization. Our goal is to model the diversity int…
ColorizationDecoderDiversityGenerative Adversarial Network+1Style-Structure Disentangled Features and Normalizing Flows for Diverse Icon Colorization
In this study, we present a colorization network that generates flat-color icons according to given sketches and semantic colorization styles. Specifically, our network contains a style-structure disentangled coloriz…
ColorizationDecoderDiversityImage-to-Image TranslationImage Colorization: A Survey and Dataset
Image colorization estimates RGB colors for grayscale images or video frames to improve their aesthetic and perceptual quality. Over the last decade, deep learning techniques for image colorization have significantly pro…
BenchmarkingColorizationImage ColorizationSurvey