Addax-Data-Science commited on
Commit
61ccfef
·
verified ·
1 Parent(s): 1588be4

Upload 2 files

Browse files
Files changed (2) hide show
  1. inference.py +4 -4
  2. taxonomy.csv +0 -0
inference.py CHANGED
@@ -1,12 +1,12 @@
1
  """
2
- Inference script for SPECIESNET-v4-0-1-A-v1 (SpeciesNet classifier)
3
 
4
  SpeciesNet is an image classifier designed to accelerate the review of images
5
  from camera traps. Trained at Google using a large dataset of camera trap images
6
  and an EfficientNet V2 M architecture. Classifies images into one of 2,498 labels
7
  covering diverse animal species, higher-level taxa, and non-animal classes.
8
 
9
- Model: SpeciesNet v4.0.1a (always_crop variant)
10
  Input: 480x480 RGB images (NHWC layout)
11
  Framework: PyTorch (torch.fx GraphModule)
12
  Classes: 2,498
@@ -36,8 +36,8 @@ ImageFile.LOAD_TRUNCATED_IMAGES = True
36
  if platform.system() != "Windows":
37
  pathlib.WindowsPath = pathlib.PosixPath
38
 
39
- # Hardcoded model parameters for SpeciesNet v4.0.1a
40
- LABELS_FILENAME = "always_crop_99710272_22x8_v12_epoch_00148.labels.txt"
41
  IMG_SIZE = 480
42
 
43
 
 
1
  """
2
+ Inference script for SPECIESNET-v4-0-2-A (SpeciesNet classifier)
3
 
4
  SpeciesNet is an image classifier designed to accelerate the review of images
5
  from camera traps. Trained at Google using a large dataset of camera trap images
6
  and an EfficientNet V2 M architecture. Classifies images into one of 2,498 labels
7
  covering diverse animal species, higher-level taxa, and non-animal classes.
8
 
9
+ Model: SpeciesNet v4.0.2a (always_crop variant)
10
  Input: 480x480 RGB images (NHWC layout)
11
  Framework: PyTorch (torch.fx GraphModule)
12
  Classes: 2,498
 
36
  if platform.system() != "Windows":
37
  pathlib.WindowsPath = pathlib.PosixPath
38
 
39
+ # Hardcoded model parameters for SpeciesNet v4.0.2a
40
+ LABELS_FILENAME = "always_crop_99710272_22x8_v12_epoch_00148.labels.20251208.txt"
41
  IMG_SIZE = 480
42
 
43
 
taxonomy.csv ADDED
The diff for this file is too large to render. See raw diff