Instructions to use Modotte/AIRealNet-2 with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Transformers
How to use Modotte/AIRealNet-2 with Transformers:
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("image-classification", model="Modotte/AIRealNet-2") pipe("https://huggingface.co/datasets/huggingface/documentation-images/resolve/main/hub/parrots.png")# pip install -U transformers accelerate # Load model directly from transformers import AutoImageProcessor, AutoModelForImageClassification processor = AutoImageProcessor.from_pretrained("Modotte/AIRealNet-2") model = AutoModelForImageClassification.from_pretrained("Modotte/AIRealNet-2", device_map="auto") - Notebooks
- Google Colab
- Kaggle
Update the olf info
Browse files
README.md
CHANGED
|
@@ -22,11 +22,11 @@ tags:
|
|
| 22 |
|
| 23 |
---
|
| 24 |
|
| 25 |
-
> This is an upgraded version of AIRealNet. We have improved its performance on social media images of different qualities, making it a great upgrade from its previous version.
|
| 26 |
|
| 27 |
## Overview
|
| 28 |
|
| 29 |
-
In an era of rapidly advancing AI-generated imagery, deepfakes, and synthetic media, the need for reliable detection tools has never been higher. **AIRealNet** is a binary image classifier explicitly designed to distinguish **AI-generated images** from **real human photographs**. This model is optimized to detect conventional AI-generated content while adhering to strict privacy standards—avoiding personal or sensitive images.
|
| 30 |
|
| 31 |
* **Class 0:** AI-generated image
|
| 32 |
* **Class 1:** Real human image
|
|
@@ -37,14 +37,13 @@ By leveraging the robust **SwinV2 Tiny** architecture as its backbone, AIRealNet
|
|
| 37 |
|
| 38 |
## Key Features
|
| 39 |
|
| 40 |
-
1. **High Accuracy on Public Datasets:**
|
| 41 |
-
|
| 42 |
-
|
| 43 |
2. **Balanced Training Split:**
|
| 44 |
The dataset contains a balanced number of AI-generated and real images, ensuring unbiased training and minimizing class imbalance issues.
|
| 45 |
|
| 46 |
-
* **AI-Generated:**
|
| 47 |
-
* **Human-Images:**
|
| 48 |
|
| 49 |
4. **Ethical Design:**
|
| 50 |
No personal photos were included, even if edited or AI-modified, respecting privacy and ethical AI principles.
|
|
@@ -56,8 +55,8 @@ By leveraging the robust **SwinV2 Tiny** architecture as its backbone, AIRealNet
|
|
| 56 |
|
| 57 |
## Training Data
|
| 58 |
|
| 59 |
-
* **Dataset:**
|
| 60 |
-
* **Size:**
|
| 61 |
* **Split:** Used the train split for fine-tuning; validation performed on a separate balanced subset.
|
| 62 |
* **Notes:** Images sourced from public datasets and AI generation tools. Edited personal photos were intentionally excluded.
|
| 63 |
|
|
@@ -164,7 +163,7 @@ Future iterations aim to:
|
|
| 164 |
### Citation
|
| 165 |
```bibtex
|
| 166 |
@misc{Modotte_AIRealNet_2025,
|
| 167 |
-
title={AIRealNet: A Fine-Tuned Vision Transformer for Detecting AI-Generated vs Real Human Images},
|
| 168 |
author={Parvesh Rawal},
|
| 169 |
publisher={Hugging Face},
|
| 170 |
year={2025},
|
|
|
|
| 22 |
|
| 23 |
---
|
| 24 |
|
| 25 |
+
> This is an upgraded version of AIRealNet. We have improved its performance on social media images of different qualities, making it a great upgrade from its previous version.
|
| 26 |
|
| 27 |
## Overview
|
| 28 |
|
| 29 |
+
In an era of rapidly advancing AI-generated imagery, deepfakes, and synthetic media, the need for reliable detection tools has never been higher. **AIRealNet-2** is a binary image classifier explicitly designed to distinguish **AI-generated images** from **real human photographs**. This model is optimized to detect conventional AI-generated content while adhering to strict privacy standards—avoiding personal or sensitive images.
|
| 30 |
|
| 31 |
* **Class 0:** AI-generated image
|
| 32 |
* **Class 1:** Real human image
|
|
|
|
| 37 |
|
| 38 |
## Key Features
|
| 39 |
|
| 40 |
+
1. **High Accuracy on Public Datasets:** Model is trained on data from social media
|
| 41 |
+
|
|
|
|
| 42 |
2. **Balanced Training Split:**
|
| 43 |
The dataset contains a balanced number of AI-generated and real images, ensuring unbiased training and minimizing class imbalance issues.
|
| 44 |
|
| 45 |
+
* **AI-Generated:** 50%
|
| 46 |
+
* **Human-Images:** 50%
|
| 47 |
|
| 48 |
4. **Ethical Design:**
|
| 49 |
No personal photos were included, even if edited or AI-modified, respecting privacy and ethical AI principles.
|
|
|
|
| 55 |
|
| 56 |
## Training Data
|
| 57 |
|
| 58 |
+
* **Dataset:** Collection of images from differnt plaforms like instagram, x.
|
| 59 |
+
* **Size:** 98k images (balanced between AI and human)
|
| 60 |
* **Split:** Used the train split for fine-tuning; validation performed on a separate balanced subset.
|
| 61 |
* **Notes:** Images sourced from public datasets and AI generation tools. Edited personal photos were intentionally excluded.
|
| 62 |
|
|
|
|
| 163 |
### Citation
|
| 164 |
```bibtex
|
| 165 |
@misc{Modotte_AIRealNet_2025,
|
| 166 |
+
title={AIRealNet-2: A Fine-Tuned Vision Transformer for Detecting AI-Generated vs Real Human Images},
|
| 167 |
author={Parvesh Rawal},
|
| 168 |
publisher={Hugging Face},
|
| 169 |
year={2025},
|