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

Go to file

Trung Ngo cfc8cc1581 Add more tagging / png info / gen params options		2022-10-31 23:33:25 -05:00
scripts	Add more tagging / png info / gen params options	2022-10-31 23:33:25 -05:00
tools	Add more tagging / png info / gen params options	2022-10-31 23:33:25 -05:00
.gitignore	Initial commit	2022-10-23 15:48:55 -05:00
.gitmodules	Add more tagging / png info / gen params options	2022-10-31 23:33:25 -05:00
LICENSE	Initial commit	2022-10-23 15:48:55 -05:00
README.md	Update readme	2022-10-27 22:49:04 -05:00
tag_group_by.png	First usable version	2022-10-25 00:27:12 -05:00

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 --recurse-submodules https://github.com/tsngo/stable-diffusion-webui-aesthetic-image-scorer extensions/aesthetic-image-scorer

To upgrade do:

git pull

After restarting the ui, see settings for options

Save score as EXIF or PNG Info Chunk (Only PNG Chunk Info currently)
Save score 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
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
- The tools submodule didn't get pulled. Make sure to use follow the installation instructions