Downloads · 30 days
0
Anime-Aesthetic-Predictor-Medium
Blackroot/Anime-Aesthetic-Predictor-Medium
Anime-Aesthetic-Predictor-Medium is a machine learning model from Blackroot. Use it for the machine learning task on the model card, and read the license before you ship it in a product. The card lists the license as mit.
An aesthetic predictor that takes anime images and gives it a score out of 10. Ideally you take this model and fine-tune it to your own preferences. There's a simple gradio demo and full training script here: <https:/…
Access
Public
Updated Feb 18, 2024
Repo size
350 MB
Likes
2
Public
Repo makeup
Click a slice to open those files.
.pth350 MB · 100%
At a glance
- License
- mit
- Access
- Public
- Created
- Feb 18, 2024
- Updated
- Feb 18, 2024
- SHA
- 5d732eb0
Model card
From the Hugging Face model README
An aesthetic predictor that takes anime images and gives it a score out of 10. Ideally you take this model and fine-tune it to your own preferences. There's a simple gradio demo and full training script here: https://github.com/CoffeeVampir3/easy-aesthetic-predictor
This model is based on ConVneXt: https://github.com/facebookresearch/ConvNeXt
This model was trained for 3 epochs at a learning rate of 1e-6 and then 9 epochs at 1e-5, training took about 6 hours on a 3090 TI with a dataset of 20,000 labeled images.
- License
- mit
- Created
- Feb 18, 2024
- Updated
- Feb 18, 2024