Update README.md
Browse files
README.md
CHANGED
|
@@ -8,7 +8,7 @@ license: apache-2.0
|
|
| 8 |
|
| 9 |
# Model Card for convnext_v2_tiny_intermediate-eu-common
|
| 10 |
|
| 11 |
-
ConvNext v2 image classification model. The model follows a two-stage training process: first undergoing intermediate training on a large-scale dataset containing diverse bird species from around the world, then fine-tuned specifically on the `eu-common` dataset containing common European bird species.
|
| 12 |
|
| 13 |
The species list is derived from the Collins bird guide [^1].
|
| 14 |
|
|
@@ -41,9 +41,9 @@ size = birder.get_size_from_signature(signature)
|
|
| 41 |
# Create an inference transform
|
| 42 |
transform = birder.classification_transform(size, rgb_stats)
|
| 43 |
|
| 44 |
-
image = "path/to/image.jpeg" # or a PIL image
|
| 45 |
(out, _) = infer_image(net, image, transform)
|
| 46 |
-
# out is a NumPy array with shape of (1, num_classes)
|
| 47 |
```
|
| 48 |
|
| 49 |
### Image Embeddings
|
|
|
|
| 8 |
|
| 9 |
# Model Card for convnext_v2_tiny_intermediate-eu-common
|
| 10 |
|
| 11 |
+
A ConvNext v2 image classification model. The model follows a two-stage training process: first undergoing intermediate training on a large-scale dataset containing diverse bird species from around the world, then fine-tuned specifically on the `eu-common` dataset containing common European bird species.
|
| 12 |
|
| 13 |
The species list is derived from the Collins bird guide [^1].
|
| 14 |
|
|
|
|
| 41 |
# Create an inference transform
|
| 42 |
transform = birder.classification_transform(size, rgb_stats)
|
| 43 |
|
| 44 |
+
image = "path/to/image.jpeg" # or a PIL image, must be loaded in RGB format
|
| 45 |
(out, _) = infer_image(net, image, transform)
|
| 46 |
+
# out is a NumPy array with shape of (1, num_classes), representing class probabilities.
|
| 47 |
```
|
| 48 |
|
| 49 |
### Image Embeddings
|