Skip to content

kamse/Color_Transfer_Histogram_Analogy

 
 

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

12 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Deep Color Transfer using Histogram Analogy — Official PyTorch Implementation

Python 2.7.12 PyTorch 0.4.0 CUDA 8.0.61 License CC BY-NC

Teaser image Picture: Color transfer results on various source and reference image pairs. For visualization, the reference image is cropped to make a same size with other images.

This repository contains the official PyTorch implementation of the following paper:

Deep Color Transfer using Historgram Analogy
Junyong Lee (POSTECH), Hyeongseok Son (POSTECH), Gunhee Lee (NCSoft), Jonghyeop Lee (POSTECH), Sunghyun Cho (POSTECH) Seungyong Lee (POSTECH)

http://cg.postech.ac.kr/papers/2020_CGI_JY.pdf

Abstract: We propose a novel approach to transferring the color of a reference image to a given source image. Although there can be diverse pairs of source and reference images in terms of content and composition similarity,previous methods are not capable of covering the whole diversity. To resolve this limitation, we propose a deep neural network that leverages color histogram analogy for color transfer. A histogram contains essential color information of an image, and our network utilizes the analogy between the source and reference histograms to modulate the color of the source image with abstract color features of the reference image. In our approach, histogram analogy is exploited basically among the whole images, but it can also be applied to semantically corresponding regions in the case that the source and reference images have similar contents with different compositions. Experimental results show that our approach effectively transfers the reference colors to the source images in a variety of settings. We also demonstrate a few applications of our approach, such asalette-based recolorization, color enhancement, and color editing.

For any inquiries, please contact junyonglee@postech.ac.kr

Resources

All material related to our paper is available via the following links:

Link
Paper PDF
Supplementary Files
Checkpoint Files

Testing the network

  1. Download pretrained weights for IRN and HEN from here. Then, place the checkpoint under ./checkpoints (one may change the offset in ./options/base_options.py).

  2. To test the network, type

python test.py --dataroot [test folder path] --checkpoints_dir [ckpt path]

Using pre-trained networks

BIBTEX

If you find this code useful, please consider citing:

@InProceedings{Lee_2020_VC,
author = {Lee, Junyong and Son, Hyeongseok and Lee, Gunhee and Lee, Jonghyeop and Cho, Sunghyun and Lee, Seungyong},
title = {Deep Color Transfer using Histogram Analogy},
booktitle = {The Visual Computer},
month = {Aug},
year = {2020}
}

License

This software is being made available under the terms in the LICENSE file.

Any exemptions to these terms requires a license from the Pohang University of Science and Technology.

About Coupe Project

Project ‘COUPE’ aims to develop software that evaluates and improves the quality of images and videos based on big visual data. To achieve the goal, we extract sharpness, color, composition features from images and develop technologies for restoring and improving by using it. In addition,ersonalization technology through userreference analysis is under study.

Please checkout out other Coupe repositories in our Posgraph github organization.

Useful Links

About

No description, website, or topics provided.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • Python 100.0%