Generate Dataset

Python Modules

recipes.cad_icassp_2024.generate_dataset.generate_at_mic_musdb18

Script creates the dataset for the ICASSP 2024 Grand Challenge.

recipes.cad_icassp_2024.generate_dataset.generate_train_scenes

Module to generate the scenes and scene-listeners metadata files for training

Configuration Files

config.yaml

 1path:
 2  root: ???
 3  metadata_dir: ${path.root}/metadata
 4  music_dir: ${path.root}/audio/music # musdb18 dataset
 5  hrtf_dir: ${path.root}/audio/hrtf
 6  scene_file: ${path.metadata_dir}/scenes.train.json
 7  music_file: ${path.metadata_dir}/musdb18.train.json
 8  head_loudspeaker_positions_file: ${path.metadata_dir}/head_loudspeaker_positions.json
 9  tracklist_file: ./musdb18_tracklist.csv
10  gains_file: ${path.metadata_dir}/gains.json
11  output_music_dir: ${path.root}/audio/at_mic_music # at microphone musdb18 dataset
12  output_music_file: ${path.metadata_dir}/at_mic_music.train.json
13
14sample_rate: 44100
15
16scene:
17  number_scenes_per_song: 4
18
19scene_listener:
20  number_listeners_per_scene: 2
21
22hydra:
23  run:
24    dir: .
25  job:
26    chdir: True