Update README.md
Browse files
README.md
CHANGED
@@ -8,7 +8,7 @@ license: apache-2.0
|
|
8 |
|
9 |
# Model Card for regnet_y_8g_intermediate-eu-common
|
10 |
|
11 |
-
RegNet Y 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 |
|
@@ -18,12 +18,12 @@ The species list is derived from the Collins bird guide [^1].
|
|
18 |
|
19 |
- **Model Type:** Image classification and detection backbone
|
20 |
- **Model Stats:**
|
21 |
-
|
22 |
-
|
23 |
- **Dataset:** eu-common (707 classes)
|
24 |
|
25 |
- **Papers:**
|
26 |
-
|
27 |
|
28 |
## Model Usage
|
29 |
|
@@ -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 regnet_y_8g_intermediate-eu-common
|
10 |
|
11 |
+
A RegNet Y 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 |
|
|
|
18 |
|
19 |
- **Model Type:** Image classification and detection backbone
|
20 |
- **Model Stats:**
|
21 |
+
- Params (M): 38.8
|
22 |
+
- Input image size: 384 x 384
|
23 |
- **Dataset:** eu-common (707 classes)
|
24 |
|
25 |
- **Papers:**
|
26 |
+
- Designing Network Design Spaces: <https://arxiv.org/abs/2003.13678>
|
27 |
|
28 |
## Model Usage
|
29 |
|
|
|
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
|