OuiAhmed commited on
Commit
effa125
·
verified ·
1 Parent(s): 99b9225

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +21 -1
README.md CHANGED
@@ -6,4 +6,24 @@ base_model:
6
  - microsoft/resnet-50
7
  pipeline_tag: image-classification
8
  library_name: keras
9
- ---
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
6
  - microsoft/resnet-50
7
  pipeline_tag: image-classification
8
  library_name: keras
9
+ ---
10
+
11
+
12
+ This model is a ResNet-50 deep convolutional neural network fine-tuned for the FER-2013 (Facial Expression Recognition 2013) dataset. The dataset consists of low-resolution (48×48) grayscale images of faces categorized into seven core emotional states.
13
+
14
+ This project focused on maximizing the performance of the pre-trained ResNet-50 architecture on this particularly challenging, noisy, and imbalanced dataset.
15
+ Training Details
16
+ Architecture
17
+
18
+ Base Model: ResNet-50 (pre-trained on ImageNet).
19
+
20
+ Head: Custom dense layers (224 units) with a high 0.5 dropout rate.
21
+
22
+ Transfer Learning Strategy: Deep Freezing. The model base was frozen up to the conv5 block, meaning only the final convolutional block (conv5) and the custom head were fine-tuned. This prevents early layers, which are optimized for high-resolution images, from being corrupted by the 48×48 input.
23
+
24
+ Optimization & Regularization
25
+ Technique Rationale
26
+ Class Weighting Applied inverse frequency weights to mitigate the severe class imbalance (e.g., Disgust is rare, Happy is abundant).
27
+ Data Augmentation Used random flips, translations, rotations, and zooms to artificially expand the small dataset and combat overfitting.
28
+ High Dropout Increased dropout to 0.5 to aggressively regularize the model and prevent the divergence seen in earlier training runs.
29
+ Optimizer Adam with a very low fine-tuning learning rate of 5e−6.