Update README.md
Browse files
README.md
CHANGED
@@ -46,7 +46,7 @@ Transformer-based vision model for pokemon image classification.
|
|
46 |
|
47 |
This model is intended to classify between pokemons from the 1st Generation. Therefore, when provided with images of pokemon from posterior generation, the model outputs won't be usable as such.
|
48 |
Moreover, the model was not designed to handle non pokemon images as well as images presenting several entities.
|
49 |
-
However, an additional layer can help mitigate the risk of wrongly classifying non pokemon images by analyzing the spread of the output (the confusion of the model), such a layer can be found in my implementation, available [here]
|
50 |
|
51 |
## Training procedure
|
52 |
|
|
|
46 |
|
47 |
This model is intended to classify between pokemons from the 1st Generation. Therefore, when provided with images of pokemon from posterior generation, the model outputs won't be usable as such.
|
48 |
Moreover, the model was not designed to handle non pokemon images as well as images presenting several entities.
|
49 |
+
However, an additional layer can help mitigate the risk of wrongly classifying non pokemon images by analyzing the spread of the output (the confusion of the model), such a layer can be found in my implementation, available [here](https://github.com/A-Duss/GottaClassifyEmAll).
|
50 |
|
51 |
## Training procedure
|
52 |
|