|
--- |
|
license: apache-2.0 |
|
metrics: |
|
- accuracy |
|
- f1 |
|
base_model: |
|
- google/vit-base-patch16-224-in21k |
|
pipeline_tag: image-classification |
|
library_name: transformers |
|
--- |
|
Predicts with about 98% accuracy whether an attached image is AI-generated. |
|
|
|
See https://www.kaggle.com/code/dima806/ai-vs-human-generated-images-prediction-vit for details. |
|
|
|
 |
|
|
|
``` |
|
Classification report: |
|
|
|
precision recall f1-score support |
|
|
|
human 0.9655 0.9930 0.9790 3998 |
|
AI-generated 0.9928 0.9645 0.9784 3997 |
|
|
|
accuracy 0.9787 7995 |
|
macro avg 0.9791 0.9787 0.9787 7995 |
|
weighted avg 0.9791 0.9787 0.9787 7995 |
|
``` |
