├── AttentionBlock.py ├── AttentionBlockConditional.py ├── ColTranColorUpsampler.py ├── ColTranCore.py ├── ColTranSpatialUpsampler.py ├── GrayscaleEncoder.py ├── InnerDecoder.py ├── Main.py ├── OuterDecoder.py ├── README.md ├── coltran.png ├── dataset ├── tiny_imagenet_256 │ ├── 187922_5d1e159f39.jpg │ ├── 196860_1a9c864110.jpg │ ├── 202290_804be7ec8f.jpg │ ├── 235344_49d89a0ee6.jpg │ ├── 245481_d974325334.jpg │ ├── 250459_5995a47cef.jpg │ ├── 266774_cec9f5ec45.jpg │ ├── 267004_4eef6e9577.jpg │ ├── 267530_4cec93ad32.jpg │ └── 64986_fc645b3742.jpg ├── tiny_imagenet_256_g │ ├── 187922_5d1e159f39.jpg │ ├── 196860_1a9c864110.jpg │ ├── 202290_804be7ec8f.jpg │ ├── 235344_49d89a0ee6.jpg │ ├── 245481_d974325334.jpg │ ├── 250459_5995a47cef.jpg │ ├── 266774_cec9f5ec45.jpg │ ├── 267004_4eef6e9577.jpg │ ├── 267530_4cec93ad32.jpg │ └── 64986_fc645b3742.jpg ├── tiny_imagenet_64 │ ├── 187922_5d1e159f39.jpg │ ├── 196860_1a9c864110.jpg │ ├── 202290_804be7ec8f.jpg │ ├── 235344_49d89a0ee6.jpg │ ├── 245481_d974325334.jpg │ ├── 250459_5995a47cef.jpg │ ├── 266774_cec9f5ec45.jpg │ ├── 267004_4eef6e9577.jpg │ ├── 267530_4cec93ad32.jpg │ └── 64986_fc645b3742.jpg ├── tiny_imagenet_64_c │ ├── 187922_5d1e159f39.jpg │ ├── 196860_1a9c864110.jpg │ ├── 202290_804be7ec8f.jpg │ ├── 235344_49d89a0ee6.jpg │ ├── 245481_d974325334.jpg │ ├── 250459_5995a47cef.jpg │ ├── 266774_cec9f5ec45.jpg │ ├── 267004_4eef6e9577.jpg │ ├── 267530_4cec93ad32.jpg │ └── 64986_fc645b3742.jpg └── tiny_imagenet_64_g │ ├── 187922_5d1e159f39.jpg │ ├── 196860_1a9c864110.jpg │ ├── 202290_804be7ec8f.jpg │ ├── 235344_49d89a0ee6.jpg │ ├── 245481_d974325334.jpg │ ├── 250459_5995a47cef.jpg │ ├── 266774_cec9f5ec45.jpg │ ├── 267004_4eef6e9577.jpg │ ├── 267530_4cec93ad32.jpg │ └── 64986_fc645b3742.jpg ├── plot ├── Loss Cross-entropy [ColTran CORE].png ├── Loss Cross-entropy [ColTran UPSAMPLER COLOR].png └── Loss Cross-entropy [ColTran UPSAMPLER SPATIAL].png ├── test ├── xg_0.png ├── xg_1.png ├── xg_2.png ├── xg_pred_0.png ├── xg_pred_1.png └── xg_pred_2.png └── utils.py /AttentionBlock.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dorian25/colorization-transformer-experimental/HEAD/AttentionBlock.py -------------------------------------------------------------------------------- /AttentionBlockConditional.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dorian25/colorization-transformer-experimental/HEAD/AttentionBlockConditional.py -------------------------------------------------------------------------------- /ColTranColorUpsampler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dorian25/colorization-transformer-experimental/HEAD/ColTranColorUpsampler.py -------------------------------------------------------------------------------- /ColTranCore.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dorian25/colorization-transformer-experimental/HEAD/ColTranCore.py -------------------------------------------------------------------------------- /ColTranSpatialUpsampler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dorian25/colorization-transformer-experimental/HEAD/ColTranSpatialUpsampler.py -------------------------------------------------------------------------------- /GrayscaleEncoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dorian25/colorization-transformer-experimental/HEAD/GrayscaleEncoder.py -------------------------------------------------------------------------------- /InnerDecoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dorian25/colorization-transformer-experimental/HEAD/InnerDecoder.py -------------------------------------------------------------------------------- /Main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dorian25/colorization-transformer-experimental/HEAD/Main.py -------------------------------------------------------------------------------- /OuterDecoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dorian25/colorization-transformer-experimental/HEAD/OuterDecoder.py -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dorian25/colorization-transformer-experimental/HEAD/README.md -------------------------------------------------------------------------------- /coltran.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dorian25/colorization-transformer-experimental/HEAD/coltran.png -------------------------------------------------------------------------------- /dataset/tiny_imagenet_256/187922_5d1e159f39.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dorian25/colorization-transformer-experimental/HEAD/dataset/tiny_imagenet_256/187922_5d1e159f39.jpg -------------------------------------------------------------------------------- /dataset/tiny_imagenet_256/196860_1a9c864110.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dorian25/colorization-transformer-experimental/HEAD/dataset/tiny_imagenet_256/196860_1a9c864110.jpg -------------------------------------------------------------------------------- /dataset/tiny_imagenet_256/202290_804be7ec8f.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dorian25/colorization-transformer-experimental/HEAD/dataset/tiny_imagenet_256/202290_804be7ec8f.jpg -------------------------------------------------------------------------------- /dataset/tiny_imagenet_256/235344_49d89a0ee6.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dorian25/colorization-transformer-experimental/HEAD/dataset/tiny_imagenet_256/235344_49d89a0ee6.jpg -------------------------------------------------------------------------------- /dataset/tiny_imagenet_256/245481_d974325334.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dorian25/colorization-transformer-experimental/HEAD/dataset/tiny_imagenet_256/245481_d974325334.jpg -------------------------------------------------------------------------------- /dataset/tiny_imagenet_256/250459_5995a47cef.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dorian25/colorization-transformer-experimental/HEAD/dataset/tiny_imagenet_256/250459_5995a47cef.jpg -------------------------------------------------------------------------------- /dataset/tiny_imagenet_256/266774_cec9f5ec45.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dorian25/colorization-transformer-experimental/HEAD/dataset/tiny_imagenet_256/266774_cec9f5ec45.jpg -------------------------------------------------------------------------------- /dataset/tiny_imagenet_256/267004_4eef6e9577.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dorian25/colorization-transformer-experimental/HEAD/dataset/tiny_imagenet_256/267004_4eef6e9577.jpg -------------------------------------------------------------------------------- /dataset/tiny_imagenet_256/267530_4cec93ad32.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dorian25/colorization-transformer-experimental/HEAD/dataset/tiny_imagenet_256/267530_4cec93ad32.jpg -------------------------------------------------------------------------------- /dataset/tiny_imagenet_256/64986_fc645b3742.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dorian25/colorization-transformer-experimental/HEAD/dataset/tiny_imagenet_256/64986_fc645b3742.jpg -------------------------------------------------------------------------------- /dataset/tiny_imagenet_256_g/187922_5d1e159f39.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dorian25/colorization-transformer-experimental/HEAD/dataset/tiny_imagenet_256_g/187922_5d1e159f39.jpg -------------------------------------------------------------------------------- /dataset/tiny_imagenet_256_g/196860_1a9c864110.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dorian25/colorization-transformer-experimental/HEAD/dataset/tiny_imagenet_256_g/196860_1a9c864110.jpg -------------------------------------------------------------------------------- /dataset/tiny_imagenet_256_g/202290_804be7ec8f.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dorian25/colorization-transformer-experimental/HEAD/dataset/tiny_imagenet_256_g/202290_804be7ec8f.jpg -------------------------------------------------------------------------------- /dataset/tiny_imagenet_256_g/235344_49d89a0ee6.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dorian25/colorization-transformer-experimental/HEAD/dataset/tiny_imagenet_256_g/235344_49d89a0ee6.jpg -------------------------------------------------------------------------------- /dataset/tiny_imagenet_256_g/245481_d974325334.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dorian25/colorization-transformer-experimental/HEAD/dataset/tiny_imagenet_256_g/245481_d974325334.jpg -------------------------------------------------------------------------------- /dataset/tiny_imagenet_256_g/250459_5995a47cef.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dorian25/colorization-transformer-experimental/HEAD/dataset/tiny_imagenet_256_g/250459_5995a47cef.jpg -------------------------------------------------------------------------------- /dataset/tiny_imagenet_256_g/266774_cec9f5ec45.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dorian25/colorization-transformer-experimental/HEAD/dataset/tiny_imagenet_256_g/266774_cec9f5ec45.jpg -------------------------------------------------------------------------------- /dataset/tiny_imagenet_256_g/267004_4eef6e9577.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dorian25/colorization-transformer-experimental/HEAD/dataset/tiny_imagenet_256_g/267004_4eef6e9577.jpg -------------------------------------------------------------------------------- /dataset/tiny_imagenet_256_g/267530_4cec93ad32.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dorian25/colorization-transformer-experimental/HEAD/dataset/tiny_imagenet_256_g/267530_4cec93ad32.jpg -------------------------------------------------------------------------------- /dataset/tiny_imagenet_256_g/64986_fc645b3742.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dorian25/colorization-transformer-experimental/HEAD/dataset/tiny_imagenet_256_g/64986_fc645b3742.jpg -------------------------------------------------------------------------------- /dataset/tiny_imagenet_64/187922_5d1e159f39.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dorian25/colorization-transformer-experimental/HEAD/dataset/tiny_imagenet_64/187922_5d1e159f39.jpg -------------------------------------------------------------------------------- /dataset/tiny_imagenet_64/196860_1a9c864110.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dorian25/colorization-transformer-experimental/HEAD/dataset/tiny_imagenet_64/196860_1a9c864110.jpg -------------------------------------------------------------------------------- /dataset/tiny_imagenet_64/202290_804be7ec8f.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dorian25/colorization-transformer-experimental/HEAD/dataset/tiny_imagenet_64/202290_804be7ec8f.jpg -------------------------------------------------------------------------------- /dataset/tiny_imagenet_64/235344_49d89a0ee6.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dorian25/colorization-transformer-experimental/HEAD/dataset/tiny_imagenet_64/235344_49d89a0ee6.jpg -------------------------------------------------------------------------------- /dataset/tiny_imagenet_64/245481_d974325334.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dorian25/colorization-transformer-experimental/HEAD/dataset/tiny_imagenet_64/245481_d974325334.jpg -------------------------------------------------------------------------------- /dataset/tiny_imagenet_64/250459_5995a47cef.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dorian25/colorization-transformer-experimental/HEAD/dataset/tiny_imagenet_64/250459_5995a47cef.jpg -------------------------------------------------------------------------------- /dataset/tiny_imagenet_64/266774_cec9f5ec45.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dorian25/colorization-transformer-experimental/HEAD/dataset/tiny_imagenet_64/266774_cec9f5ec45.jpg -------------------------------------------------------------------------------- /dataset/tiny_imagenet_64/267004_4eef6e9577.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dorian25/colorization-transformer-experimental/HEAD/dataset/tiny_imagenet_64/267004_4eef6e9577.jpg -------------------------------------------------------------------------------- /dataset/tiny_imagenet_64/267530_4cec93ad32.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dorian25/colorization-transformer-experimental/HEAD/dataset/tiny_imagenet_64/267530_4cec93ad32.jpg -------------------------------------------------------------------------------- /dataset/tiny_imagenet_64/64986_fc645b3742.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dorian25/colorization-transformer-experimental/HEAD/dataset/tiny_imagenet_64/64986_fc645b3742.jpg -------------------------------------------------------------------------------- /dataset/tiny_imagenet_64_c/187922_5d1e159f39.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dorian25/colorization-transformer-experimental/HEAD/dataset/tiny_imagenet_64_c/187922_5d1e159f39.jpg -------------------------------------------------------------------------------- /dataset/tiny_imagenet_64_c/196860_1a9c864110.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dorian25/colorization-transformer-experimental/HEAD/dataset/tiny_imagenet_64_c/196860_1a9c864110.jpg -------------------------------------------------------------------------------- /dataset/tiny_imagenet_64_c/202290_804be7ec8f.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dorian25/colorization-transformer-experimental/HEAD/dataset/tiny_imagenet_64_c/202290_804be7ec8f.jpg -------------------------------------------------------------------------------- /dataset/tiny_imagenet_64_c/235344_49d89a0ee6.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dorian25/colorization-transformer-experimental/HEAD/dataset/tiny_imagenet_64_c/235344_49d89a0ee6.jpg -------------------------------------------------------------------------------- /dataset/tiny_imagenet_64_c/245481_d974325334.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dorian25/colorization-transformer-experimental/HEAD/dataset/tiny_imagenet_64_c/245481_d974325334.jpg -------------------------------------------------------------------------------- /dataset/tiny_imagenet_64_c/250459_5995a47cef.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dorian25/colorization-transformer-experimental/HEAD/dataset/tiny_imagenet_64_c/250459_5995a47cef.jpg -------------------------------------------------------------------------------- /dataset/tiny_imagenet_64_c/266774_cec9f5ec45.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dorian25/colorization-transformer-experimental/HEAD/dataset/tiny_imagenet_64_c/266774_cec9f5ec45.jpg -------------------------------------------------------------------------------- /dataset/tiny_imagenet_64_c/267004_4eef6e9577.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dorian25/colorization-transformer-experimental/HEAD/dataset/tiny_imagenet_64_c/267004_4eef6e9577.jpg -------------------------------------------------------------------------------- /dataset/tiny_imagenet_64_c/267530_4cec93ad32.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dorian25/colorization-transformer-experimental/HEAD/dataset/tiny_imagenet_64_c/267530_4cec93ad32.jpg -------------------------------------------------------------------------------- /dataset/tiny_imagenet_64_c/64986_fc645b3742.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dorian25/colorization-transformer-experimental/HEAD/dataset/tiny_imagenet_64_c/64986_fc645b3742.jpg -------------------------------------------------------------------------------- /dataset/tiny_imagenet_64_g/187922_5d1e159f39.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dorian25/colorization-transformer-experimental/HEAD/dataset/tiny_imagenet_64_g/187922_5d1e159f39.jpg -------------------------------------------------------------------------------- /dataset/tiny_imagenet_64_g/196860_1a9c864110.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dorian25/colorization-transformer-experimental/HEAD/dataset/tiny_imagenet_64_g/196860_1a9c864110.jpg -------------------------------------------------------------------------------- /dataset/tiny_imagenet_64_g/202290_804be7ec8f.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dorian25/colorization-transformer-experimental/HEAD/dataset/tiny_imagenet_64_g/202290_804be7ec8f.jpg -------------------------------------------------------------------------------- /dataset/tiny_imagenet_64_g/235344_49d89a0ee6.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dorian25/colorization-transformer-experimental/HEAD/dataset/tiny_imagenet_64_g/235344_49d89a0ee6.jpg -------------------------------------------------------------------------------- /dataset/tiny_imagenet_64_g/245481_d974325334.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dorian25/colorization-transformer-experimental/HEAD/dataset/tiny_imagenet_64_g/245481_d974325334.jpg -------------------------------------------------------------------------------- /dataset/tiny_imagenet_64_g/250459_5995a47cef.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dorian25/colorization-transformer-experimental/HEAD/dataset/tiny_imagenet_64_g/250459_5995a47cef.jpg -------------------------------------------------------------------------------- /dataset/tiny_imagenet_64_g/266774_cec9f5ec45.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dorian25/colorization-transformer-experimental/HEAD/dataset/tiny_imagenet_64_g/266774_cec9f5ec45.jpg -------------------------------------------------------------------------------- /dataset/tiny_imagenet_64_g/267004_4eef6e9577.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dorian25/colorization-transformer-experimental/HEAD/dataset/tiny_imagenet_64_g/267004_4eef6e9577.jpg -------------------------------------------------------------------------------- /dataset/tiny_imagenet_64_g/267530_4cec93ad32.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dorian25/colorization-transformer-experimental/HEAD/dataset/tiny_imagenet_64_g/267530_4cec93ad32.jpg -------------------------------------------------------------------------------- /dataset/tiny_imagenet_64_g/64986_fc645b3742.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dorian25/colorization-transformer-experimental/HEAD/dataset/tiny_imagenet_64_g/64986_fc645b3742.jpg -------------------------------------------------------------------------------- /plot/Loss Cross-entropy [ColTran CORE].png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dorian25/colorization-transformer-experimental/HEAD/plot/Loss Cross-entropy [ColTran CORE].png -------------------------------------------------------------------------------- /plot/Loss Cross-entropy [ColTran UPSAMPLER COLOR].png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dorian25/colorization-transformer-experimental/HEAD/plot/Loss Cross-entropy [ColTran UPSAMPLER COLOR].png -------------------------------------------------------------------------------- /plot/Loss Cross-entropy [ColTran UPSAMPLER SPATIAL].png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dorian25/colorization-transformer-experimental/HEAD/plot/Loss Cross-entropy [ColTran UPSAMPLER SPATIAL].png -------------------------------------------------------------------------------- /test/xg_0.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dorian25/colorization-transformer-experimental/HEAD/test/xg_0.png -------------------------------------------------------------------------------- /test/xg_1.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dorian25/colorization-transformer-experimental/HEAD/test/xg_1.png -------------------------------------------------------------------------------- /test/xg_2.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dorian25/colorization-transformer-experimental/HEAD/test/xg_2.png -------------------------------------------------------------------------------- /test/xg_pred_0.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dorian25/colorization-transformer-experimental/HEAD/test/xg_pred_0.png -------------------------------------------------------------------------------- /test/xg_pred_1.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dorian25/colorization-transformer-experimental/HEAD/test/xg_pred_1.png -------------------------------------------------------------------------------- /test/xg_pred_2.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dorian25/colorization-transformer-experimental/HEAD/test/xg_pred_2.png -------------------------------------------------------------------------------- /utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dorian25/colorization-transformer-experimental/HEAD/utils.py --------------------------------------------------------------------------------