├── .gitignore ├── LICENSE ├── README.md ├── ckpts └── .gitignore ├── colorize.py ├── colorize_multi_c.py ├── colorize_multi_z.py ├── colorize_real.py ├── download-bigcolor.sh ├── download-pretrained.sh ├── environment.yml ├── images_gray ├── anold.jpg ├── ansel_adams3.jpg ├── chaplin_1.jpg ├── chaplin_2.jpg ├── chaplin_3.jpg ├── chaplin_4.jpg ├── corbusier_1.jpg ├── corbusier_2.jpg ├── einstein_1.jpg ├── einstein_2.jpg ├── einstein_3.jpg ├── elvis_presley_1.jpg ├── elvis_presley_2.jpg ├── kiss.jpg ├── marilyn_monroe_1.jpg ├── marilyn_monroe_2.jpg ├── marilyn_monroe_3.jpg ├── marilyn_monroe_4.jpg ├── migrant_mother_1.jpeg ├── migrant_mother_2.jpeg ├── out_door.jpg ├── toyota_camry.jpg ├── unknown_4.jpg ├── unknown_d.png ├── vanderohe_1.jpg ├── vanderohe_2.png ├── war_1.jpg └── war_2.jpg ├── images_multi_c └── ILSVRC2012_val_00002071.JPEG ├── images_multi_z ├── 03830.jpg ├── 16191.jpg ├── 16358.jpg ├── 20557.jpg ├── 20564.jpg ├── 21273.jpg ├── 26932.jpg ├── 32940.jpg ├── 35906.jpg └── 45630.jpg ├── models ├── __init__.py ├── biggan.py ├── common.py ├── encoders.py └── layers.py ├── pretrained └── .gitignore ├── scripts ├── colorize.multi_c.sh ├── colorize.multi_z.sh ├── colorize.real.sh ├── infer.bigcolor.e011.sh ├── train_v1.0.0.sh └── train_v1.0.1.sh ├── srcs ├── cplx_img_ids.txt ├── list_low_colorful.csv ├── teaser_1.png └── teaser_2.png ├── train.py └── utils ├── __init__.py ├── biggan_utils.py ├── color_models.py ├── common_utils.py ├── data.py ├── distributed_utils.py ├── logger.py └── losses.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/KIMGEONUNG/BigColor/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/KIMGEONUNG/BigColor/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/KIMGEONUNG/BigColor/HEAD/README.md -------------------------------------------------------------------------------- /ckpts/.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/KIMGEONUNG/BigColor/HEAD/ckpts/.gitignore -------------------------------------------------------------------------------- /colorize.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/KIMGEONUNG/BigColor/HEAD/colorize.py -------------------------------------------------------------------------------- /colorize_multi_c.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/KIMGEONUNG/BigColor/HEAD/colorize_multi_c.py -------------------------------------------------------------------------------- /colorize_multi_z.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/KIMGEONUNG/BigColor/HEAD/colorize_multi_z.py -------------------------------------------------------------------------------- /colorize_real.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/KIMGEONUNG/BigColor/HEAD/colorize_real.py -------------------------------------------------------------------------------- /download-bigcolor.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/KIMGEONUNG/BigColor/HEAD/download-bigcolor.sh -------------------------------------------------------------------------------- /download-pretrained.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/KIMGEONUNG/BigColor/HEAD/download-pretrained.sh -------------------------------------------------------------------------------- /environment.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/KIMGEONUNG/BigColor/HEAD/environment.yml -------------------------------------------------------------------------------- /images_gray/anold.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/KIMGEONUNG/BigColor/HEAD/images_gray/anold.jpg -------------------------------------------------------------------------------- /images_gray/ansel_adams3.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/KIMGEONUNG/BigColor/HEAD/images_gray/ansel_adams3.jpg -------------------------------------------------------------------------------- /images_gray/chaplin_1.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/KIMGEONUNG/BigColor/HEAD/images_gray/chaplin_1.jpg -------------------------------------------------------------------------------- /images_gray/chaplin_2.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/KIMGEONUNG/BigColor/HEAD/images_gray/chaplin_2.jpg -------------------------------------------------------------------------------- /images_gray/chaplin_3.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/KIMGEONUNG/BigColor/HEAD/images_gray/chaplin_3.jpg -------------------------------------------------------------------------------- /images_gray/chaplin_4.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/KIMGEONUNG/BigColor/HEAD/images_gray/chaplin_4.jpg -------------------------------------------------------------------------------- /images_gray/corbusier_1.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/KIMGEONUNG/BigColor/HEAD/images_gray/corbusier_1.jpg -------------------------------------------------------------------------------- /images_gray/corbusier_2.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/KIMGEONUNG/BigColor/HEAD/images_gray/corbusier_2.jpg -------------------------------------------------------------------------------- /images_gray/einstein_1.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/KIMGEONUNG/BigColor/HEAD/images_gray/einstein_1.jpg -------------------------------------------------------------------------------- /images_gray/einstein_2.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/KIMGEONUNG/BigColor/HEAD/images_gray/einstein_2.jpg -------------------------------------------------------------------------------- /images_gray/einstein_3.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/KIMGEONUNG/BigColor/HEAD/images_gray/einstein_3.jpg -------------------------------------------------------------------------------- /images_gray/elvis_presley_1.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/KIMGEONUNG/BigColor/HEAD/images_gray/elvis_presley_1.jpg -------------------------------------------------------------------------------- /images_gray/elvis_presley_2.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/KIMGEONUNG/BigColor/HEAD/images_gray/elvis_presley_2.jpg -------------------------------------------------------------------------------- /images_gray/kiss.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/KIMGEONUNG/BigColor/HEAD/images_gray/kiss.jpg -------------------------------------------------------------------------------- /images_gray/marilyn_monroe_1.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/KIMGEONUNG/BigColor/HEAD/images_gray/marilyn_monroe_1.jpg -------------------------------------------------------------------------------- /images_gray/marilyn_monroe_2.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/KIMGEONUNG/BigColor/HEAD/images_gray/marilyn_monroe_2.jpg -------------------------------------------------------------------------------- /images_gray/marilyn_monroe_3.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/KIMGEONUNG/BigColor/HEAD/images_gray/marilyn_monroe_3.jpg -------------------------------------------------------------------------------- /images_gray/marilyn_monroe_4.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/KIMGEONUNG/BigColor/HEAD/images_gray/marilyn_monroe_4.jpg -------------------------------------------------------------------------------- /images_gray/migrant_mother_1.jpeg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/KIMGEONUNG/BigColor/HEAD/images_gray/migrant_mother_1.jpeg -------------------------------------------------------------------------------- /images_gray/migrant_mother_2.jpeg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/KIMGEONUNG/BigColor/HEAD/images_gray/migrant_mother_2.jpeg -------------------------------------------------------------------------------- /images_gray/out_door.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/KIMGEONUNG/BigColor/HEAD/images_gray/out_door.jpg -------------------------------------------------------------------------------- /images_gray/toyota_camry.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/KIMGEONUNG/BigColor/HEAD/images_gray/toyota_camry.jpg -------------------------------------------------------------------------------- /images_gray/unknown_4.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/KIMGEONUNG/BigColor/HEAD/images_gray/unknown_4.jpg -------------------------------------------------------------------------------- /images_gray/unknown_d.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/KIMGEONUNG/BigColor/HEAD/images_gray/unknown_d.png -------------------------------------------------------------------------------- /images_gray/vanderohe_1.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/KIMGEONUNG/BigColor/HEAD/images_gray/vanderohe_1.jpg -------------------------------------------------------------------------------- /images_gray/vanderohe_2.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/KIMGEONUNG/BigColor/HEAD/images_gray/vanderohe_2.png -------------------------------------------------------------------------------- /images_gray/war_1.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/KIMGEONUNG/BigColor/HEAD/images_gray/war_1.jpg -------------------------------------------------------------------------------- /images_gray/war_2.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/KIMGEONUNG/BigColor/HEAD/images_gray/war_2.jpg -------------------------------------------------------------------------------- /images_multi_c/ILSVRC2012_val_00002071.JPEG: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/KIMGEONUNG/BigColor/HEAD/images_multi_c/ILSVRC2012_val_00002071.JPEG -------------------------------------------------------------------------------- /images_multi_z/03830.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/KIMGEONUNG/BigColor/HEAD/images_multi_z/03830.jpg -------------------------------------------------------------------------------- /images_multi_z/16191.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/KIMGEONUNG/BigColor/HEAD/images_multi_z/16191.jpg -------------------------------------------------------------------------------- /images_multi_z/16358.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/KIMGEONUNG/BigColor/HEAD/images_multi_z/16358.jpg -------------------------------------------------------------------------------- /images_multi_z/20557.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/KIMGEONUNG/BigColor/HEAD/images_multi_z/20557.jpg -------------------------------------------------------------------------------- /images_multi_z/20564.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/KIMGEONUNG/BigColor/HEAD/images_multi_z/20564.jpg -------------------------------------------------------------------------------- /images_multi_z/21273.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/KIMGEONUNG/BigColor/HEAD/images_multi_z/21273.jpg -------------------------------------------------------------------------------- /images_multi_z/26932.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/KIMGEONUNG/BigColor/HEAD/images_multi_z/26932.jpg -------------------------------------------------------------------------------- /images_multi_z/32940.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/KIMGEONUNG/BigColor/HEAD/images_multi_z/32940.jpg -------------------------------------------------------------------------------- /images_multi_z/35906.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/KIMGEONUNG/BigColor/HEAD/images_multi_z/35906.jpg -------------------------------------------------------------------------------- /images_multi_z/45630.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/KIMGEONUNG/BigColor/HEAD/images_multi_z/45630.jpg -------------------------------------------------------------------------------- /models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/KIMGEONUNG/BigColor/HEAD/models/__init__.py -------------------------------------------------------------------------------- /models/biggan.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/KIMGEONUNG/BigColor/HEAD/models/biggan.py -------------------------------------------------------------------------------- /models/common.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/KIMGEONUNG/BigColor/HEAD/models/common.py -------------------------------------------------------------------------------- /models/encoders.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/KIMGEONUNG/BigColor/HEAD/models/encoders.py -------------------------------------------------------------------------------- /models/layers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/KIMGEONUNG/BigColor/HEAD/models/layers.py -------------------------------------------------------------------------------- /pretrained/.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/KIMGEONUNG/BigColor/HEAD/pretrained/.gitignore -------------------------------------------------------------------------------- /scripts/colorize.multi_c.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/KIMGEONUNG/BigColor/HEAD/scripts/colorize.multi_c.sh -------------------------------------------------------------------------------- /scripts/colorize.multi_z.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/KIMGEONUNG/BigColor/HEAD/scripts/colorize.multi_z.sh -------------------------------------------------------------------------------- /scripts/colorize.real.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/KIMGEONUNG/BigColor/HEAD/scripts/colorize.real.sh -------------------------------------------------------------------------------- /scripts/infer.bigcolor.e011.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/KIMGEONUNG/BigColor/HEAD/scripts/infer.bigcolor.e011.sh -------------------------------------------------------------------------------- /scripts/train_v1.0.0.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/KIMGEONUNG/BigColor/HEAD/scripts/train_v1.0.0.sh -------------------------------------------------------------------------------- /scripts/train_v1.0.1.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/KIMGEONUNG/BigColor/HEAD/scripts/train_v1.0.1.sh -------------------------------------------------------------------------------- /srcs/cplx_img_ids.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/KIMGEONUNG/BigColor/HEAD/srcs/cplx_img_ids.txt -------------------------------------------------------------------------------- /srcs/list_low_colorful.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/KIMGEONUNG/BigColor/HEAD/srcs/list_low_colorful.csv -------------------------------------------------------------------------------- /srcs/teaser_1.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/KIMGEONUNG/BigColor/HEAD/srcs/teaser_1.png -------------------------------------------------------------------------------- /srcs/teaser_2.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/KIMGEONUNG/BigColor/HEAD/srcs/teaser_2.png -------------------------------------------------------------------------------- /train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/KIMGEONUNG/BigColor/HEAD/train.py -------------------------------------------------------------------------------- /utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/KIMGEONUNG/BigColor/HEAD/utils/__init__.py -------------------------------------------------------------------------------- /utils/biggan_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/KIMGEONUNG/BigColor/HEAD/utils/biggan_utils.py -------------------------------------------------------------------------------- /utils/color_models.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/KIMGEONUNG/BigColor/HEAD/utils/color_models.py -------------------------------------------------------------------------------- /utils/common_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/KIMGEONUNG/BigColor/HEAD/utils/common_utils.py -------------------------------------------------------------------------------- /utils/data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/KIMGEONUNG/BigColor/HEAD/utils/data.py -------------------------------------------------------------------------------- /utils/distributed_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/KIMGEONUNG/BigColor/HEAD/utils/distributed_utils.py -------------------------------------------------------------------------------- /utils/logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/KIMGEONUNG/BigColor/HEAD/utils/logger.py -------------------------------------------------------------------------------- /utils/losses.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/KIMGEONUNG/BigColor/HEAD/utils/losses.py --------------------------------------------------------------------------------