Aesthetic Image Scorer

Calculates aestetic score for generated images using CLIP+MLP Aesthetic Score Predictor based on Chad Scorer

Saves score to windows tags with other options planned

Installation

Clone the repo into the extensions directory and restart the web ui:

git clone https://github.com/tsngo/stable-diffusion-webui-aesthetic-image-scorer extensions/aesthetic-image-scorer

or use the Extensions tab of the webui to Install from URL

https://github.com/tsngo/stable-diffusion-webui-aesthetic-image-scorer

To upgrade do:

git pull

or use Extensions tab to upgrade.

After restarting the ui, see settings for options

Save aesthetic score and other things
- hash is a md5 hash of prompt, negative prompt, dimensions, steps, cfg scale, seed, model hash (experimental). Don't rely on this.
- rest should be obvious
Save as EXIF or PNG Info Chunk (Only PNG Chunk Info currently)
Save as tag (Windows Only)
- Added to tags as aesthetic_score_5.9
- JPG supports by default. PNG tags requires a 3rd party software like File Metadata
Save as category (Windows Only)
Save as generation parameter text
Run prediction using CPU

There is a bug currently preventing writing to PNG Info Chuck
- See AUTOMATIC1111/stable-diffusion-webui#3723
If seeing this error Aesthetic Image Scorer: Unable to write tag or category
- probably related to above bug but the windows tagging should still work
If seeing this error Aesthetic Image Scorer: Unable to load Windows tagging script from tools directory
- check if the tools directory has files. Submodules no longer required.