AlexanderGbd commited on
Commit
6510918
·
verified ·
1 Parent(s): c4c3645

Upload 9 files

Browse files

add model configs

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