Commit
·
3c63c8f
1
Parent(s):
8b6ad61
Create README.md
Browse files
README.md
ADDED
|
@@ -0,0 +1,10 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
**Model Description:**
|
| 2 |
+
This model is a Resnet18 trained in Pytorch to classify human faces orientation (Flipped or not flipped).
|
| 3 |
+
**Dataset:**
|
| 4 |
+
The model is pretrained on ImageNet and then finetuned on LFWPeople dataset. LFWPeople is a dataset of human faces. The dataset is labelled as follows:
|
| 5 |
+
* Flipped image -> label = 1
|
| 6 |
+
* Not flipped -> label = 0
|
| 7 |
+
|
| 8 |
+
**Accuracy:**
|
| 9 |
+
The validation accuracy of the model is 98%
|
| 10 |
+
|