├── README.md ├── anime_colorization └── data │ └── .gitignore ├── config.py ├── dataset ├── dataset.py └── transforms.py ├── inference.py ├── log ├── .gitignore └── log.png ├── loss.py ├── main.py ├── model ├── Dnet.py ├── Gnet.py └── layernorm.py ├── output └── .gitignore ├── requirements.txt ├── test ├── result.png ├── result2.png └── test.png └── trainer.py /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gakkiri/SGRUnet-pytorch/HEAD/README.md -------------------------------------------------------------------------------- /anime_colorization/data/.gitignore: -------------------------------------------------------------------------------- 1 | ./train 2 | ./val 3 | -------------------------------------------------------------------------------- /config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gakkiri/SGRUnet-pytorch/HEAD/config.py -------------------------------------------------------------------------------- /dataset/dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gakkiri/SGRUnet-pytorch/HEAD/dataset/dataset.py -------------------------------------------------------------------------------- /dataset/transforms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gakkiri/SGRUnet-pytorch/HEAD/dataset/transforms.py -------------------------------------------------------------------------------- /inference.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gakkiri/SGRUnet-pytorch/HEAD/inference.py -------------------------------------------------------------------------------- /log/.gitignore: -------------------------------------------------------------------------------- 1 | ./log.txt 2 | -------------------------------------------------------------------------------- /log/log.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gakkiri/SGRUnet-pytorch/HEAD/log/log.png -------------------------------------------------------------------------------- /loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gakkiri/SGRUnet-pytorch/HEAD/loss.py -------------------------------------------------------------------------------- /main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gakkiri/SGRUnet-pytorch/HEAD/main.py -------------------------------------------------------------------------------- /model/Dnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gakkiri/SGRUnet-pytorch/HEAD/model/Dnet.py -------------------------------------------------------------------------------- /model/Gnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gakkiri/SGRUnet-pytorch/HEAD/model/Gnet.py -------------------------------------------------------------------------------- /model/layernorm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gakkiri/SGRUnet-pytorch/HEAD/model/layernorm.py -------------------------------------------------------------------------------- /output/.gitignore: -------------------------------------------------------------------------------- 1 | 2 | -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- 1 | torchvision>=0.3.1 2 | timm==0.1.14 3 | easydict==1.9 4 | Pillow==6.2.2 5 | -------------------------------------------------------------------------------- /test/result.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gakkiri/SGRUnet-pytorch/HEAD/test/result.png -------------------------------------------------------------------------------- /test/result2.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gakkiri/SGRUnet-pytorch/HEAD/test/result2.png -------------------------------------------------------------------------------- /test/test.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gakkiri/SGRUnet-pytorch/HEAD/test/test.png -------------------------------------------------------------------------------- /trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gakkiri/SGRUnet-pytorch/HEAD/trainer.py --------------------------------------------------------------------------------