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.
|