|
| 1 | +#!/bin/bash |
| 2 | + |
| 3 | +## Only run this file from the example root directory |
| 4 | +## $ ./local/data_prep.sh |
| 5 | + |
| 6 | +mkdir -p "data/train" "data/test" "data/local" |
| 7 | + |
| 8 | +source ./path.sh |
| 9 | + |
| 10 | +# Dimex100 unziped corpus root directory |
| 11 | +CORPUS_DIR="$1" |
| 12 | + |
| 13 | +# Corpus data |
| 14 | +# |
| 15 | +# Number of Different speakers: 100 |
| 16 | +# Speakers common utterances: 10 |
| 17 | +# Speakers individual utterances: 50 |
| 18 | +# |
| 19 | +# Training/testing split |
| 20 | +# |
| 21 | +# Common utterances for training: 10 (100%) |
| 22 | +# Individual utterances for training: 40 (80%) |
| 23 | +# Individual utterances for testing: 10 (20%) |
| 24 | +N_SPEAKERS=100 |
| 25 | +N_COMMON_UTTERANCES=10 |
| 26 | +N_INDIVIDUAL_UTTERANCES=50 |
| 27 | +N_INDIVIDUAL_UTTERANCES_TRAINING=40 |
| 28 | +N_INDIVIDUAL_UTTERANCES_TESTING=10 |
| 29 | + |
| 30 | +# speakerId-utteranceId-[c|i] |
| 31 | +# c = speaker common utterances (10) |
| 32 | +# i = speaker individual utterances (50) |
| 33 | +# |
| 34 | +# e.g.: |
| 35 | +# s001-01-c |
| 36 | +# ... |
| 37 | +# s001-10-c |
| 38 | +# ... |
| 39 | +# s001-01-i |
| 40 | +# ... |
| 41 | +# s001-50-i |
| 42 | + |
| 43 | +## 80-20 train-test split |
| 44 | +## Only individual utterances are used in testing |
| 45 | +# 10/10 common utterances go into training |
| 46 | +# 40/50 individual utterances go into training |
| 47 | +# 10/50 individual utterances go into testing |
| 48 | + |
| 49 | +function make_speaker_id |
| 50 | +{ |
| 51 | + printf "s%03d" "$1" |
| 52 | +} |
| 53 | + |
| 54 | +function make_sentence_id |
| 55 | +{ |
| 56 | + printf "%02d" "$1" |
| 57 | +} |
| 58 | + |
| 59 | +##################################### |
| 60 | +# Convert wave audio to 16-bit, 16kHz |
| 61 | +##################################### |
| 62 | + |
| 63 | +function convert_to_16khz |
| 64 | +{ |
| 65 | + for i in $(seq 1 $N_SPEAKERS); do |
| 66 | + speaker_id=$(make_speaker_id $i) |
| 67 | + |
| 68 | + mkdir -p "$CORPUS_DIR/$speaker_id/audio_16k/comunes" |
| 69 | + mkdir -p "$CORPUS_DIR/$speaker_id/audio_16k/individuales" |
| 70 | + |
| 71 | + # Common utterances |
| 72 | + for j in $(seq 1 $N_COMMON_UTTERANCES); do |
| 73 | + sentence_id=$(make_sentence_id $j) |
| 74 | + old_wav_file="$CORPUS_DIR/$speaker_id/audio_editado/comunes/$speaker_id$sentence_id.wav" |
| 75 | + new_wav_file="$CORPUS_DIR/$speaker_id/audio_16k/comunes/$speaker_id$sentence_id.wav" |
| 76 | + sox "$old_wav_file" -r 16k "$new_wav_file" |
| 77 | + done |
| 78 | + |
| 79 | + # Individual utterances |
| 80 | + for k in $(seq 1 $N_INDIVIDUAL_UTTERANCES_TRAINING); do |
| 81 | + sentence_id=$(make_sentence_id $k) |
| 82 | + old_wav_file="$CORPUS_DIR/$speaker_id/audio_editado/individuales/$speaker_id$sentence_id.wav" |
| 83 | + new_wav_file="$CORPUS_DIR/$speaker_id/audio_16k/individuales/$speaker_id$sentence_id.wav" |
| 84 | + sox "$old_wav_file" -r 16k "$new_wav_file" |
| 85 | + done |
| 86 | + done |
| 87 | +} |
| 88 | + |
| 89 | +if [[ ! -d "$CORPUS_DIR/s001/audio_16k" ]]; then |
| 90 | + echo |
| 91 | + echo Converting audio from 44.1kHz to 16kHz |
| 92 | + echo |
| 93 | + convert_to_16khz |
| 94 | +fi |
| 95 | + |
| 96 | + |
| 97 | + |
| 98 | +################# |
| 99 | +# data/train/text |
| 100 | +# data/test/text |
| 101 | +################# |
| 102 | + |
| 103 | +# speakerId-utteranceId-[c|i] |
| 104 | +# c = speaker common utterances (10) |
| 105 | +# i = speaker individual utterances (50) |
| 106 | +# |
| 107 | +# e.g.: |
| 108 | +# s001-01-c |
| 109 | +# ... |
| 110 | +# s001-10-c |
| 111 | +# ... |
| 112 | +# s001-01-i |
| 113 | +# ... |
| 114 | +# s001-50-i |
| 115 | + |
| 116 | +## 80-20 train-test split |
| 117 | +## Only individual utterances are used in testing |
| 118 | +# 10/10 common utterances go into training |
| 119 | +# 40/50 individual utterances go into training |
| 120 | +# 10/50 individual utterances go into testing |
| 121 | + |
| 122 | + |
| 123 | + |
| 124 | +function clean |
| 125 | +{ |
| 126 | + echo "$1" \ |
| 127 | + | tr -d '\r' \ |
| 128 | + | tr '[:upper:]' '[:lower:]' \ |
| 129 | + | sed \ |
| 130 | + -e 's/á/a/g' -e 's/é/e/g' -e 's/í/i/g' -e 's/ó/o/g' -e 's/ú/u/g' \ |
| 131 | + -e 's/Á/a/g' -e 's/É/e/g' -e 's/Í/i/g' -e 's/Ó/o/g' -e 's/Ú/u/g' \ |
| 132 | + -e 's/ñ/n/g' -e 's/Ñ/n/g' -e 's/ü/u/g' -e 's/Ü/u/g' \ |
| 133 | + | tr -d -c "a-zA-Z0-9 \r\n" |
| 134 | + # | tr -d -c "_,.;:\-?¿!'\"()" \ |
| 135 | +} |
| 136 | + |
| 137 | +### Generate data/train/text |
| 138 | +for i in $(seq 1 $N_SPEAKERS); do |
| 139 | + speaker_id=$(make_speaker_id $i) |
| 140 | + |
| 141 | + # Common utterances |
| 142 | + for j in $(seq 1 $N_COMMON_UTTERANCES); do |
| 143 | + sentence_id=$(make_sentence_id $j) |
| 144 | + utterance_id="$speaker_id-$sentence_id-c" |
| 145 | + trans_file="$CORPUS_DIR/$speaker_id/texto/comunes/$speaker_id$sentence_id.txt" |
| 146 | + iconv -f iso-8859-1 -t utf-8 "$trans_file" > "$trans_file.utf8" |
| 147 | + if [ -f "$trans_file.utf8" ]; then |
| 148 | + transcription=$(cat "$trans_file.utf8") |
| 149 | + transcription=$(clean "$transcription") |
| 150 | + echo "$utterance_id $transcription" >> "data/train/text" |
| 151 | + fi |
| 152 | + done |
| 153 | + |
| 154 | + # Individual utterances |
| 155 | + for k in $(seq 1 $N_INDIVIDUAL_UTTERANCES_TRAINING); do |
| 156 | + sentence_id=$(make_sentence_id $k) |
| 157 | + utterance_id="$speaker_id-$sentence_id-i" |
| 158 | + trans_file="$CORPUS_DIR/$speaker_id/texto/individuales/$speaker_id$sentence_id.txt" |
| 159 | + iconv -f iso-8859-1 -t utf-8 "$trans_file" > "$trans_file.utf8" |
| 160 | + if [ -f "$trans_file.utf8" ]; then |
| 161 | + transcription=$(cat "$trans_file.utf8") |
| 162 | + transcription=$(clean "$transcription") |
| 163 | + echo "$utterance_id $transcription" >> "data/train/text" |
| 164 | + fi |
| 165 | + done |
| 166 | + |
| 167 | +done |
| 168 | + |
| 169 | + |
| 170 | +### Generate data/test/text |
| 171 | +for i in $(seq 1 $N_SPEAKERS); do |
| 172 | + speaker_id=$(make_speaker_id $i) |
| 173 | + |
| 174 | + # Individual utterances |
| 175 | + for k in $(seq $N_INDIVIDUAL_UTTERANCES_TRAINING $N_INDIVIDUAL_UTTERANCES); do |
| 176 | + sentence_id=$(make_sentence_id $k) |
| 177 | + utterance_id="$speaker_id-$sentence_id-i" |
| 178 | + trans_file="$CORPUS_DIR/$speaker_id/texto/individuales/$speaker_id$sentence_id.txt" |
| 179 | + iconv -f iso-8859-1 -t utf-8 "$trans_file" > "$trans_file.utf8" |
| 180 | + if [ -f "$trans_file.utf8" ]; then |
| 181 | + transcription=$(cat "$trans_file.utf8") |
| 182 | + transcription=$(clean "$transcription") |
| 183 | + echo "$utterance_id $transcription" >> "data/test/text" |
| 184 | + fi |
| 185 | + done |
| 186 | + |
| 187 | +done |
| 188 | + |
| 189 | + |
| 190 | + |
| 191 | + |
| 192 | +#################### |
| 193 | +# data/train/wav.scp |
| 194 | +# data/test/wav.scp |
| 195 | +#################### |
| 196 | + |
| 197 | + |
| 198 | +### Generate data/train/wav.scp |
| 199 | +for i in $(seq 1 $N_SPEAKERS); do |
| 200 | + speaker_id=$(make_speaker_id $i) |
| 201 | + |
| 202 | + # Common utterances |
| 203 | + for j in $(seq 1 $N_COMMON_UTTERANCES); do |
| 204 | + sentence_id=$(make_sentence_id $j) |
| 205 | + utterance_id="$speaker_id-$sentence_id-c" |
| 206 | + wav_file="$CORPUS_DIR/$speaker_id/audio_16k/comunes/$speaker_id$sentence_id.wav" |
| 207 | + if [ -f "$wav_file" ]; then |
| 208 | + echo "$utterance_id $wav_file" >> "data/train/wav.scp" |
| 209 | + fi |
| 210 | + done |
| 211 | + |
| 212 | + # Individual utterances |
| 213 | + for k in $(seq 1 $N_INDIVIDUAL_UTTERANCES_TRAINING); do |
| 214 | + sentence_id=$(make_sentence_id $k) |
| 215 | + utterance_id="$speaker_id-$sentence_id-i" |
| 216 | + wav_file="$CORPUS_DIR/$speaker_id/audio_16k/individuales/$speaker_id$sentence_id.wav" |
| 217 | + if [ -f "$wav_file" ]; then |
| 218 | + echo "$utterance_id $wav_file" >> "data/train/wav.scp" |
| 219 | + fi |
| 220 | + done |
| 221 | + |
| 222 | +done |
| 223 | + |
| 224 | + |
| 225 | +### Generate data/test/wav.scp |
| 226 | +for i in $(seq 1 $N_SPEAKERS); do |
| 227 | + speaker_id=$(make_speaker_id $i) |
| 228 | + |
| 229 | + # Individual utterances |
| 230 | + for k in $(seq $N_INDIVIDUAL_UTTERANCES_TRAINING $N_INDIVIDUAL_UTTERANCES); do |
| 231 | + sentence_id=$(make_sentence_id $k) |
| 232 | + utterance_id="$speaker_id-$sentence_id-i" |
| 233 | + wav_file="$CORPUS_DIR/$speaker_id/audio_16k/individuales/$speaker_id$sentence_id.wav" |
| 234 | + if [ -f "$wav_file" ]; then |
| 235 | + echo "$utterance_id $wav_file" >> "data/test/wav.scp" |
| 236 | + fi |
| 237 | + done |
| 238 | + |
| 239 | +done |
| 240 | + |
| 241 | + |
| 242 | + |
| 243 | + |
| 244 | +#################### |
| 245 | +# data/train/utt2spk |
| 246 | +# data/test/utt2spk |
| 247 | +#################### |
| 248 | + |
| 249 | +# Take IDs from 'text' file to avoid including missing data's IDs |
| 250 | + |
| 251 | +### Generate data/train/utt2spk |
| 252 | +utterance_ids=$(cat "data/train/text" | cut -d' ' -f1) |
| 253 | + |
| 254 | +while read -r utterance_id; do |
| 255 | + speaker_id=$(echo "$utterance_id" | cut -d'-' -f1) |
| 256 | + echo "$utterance_id $speaker_id" >> "data/train/utt2spk" |
| 257 | +done <<< "$utterance_ids" |
| 258 | + |
| 259 | + |
| 260 | +### Generate data/test/utt2spk |
| 261 | +utterance_ids=$(cat "data/test/text" | cut -d' ' -f1) |
| 262 | + |
| 263 | +while read -r utterance_id; do |
| 264 | + speaker_id=$(echo "$utterance_id" | cut -d'-' -f1) |
| 265 | + echo "$utterance_id $speaker_id" >> "data/test/utt2spk" |
| 266 | +done <<< "$utterance_ids" |
| 267 | + |
| 268 | + |
| 269 | +############ |
| 270 | +# Sort files |
| 271 | +############ |
| 272 | + |
| 273 | +LC_ALL=C sort -o "data/train/text" "data/train/text" |
| 274 | +LC_ALL=C sort -o "data/test/text" "data/test/text" |
| 275 | +LC_ALL=C sort -o "data/train/wav.scp" "data/train/wav.scp" |
| 276 | +LC_ALL=C sort -o "data/test/wav.scp" "data/test/wav.scp" |
| 277 | +LC_ALL=C sort -o "data/train/utt2spk" "data/train/utt2spk" |
| 278 | +LC_ALL=C sort -o "data/test/utt2spk" "data/test/utt2spk" |
| 279 | + |
| 280 | + |
| 281 | +#################### |
| 282 | +# data/train/spk2utt |
| 283 | +# data/test/spk2utt |
| 284 | +#################### |
| 285 | +utils/utt2spk_to_spk2utt.pl "data/train/utt2spk" > "data/train/spk2utt" |
| 286 | +utils/utt2spk_to_spk2utt.pl "data/test/utt2spk" > "data/test/spk2utt" |
0 commit comments