Predicts aesthetic scores for images. Trained on Stable Hord user ratings of Stable Diffusion generated images.
Model files in aesthetics_scorer folder
Simple gradio demo
python aesthetics_scorer/demo.py
- Inspired by https://github.com/christophschuhmann/improved-aesthetic-predictor
- Image dataset https://poloclub.github.io/diffusiondb/
- Image ratings by https://aihorde.net/
@article{wangDiffusionDBLargescalePrompt2022,
title = {Large-Scale Prompt Gallery Dataset for Text-to-Image Generative Models},
author = {Wang, Zijie J. and Montoya, Evan and Munechika, David and Yang, Haoyang and Hoover, Benjamin and Chau, Duen Horng},
year = {2022},
journal = {arXiv:2210.14896 [cs]},
url = {https://arxiv.org/abs/2210.14896}
}
@software{ilharco_gabriel_2021_5143773,
author = {Ilharco, Gabriel and
Wortsman, Mitchell and
Wightman, Ross and
Gordon, Cade and
Carlini, Nicholas and
Taori, Rohan and
Dave, Achal and
Shankar, Vaishaal and
Namkoong, Hongseok and
Miller, John and
Hajishirzi, Hannaneh and
Farhadi, Ali and
Schmidt, Ludwig},
title = {OpenCLIP},
month = jul,
year = 2021,
note = {If you use this software, please cite it as below.},
publisher = {Zenodo},
version = {0.1},
doi = {10.5281/zenodo.5143773},
url = {https://doi.org/10.5281/zenodo.5143773}
}