Baseline

Python Modules

recipes.cad2.task2.baseline.enhance

Run the dummy enhancement.

recipes.cad2.task2.baseline.evaluate

Evaluate the enhanced signals using the HAAQI metric.

recipes.cad2.task2.baseline.merge_batches_results

Join batches scores into a single file.

Configuration Files

config.yaml

 1path:
 2  root: ???  # Set to the root of the dataset
 3  metadata_dir: ${path.root}/metadata
 4  music_dir: ${path.root}/audio
 5  gains_file: ${path.metadata_dir}/gains.json
 6  listeners_file: ${path.metadata_dir}/listeners.valid.json
 7  enhancer_params_file: ${path.metadata_dir}/compressor_params.valid.json
 8  music_file: ${path.metadata_dir}/music.valid.json
 9  scenes_file: ${path.metadata_dir}/scenes.valid.json
10  scene_listeners_file: ${path.metadata_dir}/scene_listeners.valid.json
11  exp_folder: ./exp # folder to store enhanced signals and final results
12
13input_sample_rate: 44100
14remix_sample_rate: 32000
15HAAQI_sample_rate: 24000
16
17separator:
18  force_redownload: True
19  add_residual: 0.1
20  causality: noncausal
21  device: ~
22  separation:
23    number_sources: 2
24    segment: 6.0
25    overlap: 0.1
26    sample_rate: ${input_sample_rate}
27
28enhancer:
29  crossover_frequencies: [353.55, 707.11, 1414.21, 2828.43, 5656.85] # [250, 500, 1000, 2000, 4000] * sqrt(2)
30  attack: [11, 11, 14, 13, 11, 11]
31  release: [80, 80, 80, 80, 100, 100]
32  threshold: [-30, -30, -30, -30, -30, -30]
33
34soft_clip: False
35
36evaluate:
37  set_random_seed: True
38  small_test: False
39  batch_size: 1  # Number of batches
40  batch: 0       # Batch number to evaluate
41
42# hydra config
43hydra:
44  run:
45    dir: ${path.exp_folder}
46  job:
47    chdir: True