Upload 9 files
Browse filesadd model configs
- conf/augmentation/CustomAugPipeline_woAddSpec.yaml +15 -0
- conf/config.yaml +22 -0
- conf/preprocessing/log_mel_96k.yaml +15 -0
- file_handler.yaml +1 -0
- inference_transform.yaml +7 -0
- model.yaml +5 -0
- preprocess_file_handler.yaml +2 -0
- preprocess_pipeline.yaml +15 -0
- target_transform.yaml +33 -0
conf/augmentation/CustomAugPipeline_woAddSpec.yaml
ADDED
|
@@ -0,0 +1,15 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
id: CustomAugPipeline_woAddSpec
|
| 2 |
+
_target_: autrainer.augmentations.AugmentationPipeline
|
| 3 |
+
|
| 4 |
+
generator_seed: 0
|
| 5 |
+
|
| 6 |
+
pipeline:
|
| 7 |
+
- autrainer.augmentations.Sequential:
|
| 8 |
+
sequence:
|
| 9 |
+
- autrainer.augmentations.Choice:
|
| 10 |
+
weights: [0.3, 0.7]
|
| 11 |
+
choices:
|
| 12 |
+
- autrainer.augmentations.GaussianNoise:
|
| 13 |
+
p: 0.8
|
| 14 |
+
- autrainer.augmentations.SpecAugment:
|
| 15 |
+
p: 0.8
|
conf/config.yaml
ADDED
|
@@ -0,0 +1,22 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
defaults:
|
| 2 |
+
- _autrainer_
|
| 3 |
+
- _self_
|
| 4 |
+
|
| 5 |
+
|
| 6 |
+
results_dir: results
|
| 7 |
+
experiment_id: default
|
| 8 |
+
iterations: 30
|
| 9 |
+
inference_batch_size: 1
|
| 10 |
+
progress_bar: false
|
| 11 |
+
|
| 12 |
+
hydra:
|
| 13 |
+
sweeper:
|
| 14 |
+
params:
|
| 15 |
+
+seed: 1
|
| 16 |
+
+batch_size: 32
|
| 17 |
+
+learning_rate: 0.001
|
| 18 |
+
dataset: INSECTNET-96k-ML-Balanced
|
| 19 |
+
model: Cnn10-32k-T
|
| 20 |
+
optimizer: Adam
|
| 21 |
+
augmentation: CustomAugPipeline_woAddSpec
|
| 22 |
+
|
conf/preprocessing/log_mel_96k.yaml
ADDED
|
@@ -0,0 +1,15 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
file_handler:
|
| 2 |
+
autrainer.datasets.utils.AudioFileHandler:
|
| 3 |
+
target_sample_rate: 96000
|
| 4 |
+
pipeline:
|
| 5 |
+
- autrainer.transforms.StereoToMono
|
| 6 |
+
- autrainer.transforms.PannMel:
|
| 7 |
+
sample_rate: 96000
|
| 8 |
+
window_size: 4096
|
| 9 |
+
hop_size: 1024
|
| 10 |
+
mel_bins: 64
|
| 11 |
+
fmin: 50
|
| 12 |
+
fmax: 48000
|
| 13 |
+
ref: 1.0
|
| 14 |
+
amin: 1e-10
|
| 15 |
+
top_db: null
|
file_handler.yaml
ADDED
|
@@ -0,0 +1 @@
|
|
|
|
|
|
|
| 1 |
+
$autrainer.datasets.utils.file_handlers.NumpyFileHandler==0.6.0: {}
|
inference_transform.yaml
ADDED
|
@@ -0,0 +1,7 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
$autrainer.transforms.smart_compose.SmartCompose==0.6.0:
|
| 2 |
+
transforms:
|
| 3 |
+
- $autrainer.transforms.specific_transforms.Expand==0.6.0:
|
| 4 |
+
size: 376
|
| 5 |
+
method: pad
|
| 6 |
+
axis: -2
|
| 7 |
+
order: -85
|
model.yaml
ADDED
|
@@ -0,0 +1,5 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
$autrainer.models.cnn_10.Cnn10==0.6.0:
|
| 2 |
+
output_dim: 30
|
| 3 |
+
segmentwise: false
|
| 4 |
+
in_channels: 1
|
| 5 |
+
transfer: https://zenodo.org/records/3987831/files/Cnn10_mAP%3D0.380.pth
|
preprocess_file_handler.yaml
ADDED
|
@@ -0,0 +1,2 @@
|
|
|
|
|
|
|
|
|
|
| 1 |
+
$autrainer.datasets.utils.file_handlers.AudioFileHandler==0.6.0:
|
| 2 |
+
target_sample_rate: 96000
|
preprocess_pipeline.yaml
ADDED
|
@@ -0,0 +1,15 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
$autrainer.transforms.smart_compose.SmartCompose==0.6.0:
|
| 2 |
+
transforms:
|
| 3 |
+
- $autrainer.transforms.specific_transforms.StereoToMono==0.6.0:
|
| 4 |
+
order: -95
|
| 5 |
+
- $autrainer.transforms.specific_transforms.PannMel==0.6.0:
|
| 6 |
+
window_size: 4096
|
| 7 |
+
hop_size: 1024
|
| 8 |
+
sample_rate: 96000
|
| 9 |
+
fmin: 50
|
| 10 |
+
fmax: 48000
|
| 11 |
+
mel_bins: 64
|
| 12 |
+
ref: 1.0
|
| 13 |
+
amin: 1.0e-10
|
| 14 |
+
top_db: null
|
| 15 |
+
order: -90
|
target_transform.yaml
ADDED
|
@@ -0,0 +1,33 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
$autrainer.datasets.utils.target_transforms.multi_label_encoder.MultiLabelEncoder==0.6.0:
|
| 2 |
+
threshold: 0.5
|
| 3 |
+
labels:
|
| 4 |
+
- Bicolorana bicolor
|
| 5 |
+
- Chorthippus albomarginatus
|
| 6 |
+
- Chorthippus apricarius
|
| 7 |
+
- Chorthippus biguttulus
|
| 8 |
+
- Chorthippus brunneus
|
| 9 |
+
- Chorthippus dorsatus
|
| 10 |
+
- Chorthippus mollis
|
| 11 |
+
- Chrysochraon dispar
|
| 12 |
+
- Conocephalus dorsalis
|
| 13 |
+
- Conocephalus fuscus
|
| 14 |
+
- Euthystira brachyptera
|
| 15 |
+
- Gomphocerippus rufus
|
| 16 |
+
- Gryllotalpa gryllotalpa
|
| 17 |
+
- Gryllus campestris
|
| 18 |
+
- Isophya kraussii
|
| 19 |
+
- Leptophyes punctatissima
|
| 20 |
+
- Nemobius sylvestris
|
| 21 |
+
- Omocestus viridulus
|
| 22 |
+
- Phaneroptera falcata
|
| 23 |
+
- Pholidoptera griseoaptera
|
| 24 |
+
- Platycleis albopunctata
|
| 25 |
+
- Pseudochorthippus montanus
|
| 26 |
+
- Pseudochorthippus parallelus
|
| 27 |
+
- Roeseliana roeselii
|
| 28 |
+
- Ruspolia nitidula
|
| 29 |
+
- Stenobothrus lineatus
|
| 30 |
+
- Stethophyma grossum
|
| 31 |
+
- Tettigonia cantans
|
| 32 |
+
- Tettigonia viridissima
|
| 33 |
+
- nocall-empty
|