Baseline

Python Modules

recipes.cad1.task2.baseline.audio_manager

A utility class for managing audio files.

recipes.cad1.task2.baseline.baseline_utils

Utility functions for the baseline model.

recipes.cad1.task2.baseline.car_scene_acoustics

A class for the car acoustic environment.

recipes.cad1.task2.baseline.enhance

Run the dummy enhancement.

recipes.cad1.task2.baseline.evaluate

Evaluate the enhanced signals using the HAAQI metric.

recipes.cad1.task2.baseline.merge_batches_results

Join batches scores into a single file.

recipes.cad1.task2.baseline.test

Run the dummy enhancement.

Configuration Files

config.yaml

 1path:
 2  root: ../../cadenza_task2_data_demo/cad1/task2
 3  audio_dir: ${path.root}/audio
 4  metadata_dir: ${path.root}/metadata
 5  music_dir: ${path.audio_dir}/music
 6  hrtf_dir: ${path.audio_dir}/eBrird
 7  listeners_file: ${path.metadata_dir}/listeners.valid.json
 8  scenes_file: ${path.metadata_dir}/scenes.json
 9  scenes_listeners_file: ${path.metadata_dir}/scenes_listeners.json
10  hrtf_file: ${path.metadata_dir}/eBrird_BRIR.json
11  exp_folder: ./exp # folder to store enhanced signals and final results
12
13team_id: T001
14
15sample_rate: 44100          # sample rate of the input signal
16enhanced_sample_rate: 32000 # sample rate for the enhanced output signal
17
18nalr:
19  nfir: 220
20  sample_rate: ${sample_rate}
21
22compressor:
23  threshold: 0.7
24  attenuation: 0.1
25  attack: 5
26  release: 20
27  rms_buffer_size: 0.064
28
29soft_clip: False
30
31enhance:
32  average_level: -14   # Average level according Spotify's levels
33  min_level: -19
34
35evaluate:
36  set_random_seed: True
37  small_test: False
38  save_intermediate_wavs: False
39  split: valid # train, valid
40  batch_size: 1   # Number of batches
41  batch: 0      # Batch number to evaluate
42
43# hydra config
44hydra:
45  run:
46    dir: ${path.exp_folder}
47  job:
48    chdir: True