InsectNet for the Biodiversity Exploratories
Model that tags audio files as belonging to one or more of 29 (t.b.d. below) prevalent Orthoptera species within the Biodiversity Exploratories.
Installation
To use the model, you have to install autrainer, e.g. via pip:
pip install autrainer
This model has been trained and tested with autrainer version 0.6.0.
For more information about autrainer, please refer to: https://autrainer.github.io/autrainer/index.html
Usage
The model can be applied on all wav files present in a folder (<data-root>) and stored in another folder (<output-root>):
autrainer inference hf:HearTheSpecies/InsectNet-BE -r <data-root> <output-root> -w 4 -s 4 -sr 96000
, where -w is the window size in seconds, -s is the step size in seconds and -sr is the sampling rate.
For other possible inference settings and all usable parameters, please have a look at the autrainer documentation.
However, the above settings are recommended.
Training
Pretraining
TODO
Dataset
TODO
Features
The audio recordings were resampled to 96kHz, as we wanted to avoid losing too much frequency information from the species. Log-Mel spectrograms were then extracted using torchlibrosa.
Training process
The model has been trained for 30 epochs. At the end of each epoch, the model was evaluated on our validation set.
We release the state that achieved the best performance on this validation set.
All training hyperparameters can be found inside conf/config.yaml inside the model folder.
Evaluation
The performance on the test set reached a (macro) f1-score of 0.70.
Acknowledgments
TODO
Please acknowledge the work which produced the original model. We would appreciate an acknowledgment to autrainer.