Upload 2 files
Browse files- inference.py +4 -4
- taxonomy.csv +0 -0
inference.py
CHANGED
|
@@ -1,12 +1,12 @@
|
|
| 1 |
"""
|
| 2 |
-
Inference script for SPECIESNET-v4-0-
|
| 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.
|
| 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.
|
| 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
|
|
|