├── .gitignore ├── .pre-commit-config.yaml ├── LICENSE ├── README.md ├── assets ├── imagenet_val_label_map.txt └── teaser.png ├── config ├── __init__.py └── defaults.py ├── data ├── __init__.py └── imagenet_inference.py ├── main.py ├── model_zoo.md ├── models ├── __init__.py ├── biggan.py ├── biggan_layers.py ├── latent_deformator.py └── ref_based_model.py ├── ops ├── __init__.py ├── dcn │ ├── __init__.py │ ├── deform_conv.py │ └── src │ │ ├── deform_conv_cuda.cpp │ │ ├── deform_conv_cuda_kernel.cu │ │ └── deform_conv_ext.cpp └── setup.py ├── predict_imagenet_label.py ├── scripts └── generate_label_map.py ├── setup.cfg ├── solvers ├── __init__.py ├── base_solver.py └── refcolor_solver.py ├── testcase_diverse ├── test_diverse_inp0.png ├── test_diverse_inp1.png ├── test_diverse_inp2.png └── test_diverse_inp3.png ├── testcase_imagenet ├── ILSVRC2012_val_00000001.JPEG ├── ILSVRC2012_val_00000002.JPEG └── ILSVRC2012_val_00003014.JPEG ├── testcase_in_the_wild ├── testcase10_from_eccv2022_ct2.jpg ├── testcase11_from_eccv2022_ct2.jpg ├── testcase12_from_eccv2022_ct2.jpg ├── testcase13_from_eccv2022_ct2.jpg ├── testcase14_from_eccv2022_ct2.jpg ├── testcase15_classical_pic.jpg ├── testcase16_from_eccv2022_bigcolor.png ├── testcase17_from_eccv2022_bigcolor.png ├── testcase18_from_eccv2022_bigcolor.png ├── testcase19_from_eccv2022_bigcolor.png ├── testcase1_from_internet.jpeg ├── testcase2_from_internet.jpeg ├── testcase3_from_internet.jpeg ├── testcase4_from_internet.jpeg ├── testcase5_from_eccv2022_ct2.jpg ├── testcase6_from_eccv2022_ct2.jpg ├── testcase7_from_eccv2022_ct2.jpg ├── testcase8_from_eccv2022_ct2.jpg └── testcase9_from_eccv2022_ct2.jpg ├── utils ├── __init__.py ├── my_logging.py ├── parser.py └── util.py └── visual_results_in_the_wild.md /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ToTheBeginning/GCP-Colorization/HEAD/.gitignore -------------------------------------------------------------------------------- /.pre-commit-config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ToTheBeginning/GCP-Colorization/HEAD/.pre-commit-config.yaml -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ToTheBeginning/GCP-Colorization/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ToTheBeginning/GCP-Colorization/HEAD/README.md -------------------------------------------------------------------------------- /assets/imagenet_val_label_map.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ToTheBeginning/GCP-Colorization/HEAD/assets/imagenet_val_label_map.txt -------------------------------------------------------------------------------- /assets/teaser.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ToTheBeginning/GCP-Colorization/HEAD/assets/teaser.png -------------------------------------------------------------------------------- /config/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /config/defaults.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ToTheBeginning/GCP-Colorization/HEAD/config/defaults.py -------------------------------------------------------------------------------- /data/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ToTheBeginning/GCP-Colorization/HEAD/data/__init__.py -------------------------------------------------------------------------------- /data/imagenet_inference.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ToTheBeginning/GCP-Colorization/HEAD/data/imagenet_inference.py -------------------------------------------------------------------------------- /main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ToTheBeginning/GCP-Colorization/HEAD/main.py -------------------------------------------------------------------------------- /model_zoo.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ToTheBeginning/GCP-Colorization/HEAD/model_zoo.md -------------------------------------------------------------------------------- /models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ToTheBeginning/GCP-Colorization/HEAD/models/__init__.py -------------------------------------------------------------------------------- /models/biggan.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ToTheBeginning/GCP-Colorization/HEAD/models/biggan.py -------------------------------------------------------------------------------- /models/biggan_layers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ToTheBeginning/GCP-Colorization/HEAD/models/biggan_layers.py -------------------------------------------------------------------------------- /models/latent_deformator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ToTheBeginning/GCP-Colorization/HEAD/models/latent_deformator.py -------------------------------------------------------------------------------- /models/ref_based_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ToTheBeginning/GCP-Colorization/HEAD/models/ref_based_model.py -------------------------------------------------------------------------------- /ops/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /ops/dcn/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ToTheBeginning/GCP-Colorization/HEAD/ops/dcn/__init__.py -------------------------------------------------------------------------------- /ops/dcn/deform_conv.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ToTheBeginning/GCP-Colorization/HEAD/ops/dcn/deform_conv.py -------------------------------------------------------------------------------- /ops/dcn/src/deform_conv_cuda.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ToTheBeginning/GCP-Colorization/HEAD/ops/dcn/src/deform_conv_cuda.cpp -------------------------------------------------------------------------------- /ops/dcn/src/deform_conv_cuda_kernel.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ToTheBeginning/GCP-Colorization/HEAD/ops/dcn/src/deform_conv_cuda_kernel.cu -------------------------------------------------------------------------------- /ops/dcn/src/deform_conv_ext.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ToTheBeginning/GCP-Colorization/HEAD/ops/dcn/src/deform_conv_ext.cpp -------------------------------------------------------------------------------- /ops/setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ToTheBeginning/GCP-Colorization/HEAD/ops/setup.py -------------------------------------------------------------------------------- /predict_imagenet_label.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ToTheBeginning/GCP-Colorization/HEAD/predict_imagenet_label.py -------------------------------------------------------------------------------- /scripts/generate_label_map.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ToTheBeginning/GCP-Colorization/HEAD/scripts/generate_label_map.py -------------------------------------------------------------------------------- /setup.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ToTheBeginning/GCP-Colorization/HEAD/setup.cfg -------------------------------------------------------------------------------- /solvers/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /solvers/base_solver.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ToTheBeginning/GCP-Colorization/HEAD/solvers/base_solver.py -------------------------------------------------------------------------------- /solvers/refcolor_solver.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ToTheBeginning/GCP-Colorization/HEAD/solvers/refcolor_solver.py -------------------------------------------------------------------------------- /testcase_diverse/test_diverse_inp0.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ToTheBeginning/GCP-Colorization/HEAD/testcase_diverse/test_diverse_inp0.png -------------------------------------------------------------------------------- /testcase_diverse/test_diverse_inp1.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ToTheBeginning/GCP-Colorization/HEAD/testcase_diverse/test_diverse_inp1.png -------------------------------------------------------------------------------- /testcase_diverse/test_diverse_inp2.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ToTheBeginning/GCP-Colorization/HEAD/testcase_diverse/test_diverse_inp2.png -------------------------------------------------------------------------------- /testcase_diverse/test_diverse_inp3.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ToTheBeginning/GCP-Colorization/HEAD/testcase_diverse/test_diverse_inp3.png -------------------------------------------------------------------------------- /testcase_imagenet/ILSVRC2012_val_00000001.JPEG: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ToTheBeginning/GCP-Colorization/HEAD/testcase_imagenet/ILSVRC2012_val_00000001.JPEG -------------------------------------------------------------------------------- /testcase_imagenet/ILSVRC2012_val_00000002.JPEG: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ToTheBeginning/GCP-Colorization/HEAD/testcase_imagenet/ILSVRC2012_val_00000002.JPEG -------------------------------------------------------------------------------- /testcase_imagenet/ILSVRC2012_val_00003014.JPEG: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ToTheBeginning/GCP-Colorization/HEAD/testcase_imagenet/ILSVRC2012_val_00003014.JPEG -------------------------------------------------------------------------------- /testcase_in_the_wild/testcase10_from_eccv2022_ct2.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ToTheBeginning/GCP-Colorization/HEAD/testcase_in_the_wild/testcase10_from_eccv2022_ct2.jpg -------------------------------------------------------------------------------- /testcase_in_the_wild/testcase11_from_eccv2022_ct2.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ToTheBeginning/GCP-Colorization/HEAD/testcase_in_the_wild/testcase11_from_eccv2022_ct2.jpg -------------------------------------------------------------------------------- /testcase_in_the_wild/testcase12_from_eccv2022_ct2.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ToTheBeginning/GCP-Colorization/HEAD/testcase_in_the_wild/testcase12_from_eccv2022_ct2.jpg -------------------------------------------------------------------------------- /testcase_in_the_wild/testcase13_from_eccv2022_ct2.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ToTheBeginning/GCP-Colorization/HEAD/testcase_in_the_wild/testcase13_from_eccv2022_ct2.jpg -------------------------------------------------------------------------------- /testcase_in_the_wild/testcase14_from_eccv2022_ct2.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ToTheBeginning/GCP-Colorization/HEAD/testcase_in_the_wild/testcase14_from_eccv2022_ct2.jpg -------------------------------------------------------------------------------- /testcase_in_the_wild/testcase15_classical_pic.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ToTheBeginning/GCP-Colorization/HEAD/testcase_in_the_wild/testcase15_classical_pic.jpg -------------------------------------------------------------------------------- /testcase_in_the_wild/testcase16_from_eccv2022_bigcolor.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ToTheBeginning/GCP-Colorization/HEAD/testcase_in_the_wild/testcase16_from_eccv2022_bigcolor.png -------------------------------------------------------------------------------- /testcase_in_the_wild/testcase17_from_eccv2022_bigcolor.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ToTheBeginning/GCP-Colorization/HEAD/testcase_in_the_wild/testcase17_from_eccv2022_bigcolor.png -------------------------------------------------------------------------------- /testcase_in_the_wild/testcase18_from_eccv2022_bigcolor.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ToTheBeginning/GCP-Colorization/HEAD/testcase_in_the_wild/testcase18_from_eccv2022_bigcolor.png -------------------------------------------------------------------------------- /testcase_in_the_wild/testcase19_from_eccv2022_bigcolor.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ToTheBeginning/GCP-Colorization/HEAD/testcase_in_the_wild/testcase19_from_eccv2022_bigcolor.png -------------------------------------------------------------------------------- /testcase_in_the_wild/testcase1_from_internet.jpeg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ToTheBeginning/GCP-Colorization/HEAD/testcase_in_the_wild/testcase1_from_internet.jpeg -------------------------------------------------------------------------------- /testcase_in_the_wild/testcase2_from_internet.jpeg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ToTheBeginning/GCP-Colorization/HEAD/testcase_in_the_wild/testcase2_from_internet.jpeg -------------------------------------------------------------------------------- /testcase_in_the_wild/testcase3_from_internet.jpeg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ToTheBeginning/GCP-Colorization/HEAD/testcase_in_the_wild/testcase3_from_internet.jpeg -------------------------------------------------------------------------------- /testcase_in_the_wild/testcase4_from_internet.jpeg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ToTheBeginning/GCP-Colorization/HEAD/testcase_in_the_wild/testcase4_from_internet.jpeg -------------------------------------------------------------------------------- /testcase_in_the_wild/testcase5_from_eccv2022_ct2.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ToTheBeginning/GCP-Colorization/HEAD/testcase_in_the_wild/testcase5_from_eccv2022_ct2.jpg -------------------------------------------------------------------------------- /testcase_in_the_wild/testcase6_from_eccv2022_ct2.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ToTheBeginning/GCP-Colorization/HEAD/testcase_in_the_wild/testcase6_from_eccv2022_ct2.jpg -------------------------------------------------------------------------------- /testcase_in_the_wild/testcase7_from_eccv2022_ct2.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ToTheBeginning/GCP-Colorization/HEAD/testcase_in_the_wild/testcase7_from_eccv2022_ct2.jpg -------------------------------------------------------------------------------- /testcase_in_the_wild/testcase8_from_eccv2022_ct2.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ToTheBeginning/GCP-Colorization/HEAD/testcase_in_the_wild/testcase8_from_eccv2022_ct2.jpg -------------------------------------------------------------------------------- /testcase_in_the_wild/testcase9_from_eccv2022_ct2.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ToTheBeginning/GCP-Colorization/HEAD/testcase_in_the_wild/testcase9_from_eccv2022_ct2.jpg -------------------------------------------------------------------------------- /utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /utils/my_logging.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ToTheBeginning/GCP-Colorization/HEAD/utils/my_logging.py -------------------------------------------------------------------------------- /utils/parser.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ToTheBeginning/GCP-Colorization/HEAD/utils/parser.py -------------------------------------------------------------------------------- /utils/util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ToTheBeginning/GCP-Colorization/HEAD/utils/util.py -------------------------------------------------------------------------------- /visual_results_in_the_wild.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ToTheBeginning/GCP-Colorization/HEAD/visual_results_in_the_wild.md --------------------------------------------------------------------------------