Coloring with limited data: Few-shot colorization via memory augmented networks

Seungjoo Yoo, Hyojin Bahng, Sunghyo Chung, Junsoo Lee, Jaehyuk Chang, Jaegul Choo

Research output: Chapter in Book/Report/Conference proceedingConference contribution

116 Citations (Scopus)

Abstract

Despite recent advancements in deep learning-based automatic colorization, they are still limited when it comes to few-shot learning. Existing models require a significant amount of training data. To tackle this issue, we present a novel memory-augmented colorization model MemoPainter that can produce high-quality colorization with limited data. In particular, our model is able to capture rare instances and successfully colorize them. Also, we propose a novel threshold triplet loss that enables unsupervised training of memory networks without the need for class labels. Experiments show that our model has superior quality in both few-shot and one-shot colorization tasks.

Original languageEnglish
Title of host publicationProceedings - 2019 IEEE/CVF Conference on Computer Vision and Pattern Recognition, CVPR 2019
PublisherIEEE Computer Society
Pages11275-11284
Number of pages10
ISBN (Electronic)9781728132938
DOIs
Publication statusPublished - 2019 Jun
Event32nd IEEE/CVF Conference on Computer Vision and Pattern Recognition, CVPR 2019 - Long Beach, United States
Duration: 2019 Jun 162019 Jun 20

Publication series

NameProceedings of the IEEE Computer Society Conference on Computer Vision and Pattern Recognition
Volume2019-June
ISSN (Print)1063-6919

Conference

Conference32nd IEEE/CVF Conference on Computer Vision and Pattern Recognition, CVPR 2019
Country/TerritoryUnited States
CityLong Beach
Period19/6/1619/6/20

Bibliographical note

Funding Information:
This work was partially supported by the National Research Foundation of Korea (NRF) grant funded by the Korean government (MSIP) (No. NRF2016R1C1B2015924).

Funding Information:
Acknowledgements. This work was partially supported bytheNationalResearchFoundationofKorea (NRF) grantfundedbytheKorean government(MSIP)(No. NRF2016R1C1B2015924). We thankall researchers at NAVER WEBTOONCorp.,especiallySungminKang. Jaegul Choo is the corresponding author.

Publisher Copyright:
© 2019 IEEE.

Keywords

  • Computational Photography
  • Computer Vision Theory
  • Deep Learning
  • Image and Video Synthesis
  • Vi
  • Vision + Graphics

ASJC Scopus subject areas

  • Software
  • Computer Vision and Pattern Recognition

Fingerprint

Dive into the research topics of 'Coloring with limited data: Few-shot colorization via memory augmented networks'. Together they form a unique fingerprint.

Cite this