Blame view
egs/wsj/s5/utils/data/get_frame_shift.sh
1.94 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 73 74 |
#!/bin/bash
# Copyright 2016 Johns Hopkins University (author: Daniel Povey)
# Apache 2.0
# This script takes as input a data directory, such as data/train/, preferably
# with utt2dur file already existing (or the utt2dur file will be created if
# not), and it attempts to work out the approximate frame shift by comparing the
# utt2dur with the output of feat-to-len on the feats.scp. It prints it out.
# if the shift is very close to, but above, 0.01 (the normal frame shift) it
# rounds it down.
. utils/parse_options.sh
. ./path.sh
if [ $# != 1 ]; then
cat >&2 <<EOF
Usage: frame_shift=\$($0 <datadir>)
e.g.: frame_shift=\$($0 data/train)
This script prints the frame-shift in seconds (e.g. 0.01) to the standard out.
Its output is intended to be captured in a shell variable.
If <datadir> does not contain the file utt2dur, this script may invoke
utils/data/get_utt2dur.sh, which will require write permission to <datadir>.
EOF
exit 1
fi
export LC_ALL=C
dir=$1
if [[ -s $dir/frame_shift ]]; then
cat $dir/frame_shift
exit
fi
if [ ! -f $dir/feats.scp ]; then
echo "$0: $dir/feats.scp does not exist" 1>&2
exit 1
fi
if [ ! -s $dir/utt2dur ]; then
if [ ! -e $dir/wav.scp ] && [ ! -s $dir/segments ]; then
echo "$0: neither $dir/wav.scp nor $dir/segments exist; assuming a frame shift of 0.01." 1>&2
echo 0.01
exit 0
fi
echo "$0: $dir/utt2dur does not exist: creating it" 1>&2
utils/data/get_utt2dur.sh 1>&2 $dir || exit 1
fi
temp=$(mktemp /tmp/tmp.XXXX) || exit 1
feat-to-len --print-args=false "scp:head -n 10 $dir/feats.scp|" ark,t:- > $temp
if [[ ! -s $temp ]]; then
rm $temp
echo "$0: error running feat-to-len" 1>&2
exit 1
fi
frame_shift=$(head -n 10 $dir/utt2dur | paste - $temp | awk '
{ dur += $2; frames += $4; }
END { shift = dur / frames;
if (shift > 0.01 && shift < 0.0102) shift = 0.01;
print shift; }') || exit 1;
rm $temp
echo $frame_shift > $dir/frame_shift
echo $frame_shift
exit 0
|