AI-generated_images_detector
This model achieves the following results on the evaluation set:
- Loss: 0.0987
- Accuracy: 0.9736
To utilize this model
from PIL import Image
from transformers import pipeline
classifier = pipeline("image-classification", model="NYUAD-ComNets/AI-generated_images_detector")
image=Image.open("path_to_image")
pred=classifier(image)
print(pred)
Training and evaluation data
Training results
Training Loss | Epoch | Step | Validation Loss | Accuracy |
---|---|---|---|---|
0.0431 | 0.55 | 100 | 0.1672 | 0.9568 |
0.0139 | 1.1 | 200 | 0.2338 | 0.9398 |
0.0201 | 1.66 | 300 | 0.1291 | 0.9655 |
0.0023 | 2.21 | 400 | 0.1147 | 0.9709 |
0.0033 | 2.76 | 500 | 0.0987 | 0.9736 |
- Downloads last month
- 458
Evaluation results
- Accuracy on imagefoldervalidation set self-reported0.974