SadraCoding commited on
Commit
1350a96
Β·
verified Β·
1 Parent(s): 83c560d

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +12 -12
README.md CHANGED
@@ -1,4 +1,4 @@
1
- # 🎭 SDXL-Deepfake-Detector
2
  ### Detecting AI-Generated Faces with Precision and Purpose
3
 
4
  >*Not just another classifier β€” a tool for digital truth.*
@@ -7,16 +7,16 @@ Developed by **[Sadra Milani Moghaddam](https://sadramilani.ir/)**
7
 
8
  ---
9
 
10
- ## 🌍 Why This Matters
11
  As generative AI (like SDXL, DALLΒ·E, and Midjourney) becomes more accessible, the line between real and synthetic media blurs β€” especially for vulnerable communities. This project started as a technical experiment but evolved into a **privacy-aware, open-source defense** against visual misinformation, with a focus on **ethical AI deployment**.
12
 
13
  ---
14
 
15
- ## πŸš€ Model Overview
16
 
17
- **SDXL-Deepfake-Detector** is a fine-tuned vision transformer that classifies human faces as **AI-Generated (0)** or **Real (1)**.
18
 
19
- ## 🧠 Training Approach
20
 
21
  This model was obtained by **fine-tuning** the [`Organika/sdxl-detector`](https://huggingface.co/Organika/sdxl-detector) β€” a vision transformer pre-trained specifically to detect SDXL-generated faces β€” on the [140k Real and Fake Faces](https://www.kaggle.com/datasets/xhlulu/140k-real-and-fake-faces) dataset.
22
 
@@ -27,7 +27,7 @@ This approach leverages:
27
 
28
  The result is a lightweight, high-accuracy detector optimized for **both SDXL and general diffusion-based deepfakes**.
29
 
30
- ### βœ… Key Highlights
31
  - **Architecture**: Fine-tuned Vision Transformer (ViT) via Hugging Face `transformers`
32
  - **Dataset**: 140k balanced real/fake face images
33
  - **License**: [MIT](https://opensource.org/licenses/MIT) β€” free for research and commercial use
@@ -35,7 +35,7 @@ The result is a lightweight, high-accuracy detector optimized for **both SDXL an
35
 
36
  ---
37
 
38
- ## πŸ’» Quick Start
39
 
40
  ### Dependencies
41
  ```bash
@@ -98,7 +98,7 @@ if __name__ == "__main__":
98
  python predict.py --image path/to/image
99
  ```
100
 
101
- ## πŸ“Š Performance & Limitations
102
 
103
  > **Note**: Final test accuracy will be reported after full evaluation. Preliminary results show strong generalization on SDXL- and diffusion-based face forgeries.
104
 
@@ -109,13 +109,13 @@ python predict.py --image path/to/image
109
  - GAN-generated faces (e.g., StyleGAN2/3)
110
  - Label mapping:
111
  - `0` β†’ `"artificial"` (AI-generated / Deepfake)
112
- - `1` β†’ `"real"` (authentic human face)
113
 
114
  > ⚠️ This tool is **not a forensic proof**, but a probabilistic detector. Use responsibly.
115
 
116
  ---
117
 
118
- ## 🌱 Philosophy & Ethics
119
 
120
  This model is open-source because:
121
  - **Transparency** is essential in the fight against synthetic media.
@@ -126,7 +126,7 @@ As a developer from a vulnerable community, I believe AI safety tools must be **
126
 
127
  ---
128
 
129
- ## πŸ™Œ Acknowledgements
130
 
131
  - **Dataset**: [140k Real and Fake Faces](https://www.kaggle.com/datasets/xhlulu/140k-real-and-fake-faces) by xhlulu
132
  - **Framework**: [Hugging Face Transformers](https://huggingface.co/docs/transformers)
@@ -134,7 +134,7 @@ As a developer from a vulnerable community, I believe AI safety tools must be **
134
 
135
  ---
136
 
137
- ## πŸ“¬ How to Contribute
138
 
139
  Fine-tune this model on your domain-specific data using Hugging Face `Trainer`.
140
 
 
1
+ # SDXL-Deepfake-Detector
2
  ### Detecting AI-Generated Faces with Precision and Purpose
3
 
4
  >*Not just another classifier β€” a tool for digital truth.*
 
7
 
8
  ---
9
 
10
+ ## Why This Matters
11
  As generative AI (like SDXL, DALLΒ·E, and Midjourney) becomes more accessible, the line between real and synthetic media blurs β€” especially for vulnerable communities. This project started as a technical experiment but evolved into a **privacy-aware, open-source defense** against visual misinformation, with a focus on **ethical AI deployment**.
12
 
13
  ---
14
 
15
+ ## Model Overview
16
 
17
+ **SDXL-Deepfake-Detector** is a fine-tuned vision transformer that classifies human faces as **artificial (0)** or **human (1)**.
18
 
19
+ ## Training Approach
20
 
21
  This model was obtained by **fine-tuning** the [`Organika/sdxl-detector`](https://huggingface.co/Organika/sdxl-detector) β€” a vision transformer pre-trained specifically to detect SDXL-generated faces β€” on the [140k Real and Fake Faces](https://www.kaggle.com/datasets/xhlulu/140k-real-and-fake-faces) dataset.
22
 
 
27
 
28
  The result is a lightweight, high-accuracy detector optimized for **both SDXL and general diffusion-based deepfakes**.
29
 
30
+ ### Key Highlights
31
  - **Architecture**: Fine-tuned Vision Transformer (ViT) via Hugging Face `transformers`
32
  - **Dataset**: 140k balanced real/fake face images
33
  - **License**: [MIT](https://opensource.org/licenses/MIT) β€” free for research and commercial use
 
35
 
36
  ---
37
 
38
+ ## Quick Start
39
 
40
  ### Dependencies
41
  ```bash
 
98
  python predict.py --image path/to/image
99
  ```
100
 
101
+ ## Performance & Limitations
102
 
103
  > **Note**: Final test accuracy will be reported after full evaluation. Preliminary results show strong generalization on SDXL- and diffusion-based face forgeries.
104
 
 
109
  - GAN-generated faces (e.g., StyleGAN2/3)
110
  - Label mapping:
111
  - `0` β†’ `"artificial"` (AI-generated / Deepfake)
112
+ - `1` β†’ `"human"` (authentic human face)
113
 
114
  > ⚠️ This tool is **not a forensic proof**, but a probabilistic detector. Use responsibly.
115
 
116
  ---
117
 
118
+ ## Philosophy & Ethics
119
 
120
  This model is open-source because:
121
  - **Transparency** is essential in the fight against synthetic media.
 
126
 
127
  ---
128
 
129
+ ## Acknowledgements
130
 
131
  - **Dataset**: [140k Real and Fake Faces](https://www.kaggle.com/datasets/xhlulu/140k-real-and-fake-faces) by xhlulu
132
  - **Framework**: [Hugging Face Transformers](https://huggingface.co/docs/transformers)
 
134
 
135
  ---
136
 
137
+ ## How to Contribute
138
 
139
  Fine-tune this model on your domain-specific data using Hugging Face `Trainer`.
140