├── .flake8 ├── .gitignore ├── .pre-commit-config.yaml ├── LICENSE ├── README.md ├── part1-code ├── minimal-app │ ├── app.py │ ├── app2.py │ └── requirements.txt └── superres-local │ ├── app.py │ ├── examples │ └── comic_lr.png │ ├── imgproc.py │ ├── model.py │ ├── requirements.txt │ └── weights │ └── SRGAN_x4-ImageNet-c71a4860.pth.tar ├── part2-code ├── pt-lightning-app │ ├── main.py │ ├── my_classifier_template │ │ ├── __init__.py │ │ ├── dataset.py │ │ └── model.py │ └── requirements.txt ├── superres-from-part1 │ ├── .lightning │ ├── app.py │ ├── examples │ │ └── comic_lr.png │ ├── imgproc.py │ ├── model.py │ ├── requirements.txt │ └── weights │ │ └── SRGAN_x4-ImageNet-c71a4860.pth.tar ├── train-and-save │ ├── .lightning │ ├── .lightningignore │ ├── app_with_drive.py │ ├── app_with_path.py │ ├── my_train_script.py │ └── requirements.txt └── train │ ├── .lightning │ ├── .lightningignore │ ├── app.py │ ├── my_train_script.py │ └── requirements.txt └── part3-code └── .gitkeep /.flake8: -------------------------------------------------------------------------------- 1 | [flake8] 2 | max-line-length = 88 -------------------------------------------------------------------------------- /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rasbt/srgan-lightning-blog/HEAD/.gitignore -------------------------------------------------------------------------------- /.pre-commit-config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rasbt/srgan-lightning-blog/HEAD/.pre-commit-config.yaml -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rasbt/srgan-lightning-blog/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- 1 | # srgan-lightning-blog -------------------------------------------------------------------------------- /part1-code/minimal-app/app.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rasbt/srgan-lightning-blog/HEAD/part1-code/minimal-app/app.py -------------------------------------------------------------------------------- /part1-code/minimal-app/app2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rasbt/srgan-lightning-blog/HEAD/part1-code/minimal-app/app2.py -------------------------------------------------------------------------------- /part1-code/minimal-app/requirements.txt: -------------------------------------------------------------------------------- 1 | lightning==2022.6.15.post2 -------------------------------------------------------------------------------- /part1-code/superres-local/app.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rasbt/srgan-lightning-blog/HEAD/part1-code/superres-local/app.py -------------------------------------------------------------------------------- /part1-code/superres-local/examples/comic_lr.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rasbt/srgan-lightning-blog/HEAD/part1-code/superres-local/examples/comic_lr.png -------------------------------------------------------------------------------- /part1-code/superres-local/imgproc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rasbt/srgan-lightning-blog/HEAD/part1-code/superres-local/imgproc.py -------------------------------------------------------------------------------- /part1-code/superres-local/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rasbt/srgan-lightning-blog/HEAD/part1-code/superres-local/model.py -------------------------------------------------------------------------------- /part1-code/superres-local/requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rasbt/srgan-lightning-blog/HEAD/part1-code/superres-local/requirements.txt -------------------------------------------------------------------------------- /part1-code/superres-local/weights/SRGAN_x4-ImageNet-c71a4860.pth.tar: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rasbt/srgan-lightning-blog/HEAD/part1-code/superres-local/weights/SRGAN_x4-ImageNet-c71a4860.pth.tar -------------------------------------------------------------------------------- /part2-code/pt-lightning-app/main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rasbt/srgan-lightning-blog/HEAD/part2-code/pt-lightning-app/main.py -------------------------------------------------------------------------------- /part2-code/pt-lightning-app/my_classifier_template/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /part2-code/pt-lightning-app/my_classifier_template/dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rasbt/srgan-lightning-blog/HEAD/part2-code/pt-lightning-app/my_classifier_template/dataset.py -------------------------------------------------------------------------------- /part2-code/pt-lightning-app/my_classifier_template/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rasbt/srgan-lightning-blog/HEAD/part2-code/pt-lightning-app/my_classifier_template/model.py -------------------------------------------------------------------------------- /part2-code/pt-lightning-app/requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rasbt/srgan-lightning-blog/HEAD/part2-code/pt-lightning-app/requirements.txt -------------------------------------------------------------------------------- /part2-code/superres-from-part1/.lightning: -------------------------------------------------------------------------------- 1 | name: superres-demo-1 2 | -------------------------------------------------------------------------------- /part2-code/superres-from-part1/app.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rasbt/srgan-lightning-blog/HEAD/part2-code/superres-from-part1/app.py -------------------------------------------------------------------------------- /part2-code/superres-from-part1/examples/comic_lr.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rasbt/srgan-lightning-blog/HEAD/part2-code/superres-from-part1/examples/comic_lr.png -------------------------------------------------------------------------------- /part2-code/superres-from-part1/imgproc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rasbt/srgan-lightning-blog/HEAD/part2-code/superres-from-part1/imgproc.py -------------------------------------------------------------------------------- /part2-code/superres-from-part1/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rasbt/srgan-lightning-blog/HEAD/part2-code/superres-from-part1/model.py -------------------------------------------------------------------------------- /part2-code/superres-from-part1/requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rasbt/srgan-lightning-blog/HEAD/part2-code/superres-from-part1/requirements.txt -------------------------------------------------------------------------------- /part2-code/superres-from-part1/weights/SRGAN_x4-ImageNet-c71a4860.pth.tar: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rasbt/srgan-lightning-blog/HEAD/part2-code/superres-from-part1/weights/SRGAN_x4-ImageNet-c71a4860.pth.tar -------------------------------------------------------------------------------- /part2-code/train-and-save/.lightning: -------------------------------------------------------------------------------- 1 | name: fresh-sammet-2010 2 | -------------------------------------------------------------------------------- /part2-code/train-and-save/.lightningignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rasbt/srgan-lightning-blog/HEAD/part2-code/train-and-save/.lightningignore -------------------------------------------------------------------------------- /part2-code/train-and-save/app_with_drive.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rasbt/srgan-lightning-blog/HEAD/part2-code/train-and-save/app_with_drive.py -------------------------------------------------------------------------------- /part2-code/train-and-save/app_with_path.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rasbt/srgan-lightning-blog/HEAD/part2-code/train-and-save/app_with_path.py -------------------------------------------------------------------------------- /part2-code/train-and-save/my_train_script.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rasbt/srgan-lightning-blog/HEAD/part2-code/train-and-save/my_train_script.py -------------------------------------------------------------------------------- /part2-code/train-and-save/requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rasbt/srgan-lightning-blog/HEAD/part2-code/train-and-save/requirements.txt -------------------------------------------------------------------------------- /part2-code/train/.lightning: -------------------------------------------------------------------------------- 1 | name: heavy-rhodes-8889 2 | -------------------------------------------------------------------------------- /part2-code/train/.lightningignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rasbt/srgan-lightning-blog/HEAD/part2-code/train/.lightningignore -------------------------------------------------------------------------------- /part2-code/train/app.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rasbt/srgan-lightning-blog/HEAD/part2-code/train/app.py -------------------------------------------------------------------------------- /part2-code/train/my_train_script.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rasbt/srgan-lightning-blog/HEAD/part2-code/train/my_train_script.py -------------------------------------------------------------------------------- /part2-code/train/requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rasbt/srgan-lightning-blog/HEAD/part2-code/train/requirements.txt -------------------------------------------------------------------------------- /part3-code/.gitkeep: -------------------------------------------------------------------------------- 1 | --------------------------------------------------------------------------------