Skip to content

A repository for making an aesthetic prediction model based on the ConvneXt architecture.

Notifications You must be signed in to change notification settings

CoffeeVampir3/easy-aesthetic-predictor

Repository files navigation

How to use:

Use the included gradio demo to get simple predictions on images, see how it feels. Demo will automatically download pretrained weights from https://huggingface.co/Blackroot/Anime-Aesthetic-Predictor-Medium/tree/main for you to try. You can rip the inference code out of the gradio demo and use it in any python script or whatever, feel free. This model is likely very overfit on my aesthetic preferences, the idea with this is to reduce the amount of work required to prep datasets, so you likely want to tune it to your preferences instead. Additionally, this was only trained on anime images.

Training:

See the wiki, it's designed to be pretty easy. You do need a GPU to train. https://github.com/CoffeeVampir3/easy-aesthetic-predictor/wiki

About

A repository for making an aesthetic prediction model based on the ConvneXt architecture.

Topics

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published