Update README.md
Browse files
README.md
CHANGED
|
@@ -9,7 +9,7 @@ tags:
|
|
| 9 |
|
| 10 |
# Baseline model for audio classification of orthopera and hemiptera
|
| 11 |
|
| 12 |
-
This baseline model, utilized in the ECOSoundSet paper
|
| 13 |
|
| 14 |
## Installation
|
| 15 |
|
|
@@ -25,7 +25,19 @@ The model can be applied on all wav files present in a folder (`<data-root>`) an
|
|
| 25 |
|
| 26 |
`autrainer inference hf:AlexanderGbd/insects-base-cnn10-96k-t -r <data-root> <output-root>`
|
| 27 |
|
| 28 |
-
For possible inference settings (e.g. using sliding window) and all usable parameters, please have a look at the autrainer documentation.
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 29 |
|
| 30 |
|
| 31 |
## Training
|
|
@@ -36,8 +48,7 @@ The model has been originally trained on AudioSet by Kong et. al. (https://ieeex
|
|
| 36 |
|
| 37 |
### Dataset
|
| 38 |
|
| 39 |
-
The model has been further trained (finetuned) on 4s long audio segments of the
|
| 40 |
-
and will be referenced here as soon as it is publicly available.
|
| 41 |
|
| 42 |
|
| 43 |
### Features
|
|
|
|
| 9 |
|
| 10 |
# Baseline model for audio classification of orthopera and hemiptera
|
| 11 |
|
| 12 |
+
This baseline model, utilized in the [ECOSoundSet paper](https://arxiv.org/abs/2504.20776), was trained to tag audio files with one or more of 86 species from the Orthoptera and Hemiptera insect orders.
|
| 13 |
|
| 14 |
## Installation
|
| 15 |
|
|
|
|
| 25 |
|
| 26 |
`autrainer inference hf:AlexanderGbd/insects-base-cnn10-96k-t -r <data-root> <output-root>`
|
| 27 |
|
| 28 |
+
For possible inference settings (e.g. using sliding window) and all usable parameters, please have a look at the [autrainer documentation](https://autrainer.github.io/autrainer/usage/cli_reference.html#autrainer-inference).
|
| 29 |
+
|
| 30 |
+
The following model usage is recommended:
|
| 31 |
+
|
| 32 |
+
```autrainer inference hf:AlexanderGbd/insects-base-cnn10-96k-t /path/to/input/data /path/to/store/results -r -w 4 -s 4 -sr 96000```
|
| 33 |
+
|
| 34 |
+
Here:
|
| 35 |
+
- r: recursive flag (recursively go through all subfolders as well).
|
| 36 |
+
- w: window size (here 4 seconds), i.e. length of the audio segment analyzed (in seconds).
|
| 37 |
+
- s: step size (here 4 seconds), i.e. how far the window moves forward after each analysis (in seconds).
|
| 38 |
+
- sr: target sampling rate (in Hz). If audio files don’t match this rate, they’re resampled.
|
| 39 |
+
- e: file extension (default is wav), i.e. the file format to process.
|
| 40 |
+
|
| 41 |
|
| 42 |
|
| 43 |
## Training
|
|
|
|
| 48 |
|
| 49 |
### Dataset
|
| 50 |
|
| 51 |
+
The model has been further trained (finetuned) on 4s long audio segments of the first version of the ECOSoundSet dataset.
|
|
|
|
| 52 |
|
| 53 |
|
| 54 |
### Features
|