Update README.md
Browse files
README.md
CHANGED
|
@@ -15,6 +15,8 @@ This is a model used to differentiate between three types of insects:
|
|
| 15 |
|
| 16 |
It was created as part of the end-to-end Deep Learning Project learning process.
|
| 17 |
|
|
|
|
|
|
|
| 18 |
## Intended uses & limitations
|
| 19 |
This model was trained on roughly 150 pictures of each category and performed well. However, it was not vigorously tested and so may perform badly on some edge cases images and may contain bias e.g. when training, I noticed that most images of caterpillars were next to leaves and plantation, so it may have learned to associate that environment with a caterpillar.
|
| 20 |
|
|
|
|
| 15 |
|
| 16 |
It was created as part of the end-to-end Deep Learning Project learning process.
|
| 17 |
|
| 18 |
+
The model is a pretrained `convnext_tiny_in22k` from the [timm library](https://github.com/rwightman/pytorch-image-models) fine-tuned on the new dataset.
|
| 19 |
+
|
| 20 |
## Intended uses & limitations
|
| 21 |
This model was trained on roughly 150 pictures of each category and performed well. However, it was not vigorously tested and so may perform badly on some edge cases images and may contain bias e.g. when training, I noticed that most images of caterpillars were next to leaves and plantation, so it may have learned to associate that environment with a caterpillar.
|
| 22 |
|