DiffusionToolkit
DiffusionToolkit copied to clipboard
Aesthetic Score: calcualte for images that do not have initially score saved
Is your feature request related to a problem? Please describe. I have bunch of images that do not have "Aesthetic score"
Describe the solution you'd like
- I'd like to have a plugin or menu that will allow calculate Aesthetic Score AND save it back into metadata.
- From DiffusionToolkit a. find files by search criteria (like AestheticScore == null) b. Apply custom command for those files
- From DiffusionToolkit (sounds like min effort one) a. find files by search criteria (like AestheticScore == null) b. Export/Save search list into file
Describe alternatives you've considered
- maybe A1111 Extras can do it for batch of files, but I could not find how to do it
- Custom Python script that will iterate folders and use https://github.com/vladmandic/sd-extension-aesthetic-scorer
Additional context
that aesthetic scorer got a python script named "aesthetic-scorer-cli.py" which you can use in the command line - just feed it with a folder of images you want to score.
also a1111 extras could do that if u batch-upscale with selecting upscale by x1 and no upscaler
@Ariloum thank you for a hints. I suspect a1111 extras will not update existing image "Parameters", but produce new files so will loose Date Created, which I need for sorting.
I've actually followed the extension, but CLI had bugs that were showing improperly in DiffusionToolkit
So I did fork with bugfixes: https://github.com/DmytroSokhach/sd-extension-aesthetic-scorer , usage is the same. E.g.:
python aesthetic-scorer-cli.py --save "#" "D:\Dev\StabilityMatrix\Packages\ComfyUI\output"
It now handles images without metadata as well as certain metadata scenarios which were buggy and not showing properly. Cheers! 👌