🚀 Want to run this model with an API? Get started

cjwbw/bigcolor

Public
Colorization using a Generative Color Prior for Natural Images
21.2K runs

Run time and cost

Predictions run on Nvidia T4 GPU hardware. Predictions typically complete within 3 seconds.

This is a cog implementation of https://github.com/KIMGEONUNG/BigColor

BigColor: Colorization using a Generative Color Prior for Natural Images
Official PyTorch Implementation of the ECCV 2022 Paper

Teaser image 1

BigColor: Colorization using a Generative Color Prior for Natural Images

Geonung Kim, Kyoungkook Kang, Seongtae Kim, Hwayoon Lee, Sehoon Kim, Jonghyun Kim, Seung-Hwan Baek, Sunghyun Cho

[Paper]
[Supple]
[Project Page]

Abstract: For realistic and vivid colorization, generative priors have recently been exploited. However, such generative priors often fail for in-the-wild complex images due to their limited representation space. In this paper, we propose BigColor, a novel colorization approach that provides vivid colorization for diverse in-the-wild images with complex structures. While previous generative priors are trained to synthesize both image structures and colors, we learn a generative color prior to focus on color synthesis given the spatial structure of an image. In this way, we reduce the burden of synthesizing image structures from the generative prior and expand its representation space to cover diverse images. To this end, we propose a BigGAN-inspired encoder-generator network that uses a spatial feature map instead of a spatially-flattened BigGAN latent code, resulting in an enlarged representation space. Our method enables robust colorization for diverse inputs in a single forward pass, supports arbitrary input resolutions, and provides multi-modal colorization results. We demonstrate that BigColor significantly outperforms existing methods especially on in-the-wild images with complex structures.

Acknowledgement

The code of pretrained-BigGAN generator and discriminator and checkpoints are borrowed from
a research: DGP.

Citation

@inproceedings{Kim2022BigColor,
  title     = {BigColor: Colorization using a Generative Color Prior for Natural Images},
  author    = {Geonung Kim,Kyoungkook Kang,Seongtae Kim,Hwayoon Lee,Sehoon Kim,Jonghyun Kim,Seung-Hwan Baek,Sunghyun Cho},
  booktitle = {European Conference on Computer Vision (ECCV)},
  year      = {2022}
}