|
|
--- |
|
|
|
|
|
inference: false |
|
|
|
|
|
license: apache-2.0 |
|
|
|
|
|
tags: |
|
|
- generated_from_trainer |
|
|
datasets: |
|
|
- imagefolder |
|
|
metrics: |
|
|
- accuracy |
|
|
model-index: |
|
|
- name: AI-generated_images_detector |
|
|
results: |
|
|
- task: |
|
|
name: Image Classification |
|
|
type: image-classification |
|
|
dataset: |
|
|
name: imagefolder |
|
|
type: imagefolder |
|
|
config: default |
|
|
split: validation |
|
|
args: default |
|
|
metrics: |
|
|
- name: Accuracy |
|
|
type: accuracy |
|
|
value: 0.9735697557711609 |
|
|
--- |
|
|
|
|
|
<!-- This model card has been generated automatically according to the information the Trainer had access to. You |
|
|
should probably proofread and complete it, then remove this comment. --> |
|
|
|
|
|
# AI-generated_images_detector |
|
|
|
|
|
This model achieves the following results on the evaluation set: |
|
|
- Loss: 0.0987 |
|
|
- Accuracy: 0.9736 |
|
|
|
|
|
# To utilize this model |
|
|
|
|
|
``` python |
|
|
|
|
|
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 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 | |
|
|
|
|
|
|
|
|
# BibTeX entry and citation info |
|
|
|
|
|
``` |
|
|
@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} |
|
|
} |
|
|
|
|
|
|
