README.md

AlColor

AIColor is our code for NTIRE 天池大数据比赛AI着色 CVPR2023

Model

we propose the Mult DDIM on HourGlass model to Enhance image color

Build different Unet models for different time steps to accelerate the model inference and training process.

Train

#TODO

Results

CDC Score

FID Score

Here is our result in tianchi rank board result and we got 11/93 adn 10/130 rank in FID and CDC

we test different model to generate image after colored

DF-gan tiny result:

DF-gan middle result:

DF-Unet-Gan result

DF-HourGlass-Gan result