hassonofer commited on
Commit
20111f4
·
verified ·
1 Parent(s): 6c9a5dc

Update README.md

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