Downloads · 30 days
1.3K
2% of all-time downloads
NYUAD-ComNets/NYUAD_AI-generated_images_detector
NYUAD_AI-generated_images_detector is a image classification model from NYUAD-ComNets. Use it when you need a label for an image. It is set up for transformers. The card lists the license as apache-2.0.
should probably proofread and complete it, then remove this comment. --
Downloads · 30 days
1.3K
2% of all-time downloads
All-time downloads
60.1K
Public
Parameters
85.8M
1.7 GB on disk
Likes
11
Public
Click a slice to open those files.
.safetensors343 MB · 100%
From the Hugging Face model README
This model achieves the following results on the evaluation set:
from PIL import Image
from transformers import pipeline
classifier = pipeline("image-classification", model="NYUAD-ComNets/NYUAD_AI-generated_images_detector")
image=Image.open("path_to_image")
pred=classifier(image)
print(pred)
| 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 |
@article{aldahoul2024detecting,
title={Detecting AI-Generated Images Using Vision Transformers: A Robust Approach for Safeguarding Visual Media Integrity},
author={AlDahoul, Nouar and Zaki, Yasir},
journal={Available at SSRN},
year={2024}
}
@misc{ComNets,
url={https://huggingface.co/NYUAD-ComNets/NYUAD_AI-generated_images_detector](https://huggingface.co/NYUAD-ComNets/NYUAD_AI-generated_images_detector)},
title={NYUAD_AI-generated_images_detector},
author={Nouar AlDahoul, Yasir Zaki}
}