Installation

Environment for SNR - C50 prediction

conda env create -f env_datamaker.yml
conda activate datamaker

Environment for VAD

conda env create -f env_pyannote.yml
conda activate pyannote_datamaker

To setup the project :

pip install .

Building the datasets for the VAD / SNR predictors

Here are the instruction to build the extended datasets for both training and testing the VAD / SNR prediction model.

Downloading the datasets - Librispeech

Librispeech 1000 - train set

LibriSpeech 1000, composed of multiple datasets of librispeech, is used as a train dataset.

Download train-clean-100, train-clean-360 and train-other-500 on this page.

Librispeech dev

The dev dataset is composed of other librispeech datasets.

Download dev-clean, dev-other, test-clean, test-other on this page.

Building the data

To build the data, you need to run the script: build_vad_datasets.py

python build_vad_datasets.py init $DATASET_NAME \
                             $OUTPUT_DIR_VAD_DATASET \
                             --root-in $DIR_DOWNLOADED_DATA

Downloading the reverb dataset

We use impulse datasets to perform a convincing reverberation. We used the MIT Acoustical Reverberation Scene Statistics Survey and EchoThief.

Then we split all the impulse responses into a train set and a dev set with a 80/20 ratio.

You can use the following script to do so :

python data_preparation/reverb_data_prep.py --dataset-path desired/path/to/reverb/dataset

CPC works with 16kHz audio files, but these reverb datasets have an higher sample rate. To convert them to 16kHz run build_vad_datasets again :

python build_vad_datasets.py init standard \
                             $OUTPUT_DIR_IR_DATASET \
                             --root-in $DIR_DOWNLOADED_DATA

Downloading the noise dataset

We use Audioset to contaminate Librispeech with noise.

First download the metadata eval_segments.csv, balanced_train_segments.csv and unbalanced_train_segments.csv here

Then, to download Audioset :

.data_preparation/audioset_download.sh metadata.csv

Launching pyannote on a dataset

Inference

To run a pyannote inference on a dataset, you can use the script vad_pyannote/launch_vad_pyannote.py:

python vad_pyannote/launch_vad_pyannote.py ${DATASET_DIR}/audio_16k \
                                            --file_extension .flac
                                            -o ${DATASET_DIR}/rttm_files

This script takes advantage of all available GPUs. You can launch it on scrum to deal efficiently with large dataset.

Apply diverse transformations the dataset

To transform your dataset, you will need to use build_vad_datasets.py as follow:

python build_vad_datasets.py transform $DATASET_NAME \
                             $OUTPUT_DIR_VAD_DATASET \
                             -o $OUTPUT_DIR_TRANSFORM \
                            --transforms [ TRANSFORM_COMBINATON ]

Silence extension

You can extend the silences of your dataset by using the following command :

python build_vad_datasets.py transform $DATASET_NAME \
                             $OUTPUT_DIR_VAD_DATASET \
                             --name sil \
                             -o $OUTPUT_DIR_TRANSFORM \
                             --transforms extend_sil \
                             --expand-silence-only \  # use this option if you want to expand only the existing silences
                             --target-share-sil 0.5 \

Noise Augmentation

You will need to audioset AUDIOSET. To launch the noise augmentation use build_vad_datasets.py as follow:

python build_vad_datasets.py transform $DATASET_NAME \
                             $OUTPUT_DIR_VAD_DATASET \
                             --name noise \
                             --transforms noise \
                             --dir-noise $MUSAN_DIR \
                             --ext-noise .wav \
                             -o $OUTPUT_DIR_TRANSFORM \

Reverb augmentation

First, you will need to download impulse datasets to perform a convincing reverberation. We used the MIT Acoustical Reverberation Scene Statistics Survey for the train set and EchoThief for the train set.

CPC works with 16kHz audio files, but these reverb datasets have an higher sample rate. To convert them to 16kHz run build_vad_datasets again :

python build_vad_datasets.py init standard \
                             $OUTPUT_DIR_IR_DATASET \
                             --root-in $DIR_DOWNLOADED_DATA

Then, to apply the reverberation:

python build_vad_datasets.py transform $DATASET_NAME \
                             $OUTPUT_DIR_VAD_DATASET \
                             --name reverb
                             -o $OUTPUT_DIR_TRANSFORM \
                            --transforms reverb \
                            --dir-impulse-response $OUTPUT_DIR_IR_DATASET

Combining several transformation

You can combine different transformations any way you want. For example, to run a peak normalization, followed by some reverb augmentation and finish with noise augmentation run:

python build_vad_datasets.py transform $DATASET_NAME \
                             $OUTPUT_DIR_VAD_DATASET \
                             --name combo \
                             -o $OUTPUT_DIR_TRANSFORM \
                            --transforms peaknorm reverb noise \
                            --dir-impulse-response $OUTPUT_DIR_IR_DATASET \
                            --dir-noise $MUSAN_DIR \
                            --ext-noise .wav

Segment a dataset into smaller segments

You can segment a dataset into smaller audio segments using build_vad_datasets.py:

python build_vad_datasets.py segment $DATASET_NAME \
                             $OUTPUT_DIR_VAD_DATASET \
                             -o $OUTPUT_DIR_SEGMENT \
                            -t target_size_segment

Additional ressources:

Google drive: https://drive.google.com/drive/folders/1XXc8526sIsfg6w8h7oOUF9fWC-9ap2Uu?usp=sharing

What's next ?

Fix noise augmentation :
- either AddNoise:
  - Add self.max_size_loaded (which remains constant and indicates the amount of noise data that is being loaded at once)
  - Add self.cumulated_duration (which is updated after each run of the call function and describes the cumulated duration of segments that have been corrupted with additive noise)
  - Once self.cumulated_duration reaches self.max_size_loaded, call to self.load_noise_db() that must load M segments of noise until self.max_sized_loaded is reached
- Or use Marvin's technics and pre-process noise, concatenate in four big files with cross fading, and use these.
AddNoise should call AddReverb to corrupt noise segments with reverberation
No need to apply VAD : Audioset already has the labels

entn-at / brouhaha-maker Goto Github PK

brouhaha-maker's Introduction