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

Downloads last month: 59

Safetensors

Model size

85.8M params

Tensor type

F32

Inference Providers NEW

Image Classification

This model is not currently available via any of the supported third-party Inference Providers, and the model is not deployed on the HF Inference API.

Model tree for dima806/ai_vs_human_generated_image_detection

Base model

google/vit-base-patch16-224-in21k

Finetuned

(1858)

this model