Blame view
egs/aspire/s5/local/multi_condition/reverberate_data_dir.sh
2.04 KB
8dcb6dfcb
first commit
|
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 |
#!/bin/bash
# Copyright 2014 Johns Hopkins University (Author: Vijayaditya Peddinti)
# 2015 Tom Ko
# Apache 2.0.
# This script processes generates multi-condition training data from clean data dir
# and directory with impulse responses and noises
. ./cmd.sh;
set -e
random_seed=0
snrs="20:10:15:5:0"
log_dir=exp/make_reverb
. ./path.sh;
. ./utils/parse_options.sh
if [ $# != 3 ]; then
echo "Usage: reverberate_data_dir_speed.sh [options] <src_dir> <impulse-noise-dir> <dest_dir>"
echo "e.g.:"
echo " $0 --random-seed 12 data/train_si284 data_multicondition/impulses_noises data/train_si284p"
exit 1;
fi
src_dir=$1
impnoise_dir=$2
dest_dir=$3
mkdir -p $dest_dir
mkdir -p $log_dir
wav_prefix="rev${random_seed}_"
utt_prefix="rev${random_seed}_"
spk_prefix="rev${random_seed}_"
# create the distorted wave files
utils/copy_data_dir.sh --spk-prefix "$spk_prefix" --utt-prefix "$utt_prefix" \
$src_dir $dest_dir
if [ ! -f $src_dir/utt2uniq ]; then
cat $src_dir/utt2spk | awk -v p=$utt_prefix '{printf("%s%s %s
", p, $1, $1);}' > $dest_dir/utt2uniq
else
cat $src_dir/utt2uniq | awk -v p=$utt_prefix '{printf("%s%s %s
", p, $1, $2);}' > $dest_dir/utt2uniq
fi
# create the wav.scp files
cat $src_dir/wav.scp | sed -e "s/^\s*//g" | \
cut -d' ' -f1 | \
awk -v p2=$wav_prefix \
'{printf("%s%s
", p2, $1);}'> $log_dir/corrupted_${random_seed}.list
# modify segments file to point to the new wav files
cat $dest_dir/segments | awk -v p=$wav_prefix \
'{printf("%s %s%s %s %s
", $1, p, $2, $3, $4);}' > $log_dir/segments_temp
mv $log_dir/segments_temp $dest_dir/segments
# remove these files as we would have to extract
# features for this new audio and out audio
# is single channel
for file in cmvn.scp feats.scp reco2file_and_channel; do
rm -f $dest_dir/$file
done
python local/multi_condition/reverberate_wavs.py \
--snrs $snrs --random-seed $random_seed \
$src_dir/wav.scp $log_dir/corrupted_${random_seed}.list $impnoise_dir \
$dest_dir/wav.scp || exit 1;
echo "Successfully generated corrupted data and stored it in $dest_dir." && exit 0;
|