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.

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

Downloads last month: 13

Safetensors

Model size

85.8M params

Tensor type

F32

Inference API

Unable to determine this model's library. Check the docs .

Model tree for dima806/human_faces_ai_vs_real_image_detection

Base model

google/vit-base-patch16-224-in21k

Finetuned

(1769)

this model