Update README.md
Browse files
README.md
CHANGED
|
@@ -8,7 +8,8 @@ metrics:
|
|
| 8 |
This model is built on the efficientnet_b2 architecture.
|
| 9 |
The model uses pretrained weights of the model found in the torchvision.models library.
|
| 10 |
The classification head was changed, to be a dropout layer, followed by a linear layer with 6 target classes.
|
| 11 |
-
Using transfer learning, the model was then trained on the [intel image dataset.](https://www.kaggle.com/datasets/puneet6060/intel-image-classification)
|
|
|
|
| 12 |
|
| 13 |
#### Performance
|
| 14 |
The model achieved a test accuracy of 89,67%.
|
|
|
|
| 8 |
This model is built on the efficientnet_b2 architecture.
|
| 9 |
The model uses pretrained weights of the model found in the torchvision.models library.
|
| 10 |
The classification head was changed, to be a dropout layer, followed by a linear layer with 6 target classes.
|
| 11 |
+
Using transfer learning, the model was then trained on the [intel image dataset.](https://www.kaggle.com/datasets/puneet6060/intel-image-classification)
|
| 12 |
+
See the corresponding [hugging face space](https://huggingface.co/spaces/richardschattner/scenery_classification) for a live demo of the model.
|
| 13 |
|
| 14 |
#### Performance
|
| 15 |
The model achieved a test accuracy of 89,67%.
|