ChromaFusionNet (CFNet)

Official PyTorch Implementation of AAAI 2024 Paper "ChromaFusionNet (CFNet): Natural Fusion of Fine-Grained Color Editing"

Environment

Tested on Ubuntu 20.04 + CUDA 11.3

conda create -n cfnet python=3.9 -y
conda activate cfnet
pip install torch==1.12.1+cu113 torchvision==0.13.1+cu113 torchaudio==0.12.1 --extra-index-url https://download.pytorch.org/whl/cu113
pip install -r requirements.txt

Checkpoints

Please unzip the following zip file to project's main folder.

Google Drive

https://drive.google.com/file/d/1xK23C1k68lKrH0yeLvxKwr3MZ-71RvqM/view?usp=sharing

Baidu Disk

https://pan.baidu.com/s/1iMu2DNmiuTc9vSEafDrpLA?pwd=szz4
Password: szz4 

Inference

python cfnet_inference.py

Acknowledgments

This code repository is partially borrowed from iColoriT and ESRGAN

Citation

You can cite it as follows:

@inproceedings{dong2024chromafusionnet,
  title={ChromaFusionNet (CFNet): Natural Fusion of Fine-Grained Color Editing},
  author={Dong, Yi and Wang, Yuxi and Fan, Ruoxi and Ouyang, Wenqi and Shen, Zhiqi and Ren, Peiran and Xie, Xuansong},
  booktitle={Proceedings of the AAAI Conference on Artificial Intelligence},
  volume={38},
  number={2},
  pages={1591--1599},
  year={2024}
}