dima806's picture
Update README.md
b30b43d verified
metadata
license: apache-2.0
metrics:
  - accuracy
  - f1
base_model:
  - google/vit-base-patch16-224-in21k

Predicts with about 100% accuracy whether the human face is real or AI-generated.

See https://www.kaggle.com/code/dima806/human-faces-ai-vs-real-detection-vit for details.

image/png

Classification report:

                     precision    recall  f1-score   support

AI-Generated Images     1.0000    1.0000    1.0000      2000
        Real Images     1.0000    1.0000    1.0000      2000

           accuracy                         1.0000      4000
          macro avg     1.0000    1.0000    1.0000      4000
       weighted avg     1.0000    1.0000    1.0000      4000