hassonofer commited on
Commit
99f69d3
·
verified ·
1 Parent(s): 2083d5e

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +6 -6
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
- - 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,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