Update README.md
Browse files
README.md
CHANGED
|
@@ -74,6 +74,4 @@ model.eval()
|
|
| 74 |
## ⚠️ Limitations and Considerations
|
| 75 |
Input Dimensions: Make sure your input images are resized to the expected dimensions (100x100) before inference.
|
| 76 |
Number of Classes: The trained model supports exactly 7 classes as defined in the training dataset.
|
| 77 |
-
|
| 78 |
-
|
| 79 |
-
|
|
|
|
| 74 |
## ⚠️ Limitations and Considerations
|
| 75 |
Input Dimensions: Make sure your input images are resized to the expected dimensions (100x100) before inference.
|
| 76 |
Number of Classes: The trained model supports exactly 7 classes as defined in the training dataset.
|
| 77 |
+
Output: The model output should be a probability of each of the 7 face type labels. Don't forget to use the softmax function to make predictions. Note that softmax is not used in the last layer of this model's architecture.
|
|
|
|
|
|