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 Meta. Here is a walkthrough to add tagging for unsupported filetypes.
• Save as category (Windows Only)
• Save as generation parameter text
• Run prediction using CPU

FAQs

• If you want to add score to existing images, try running Extra->Batch from directory without upscaling. The new images will get the score.
• If seeing this error Aesthetic Image Scorer: Unable to write tag or category
  • Make sure you're using the latest webui and this extension
  • If still seeing this, post your settings from the settings page, your OS and whether you're saving PNG or JPG
• If seeing this error Aesthetic Image Scorer: Unable to load Windows tagging script from tools directory
  • Check if the tools directory has add_tags.py