1580158
/
stable-diffusion-webui-aesthetic-image-scorer
mirror of https://github.com/tsngo/stable-diffusion-webui-aesthetic-image-scorer.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity
Calculates aesthetic score for generated images using CLIP+MLP Aesthetic Score Predictor based on Chad Scorer
14 Commits
1 Branch
0 Tags
865 KiB
Python 100%
 
Go to file
Trung Ngo e6e31804e4 Update readme 2022-10-31 23:57:26 -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-31 23:57:26 -05:00
settings_section.png Update readme 2022-10-31 23:57:26 -05:00
tag_group_by.png First usable version 2022-10-25 00:27:12 -05:00

README.md

Aesthetic Image Scorer

Extension for https://github.com/AUTOMATIC1111/stable-diffusion-webui

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

See Discussion

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

Features

  • 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

FAQ

  • There is a bug currently preventing writing to PNG Info Chuck
  • 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.