├── .gitignore ├── LICENSE ├── README.md ├── assets └── example.png ├── configs └── train.yaml ├── eval.ipynb ├── requirements.txt ├── scripts └── download_aadb.sh ├── torch_aesthetics ├── __init__.py ├── aadb.py ├── losses.py ├── metrics.py └── models.py ├── train.ipynb └── train.py /.gitignore: -------------------------------------------------------------------------------- 1 | data/ 2 | runs/ 3 | models/ 4 | .mypy_cache/ 5 | *.pyc 6 | test.ipynb -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/isaaccorley/deep-aesthetics-pytorch/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/isaaccorley/deep-aesthetics-pytorch/HEAD/README.md -------------------------------------------------------------------------------- /assets/example.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/isaaccorley/deep-aesthetics-pytorch/HEAD/assets/example.png -------------------------------------------------------------------------------- /configs/train.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/isaaccorley/deep-aesthetics-pytorch/HEAD/configs/train.yaml -------------------------------------------------------------------------------- /eval.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/isaaccorley/deep-aesthetics-pytorch/HEAD/eval.ipynb -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/isaaccorley/deep-aesthetics-pytorch/HEAD/requirements.txt -------------------------------------------------------------------------------- /scripts/download_aadb.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/isaaccorley/deep-aesthetics-pytorch/HEAD/scripts/download_aadb.sh -------------------------------------------------------------------------------- /torch_aesthetics/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/isaaccorley/deep-aesthetics-pytorch/HEAD/torch_aesthetics/__init__.py -------------------------------------------------------------------------------- /torch_aesthetics/aadb.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/isaaccorley/deep-aesthetics-pytorch/HEAD/torch_aesthetics/aadb.py -------------------------------------------------------------------------------- /torch_aesthetics/losses.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/isaaccorley/deep-aesthetics-pytorch/HEAD/torch_aesthetics/losses.py -------------------------------------------------------------------------------- /torch_aesthetics/metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/isaaccorley/deep-aesthetics-pytorch/HEAD/torch_aesthetics/metrics.py -------------------------------------------------------------------------------- /torch_aesthetics/models.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/isaaccorley/deep-aesthetics-pytorch/HEAD/torch_aesthetics/models.py -------------------------------------------------------------------------------- /train.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/isaaccorley/deep-aesthetics-pytorch/HEAD/train.ipynb -------------------------------------------------------------------------------- /train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/isaaccorley/deep-aesthetics-pytorch/HEAD/train.py --------------------------------------------------------------------------------