Update README.md
Browse files
README.md
CHANGED
|
@@ -15,6 +15,8 @@ tags:
|
|
| 15 |
|
| 16 |
This is a proof of concept model for detecting anime style AI images. Using Vision Transformer, it was trained on 1M human-made real and 217K AI generated anime images. During training either type appeared in equal amount to avoid biases. The model was trained on a single RTX 3090 GPU for about 40 hours, ~35 epochs.
|
| 17 |
|
|
|
|
|
|
|
| 18 |
## Evaluation
|
| 19 |
|
| 20 |
Each checkpoint was evaluated on 500-500 real and AI images.
|
|
|
|
| 15 |
|
| 16 |
This is a proof of concept model for detecting anime style AI images. Using Vision Transformer, it was trained on 1M human-made real and 217K AI generated anime images. During training either type appeared in equal amount to avoid biases. The model was trained on a single RTX 3090 GPU for about 40 hours, ~35 epochs.
|
| 17 |
|
| 18 |
+
The training logs are available on my [wandb](https://wandb.ai/legekka/AI-Image-Detector).
|
| 19 |
+
|
| 20 |
## Evaluation
|
| 21 |
|
| 22 |
Each checkpoint was evaluated on 500-500 real and AI images.
|