mirror of
https://github.com/macaodha/batdetect2.git
synced 2025-06-29 14:41:58 +02:00
Add comprehensive conf file
This commit is contained in:
parent
136949c4e7
commit
152a577511
122
config.yaml
122
config.yaml
@ -8,3 +8,125 @@ datasets:
|
|||||||
description: Examples included for testing batdetect2
|
description: Examples included for testing batdetect2
|
||||||
annotations_dir: example_data/anns
|
annotations_dir: example_data/anns
|
||||||
audio_dir: example_data/audio
|
audio_dir: example_data/audio
|
||||||
|
|
||||||
|
targets:
|
||||||
|
classes:
|
||||||
|
classes:
|
||||||
|
- name: myomys
|
||||||
|
tags:
|
||||||
|
- value: Myotis mystacinus
|
||||||
|
- name: pippip
|
||||||
|
tags:
|
||||||
|
- value: Pipistrellus pipistrellus
|
||||||
|
- name: eptser
|
||||||
|
tags:
|
||||||
|
- value: Eptesicus serotinus
|
||||||
|
- name: rhifer
|
||||||
|
tags:
|
||||||
|
- value: Rhinolophus ferrumequinum
|
||||||
|
generic_class:
|
||||||
|
- key: class
|
||||||
|
value: Bat
|
||||||
|
|
||||||
|
filtering:
|
||||||
|
rules:
|
||||||
|
- match_type: all
|
||||||
|
tags:
|
||||||
|
- key: event
|
||||||
|
value: Echolocation
|
||||||
|
- match_type: exclude
|
||||||
|
tags:
|
||||||
|
- key: class
|
||||||
|
value: Unknown
|
||||||
|
|
||||||
|
preprocess:
|
||||||
|
audio:
|
||||||
|
resample:
|
||||||
|
samplerate: 256000
|
||||||
|
method: "poly"
|
||||||
|
scale: false
|
||||||
|
center: true
|
||||||
|
duration: null
|
||||||
|
|
||||||
|
spectrogram:
|
||||||
|
stft:
|
||||||
|
window_duration: 0.002
|
||||||
|
window_overlap: 0.75
|
||||||
|
window_fn: hann
|
||||||
|
frequencies:
|
||||||
|
max_freq: 120000
|
||||||
|
min_freq: 10000
|
||||||
|
pcen:
|
||||||
|
time_constant: 0.4
|
||||||
|
gain: 0.98
|
||||||
|
bias: 2
|
||||||
|
power: 0.5
|
||||||
|
scale: "amplitude"
|
||||||
|
size:
|
||||||
|
height: 128
|
||||||
|
resize_factor: 0.5
|
||||||
|
spectral_mean_substraction: true
|
||||||
|
peak_normalize: false
|
||||||
|
|
||||||
|
postprocess:
|
||||||
|
nms_kernel_size: 9
|
||||||
|
detection_threshold: 0.01
|
||||||
|
min_freq: 10000
|
||||||
|
max_freq: 120000
|
||||||
|
top_k_per_sec: 200
|
||||||
|
|
||||||
|
labels:
|
||||||
|
sigma: 3
|
||||||
|
|
||||||
|
model:
|
||||||
|
name: Net2DFast
|
||||||
|
input_height: 128
|
||||||
|
encoder_channels: [1, 32, 64, 128]
|
||||||
|
bottleneck_channels: 256
|
||||||
|
decoder_channels: [256, 64, 32, 32]
|
||||||
|
out_channels: 32
|
||||||
|
|
||||||
|
train:
|
||||||
|
batch_size: 8
|
||||||
|
loss:
|
||||||
|
detection:
|
||||||
|
weight: 1.0
|
||||||
|
focal:
|
||||||
|
beta: 4
|
||||||
|
alpha: 2
|
||||||
|
classification:
|
||||||
|
weight: 2.0
|
||||||
|
focal:
|
||||||
|
beta: 4
|
||||||
|
alpha: 2
|
||||||
|
size:
|
||||||
|
weight: 0.1
|
||||||
|
optimizer:
|
||||||
|
learning_rate: 0.001
|
||||||
|
t_max: 100
|
||||||
|
augmentations:
|
||||||
|
steps:
|
||||||
|
- augmentation_type: mix
|
||||||
|
probability: 0.2
|
||||||
|
min_weight: 0.3
|
||||||
|
max_weight: 0.7
|
||||||
|
- agumentation_type: echo
|
||||||
|
probability: 0.2
|
||||||
|
max_delay: 0.005
|
||||||
|
min_weight: 0.0
|
||||||
|
max_weight: 1.0
|
||||||
|
- augmentation_type: volume
|
||||||
|
probability: 0.2
|
||||||
|
min_scaling: 0.0
|
||||||
|
max_scaling: 2.0
|
||||||
|
- augmentation_type: warp
|
||||||
|
probability: 0.2
|
||||||
|
delta: 0.04
|
||||||
|
- augmentation_type: time_mask
|
||||||
|
probability: 0.2
|
||||||
|
max_perc: 0.05
|
||||||
|
max_masks: 3
|
||||||
|
- augmentation_type: frequency_mask
|
||||||
|
probability: 0.2
|
||||||
|
max_perc: 0.10
|
||||||
|
max_masks: 3
|
||||||
|
Loading…
Reference in New Issue
Block a user