Update README.md
Browse files
README.md
CHANGED
|
@@ -1,29 +1,45 @@
|
|
| 1 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 2 |
|
| 3 |
-
|
| 4 |
|
| 5 |
-
|
| 6 |
|
| 7 |
-
|
| 8 |
|
| 9 |
## π Model Description
|
| 10 |
|
| 11 |
-
The **SDXL-Deepfake-Detector** is a specialized image classification model
|
| 12 |
|
| 13 |
### Key Features & Performance
|
| 14 |
|
| 15 |
-
|
|
| 16 |
| :--- | :--- | :--- |
|
| 17 |
-
| **Task** | Binary Image Classification | Real Face vs. Deepfake Face |
|
| 18 |
-
| **Dataset
|
| 19 |
-
| **Test Accuracy** | **0.91** (91%) |
|
| 20 |
-
| **Hardware
|
| 21 |
|
| 22 |
---
|
| 23 |
|
| 24 |
## π» Usage with Hugging Face Transformers
|
| 25 |
|
| 26 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 27 |
|
| 28 |
---
|
| 29 |
license: mit
|
|
|
|
| 1 |
+
---
|
| 2 |
+
tags:
|
| 3 |
+
- image-classification
|
| 4 |
+
- computer-vision
|
| 5 |
+
- deepfake-detection
|
| 6 |
+
- fine-tuned
|
| 7 |
+
license: mit
|
| 8 |
+
datasets:
|
| 9 |
+
- 140k-real-and-fake-faces
|
| 10 |
+
metrics:
|
| 11 |
+
- accuracy
|
| 12 |
+
---
|
| 13 |
|
| 14 |
+
# π SDXL-Deepfake-Detector
|
| 15 |
|
| 16 |
+
**A high-performance deep learning model fine-tuned for the binary classification of real vs. fake faces, designed for images potentially generated by advanced synthesis models.**
|
| 17 |
|
| 18 |
+
This project was developed by **[Sadra Milani Moghadam](https://sadramilani.ir/)**.
|
| 19 |
|
| 20 |
## π Model Description
|
| 21 |
|
| 22 |
+
The **SDXL-Deepfake-Detector** is a specialized image classification model engineered to distinguish between authentic human faces and synthetically generated (deepfake) faces. The model was trained using **transfer learning** principles to provide robust and reliable detection.
|
| 23 |
|
| 24 |
### Key Features & Performance
|
| 25 |
|
| 26 |
+
| Metric | Value | Notes |
|
| 27 |
| :--- | :--- | :--- |
|
| 28 |
+
| **Task** | Binary Image Classification | Real Face (0) vs. Deepfake Face (1) |
|
| 29 |
+
| **Dataset** | [140K Real and Fake Faces (Kaggle)](https://www.kaggle.com/datasets/xhlulu/140k-real-and-fake-faces) | A large dataset containing 140,000 high-quality images. |
|
| 30 |
+
| **Test Accuracy** | **0.91** (91%) | Achieved on the independent test set. |
|
| 31 |
+
| **Hardware** | NVIDIA RTX 3060 (12GB VRAM) | Training performed on suitable hardware for efficiency. |
|
| 32 |
|
| 33 |
---
|
| 34 |
|
| 35 |
## π» Usage with Hugging Face Transformers
|
| 36 |
|
| 37 |
+
To load and use the model in your Python environment, use the standard `transformers` and `PIL` libraries.
|
| 38 |
+
|
| 39 |
+
### Installation
|
| 40 |
+
|
| 41 |
+
```bash
|
| 42 |
+
pip install transformers torch pillow
|
| 43 |
|
| 44 |
---
|
| 45 |
license: mit
|