Abdu07 commited on
Commit
83c8828
·
verified ·
1 Parent(s): 962d6d5

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +12 -9
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
- ## Model Details
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.