Update README.md
Browse files
README.md
CHANGED
|
@@ -1,3 +1,10 @@
|
|
| 1 |
---
|
| 2 |
license: mit
|
| 3 |
---
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
---
|
| 2 |
license: mit
|
| 3 |
---
|
| 4 |
+
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:
|
| 5 |
+
<https://github.com/CoffeeVampir3/easy-aesthetic-predictor>
|
| 6 |
+
|
| 7 |
+
This model is based on ConVneXt:
|
| 8 |
+
<https://github.com/facebookresearch/ConvNeXt>
|
| 9 |
+
|
| 10 |
+
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.
|