avimittal30 commited on
Commit
5203ed4
·
1 Parent(s): f2aa636

adding read me

Browse files
Files changed (1) hide show
  1. README.md +9 -26
README.md CHANGED
@@ -1,26 +1,9 @@
1
- ## Human Emotion Detection
2
- A full fledged application that detects human emotion from an Image.
3
- In this project, I have finetuned Efficient model to achieve 80% accuracy on the validation dataset.
4
-
5
- This project uses human emotions dataset from Kaggle to finetune efficientnet model
6
-
7
- The application is created using flask framework and deployed on AWS EC2 instance . The docker image was pushed to ECR and pulled into EC2 instance using Github actions as a part of CI/CD pipeline implementation
8
-
9
- I tried using TransferNet approach and Finetuning approach. With Transfernet, the model accuracy was only 63% while it increased to 80% with a finetuned model
10
-
11
- ## Evaluation Metrics
12
-
13
- Loss: 0.5
14
- Accuracy: 80%
15
- Top_k_accuracy: 93%
16
-
17
- ### Example: How finetuning improved the model performance
18
-
19
- Before finetuning, this image was incorrecly labeled as Sad
20
-
21
- ![Incorrect labeling with transfer learning](TransferLearning.png)
22
-
23
- After finetuning, this image was correctly labeled as happy
24
-
25
- ![Correct labeling with finetuned model](Finetuning.png)
26
-
 
1
+ ---
2
+ title: Human Emotion Detection
3
+ emoji: 🧠
4
+ colorFrom: blue
5
+ colorTo: indigo
6
+ sdk: docker
7
+ app_file: app.py
8
+ pinned: false
9
+ ---