Hugging Face's logo

dima806
/
ai_vs_human_generated_image_detection

Image Classification
Transformers
Safetensors
vit
Model card Files Community
ai_vs_human_generated_image_detection
File size: 815 Bytes 
b258c3c
 
 
 
 
 
 
 
 
7a52c11
b2957e6
7a52c11
 
 
b2957e6
7a52c11
 
 
 
 
 
b2957e6
 
7a52c11
b2957e6
 
 
7a52c11
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
 
---
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.

![image/png](https://cdn-uploads.huggingface.co/production/uploads/6449300e3adf50d864095b90/vaVofgP3NO3ybwG0GSbC3.png)

```
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
```