Update README.md
Browse files
README.md
CHANGED
|
@@ -17,6 +17,17 @@ This model is a **Multi-Task Image Classifier** that performs two tasks simultan
|
|
| 17 |
|
| 18 |
The model uses a **ResNet-50** backbone with two heads: one for multi-class object recognition and another for binary classification (AI-generated vs. Real). It was trained on a subset of the [Hemg/AI-Generated-vs-Real-Images-Datasets](https://huggingface.co/datasets/Hemg/AI-Generated-vs-Real-Images-Datasets) and leverages YOLO for improved pseudo-labeling across the entire dataset.
|
| 19 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 20 |
## Intended Use
|
| 21 |
This model is designed for:
|
| 22 |
- **Digital Content Verification:** Detecting AI-generated images to help prevent misinformation.
|
|
@@ -47,15 +58,7 @@ Alternatively, you can test the model directly via our interactive demo:
|
|
| 47 |
- **Object Recognition:** Pseudo-label accuracy started at around 38–40% and improved during training.
|
| 48 |
- **Evaluation:** Detailed evaluation metrics and loss curves are available in our training logs.
|
| 49 |
|
| 50 |
-
|
| 51 |
-
- **Trained by:** [Abdellahi El Moustapha](https://abmstpha.github.io/)
|
| 52 |
-
- **Programming Language:** Python
|
| 53 |
-
- **Base Model:** ResNet-50
|
| 54 |
-
- **Datasets:** Hemg/AI-Generated-vs-Real-Images-Datasets
|
| 55 |
-
- **Library:** PyTorch
|
| 56 |
-
- **Pipeline Tag:** image-classification
|
| 57 |
-
- **Metrics:** Accuracy for both binary classification and multi-class object recognition
|
| 58 |
-
- **Version:** v1.0
|
| 59 |
|
| 60 |
## Limitations and Ethical Considerations
|
| 61 |
- **Pseudo-Labeling:** The object recognition task uses pseudo-labels generated from a pretrained model, which may introduce noise or bias.
|
|
|
|
| 17 |
|
| 18 |
The model uses a **ResNet-50** backbone with two heads: one for multi-class object recognition and another for binary classification (AI-generated vs. Real). It was trained on a subset of the [Hemg/AI-Generated-vs-Real-Images-Datasets](https://huggingface.co/datasets/Hemg/AI-Generated-vs-Real-Images-Datasets) and leverages YOLO for improved pseudo-labeling across the entire dataset.
|
| 19 |
|
| 20 |
+
## Model Details
|
| 21 |
+
- **Trained by:** [Abdellahi El Moustapha](https://abmstpha.github.io/)
|
| 22 |
+
- **Programming Language:** Python
|
| 23 |
+
- **Base Model:** ResNet-50
|
| 24 |
+
- **Datasets:** Hemg/AI-Generated-vs-Real-Images-Datasets
|
| 25 |
+
- **Library:** PyTorch
|
| 26 |
+
- **Pipeline Tag:** image-classification
|
| 27 |
+
- **Metrics:** Accuracy for both binary classification and multi-class object recognition
|
| 28 |
+
- **Version:** v1.0
|
| 29 |
+
|
| 30 |
+
|
| 31 |
## Intended Use
|
| 32 |
This model is designed for:
|
| 33 |
- **Digital Content Verification:** Detecting AI-generated images to help prevent misinformation.
|
|
|
|
| 58 |
- **Object Recognition:** Pseudo-label accuracy started at around 38–40% and improved during training.
|
| 59 |
- **Evaluation:** Detailed evaluation metrics and loss curves are available in our training logs.
|
| 60 |
|
| 61 |
+
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 62 |
|
| 63 |
## Limitations and Ethical Considerations
|
| 64 |
- **Pseudo-Labeling:** The object recognition task uses pseudo-labels generated from a pretrained model, which may introduce noise or bias.
|