# stable-diffusion-webui-aesthetic-image-scorer

**Repository Path**: dbscholar0/stable-diffusion-webui-aesthetic-image-scorer

## Basic Information

- **Project Name**: stable-diffusion-webui-aesthetic-image-scorer
- **Description**: 使用 CLIP+MLP 美学得分预测器基于 Chad Scorer 计算生成图像的美学分数。
- **Primary Language**: Unknown
- **License**: MIT
- **Default Branch**: main
- **Homepage**: None
- **GVP Project**: No

## Statistics

- **Stars**: 0
- **Forks**: 0
- **Created**: 2023-04-08
- **Last Updated**: 2023-04-08

## Categories & Tags

**Categories**: Uncategorized

**Tags**: None

## README

# Aesthetic Image Scorer

![](tag_group_by.png)

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

Calculates aestetic score for generated images using [CLIP+MLP Aesthetic Score Predictor](https://github.com/christophschuhmann/improved-aesthetic-predictor) based on [Chad Scorer](https://github.com/grexzen/SD-Chad/blob/main/chad_scorer.py)

See [Discussion](https://github.com/AUTOMATIC1111/stable-diffusion-webui/discussions/1831)

Saves score to windows tags with other options planned

## Installation
Clone the repo into the `extensions` directory and restart the web ui:

```commandline
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`

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


To upgrade do:

```commandline
git pull
```

or use `Extensions` tab to upgrade.

After restarting the ui, see settings for options

## Features
![](settings_section.png)
- 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](https://github.com/Dijji/FileMeta/releases). Here is a [walkthrough](https://www.thewindowsclub.com/enable-tags-for-unsupported-files-windows) 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