Blame view

Scripts/utils/copy_data_dir.sh 2.7 KB
ec85f8892   bigot benjamin   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
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
  #!/bin/bash
  
  # Copyright 2013  Johns Hopkins University (author: Daniel Povey)
  # Apache 2.0
  
  # This script operates on a directory, such as in data/train/,
  # that contains some subset of the following files:
  #  feats.scp
  #  wav.scp
  #  spk2utt
  #  utt2spk
  #  text
  #
  # It copies to another directory, possibly adding a specified prefix or a suffix
  # to the utterance and/or speaker names.  Note, the recording-ids stay the same.
  #
  
  
  # begin configuration section
  spk_prefix=
  utt_prefix=
  spk_suffix=
  utt_suffix=
  # end configuration section
  
  . utils/parse_options.sh
  
  if [ $# != 2 ]; then
    echo "Usage: "
    echo "  $0 [options] <srcdir> <destdir>"
    echo "e.g.:"
    echo " $0 --spk-prefix=1- --utt-prefix=1- data/train data/train_1"
    echo "Options"
    echo "   --spk-prefix=<prefix>     # Prefix for speaker ids, default empty"
    echo "   --utt-prefix=<prefix>     # Prefix for utterance ids, default empty"
    echo "   --spk-suffix=<suffix>     # Suffix for speaker ids, default empty"
    echo "   --utt-suffix=<suffix>     # Suffix for utterance ids, default empty"
    exit 1;
  fi
  
  
  export LC_ALL=C
  
  srcdir=$1
  destdir=$2
  
  if [ ! -f $srcdir/utt2spk ]; then
    echo "copy_data_dir.sh: no such file $srcdir/utt2spk" 
    exit 1;
  fi
  
  set -e;
  
  mkdir -p $destdir
  
  cat $srcdir/utt2spk | awk -v p=$utt_prefix -v s=$utt_suffix '{printf("%s %s%s%s
  ", $1, p, $1, s);}' > $destdir/utt_map
  cat $srcdir/spk2utt | awk -v p=$spk_prefix -v s=$spk_suffix '{printf("%s %s%s%s
  ", $1, p, $1, s);}' > $destdir/spk_map
  
  cat $srcdir/utt2spk | utils/apply_map.pl -f 1 $destdir/utt_map  | \
    utils/apply_map.pl -f 2 $destdir/spk_map >$destdir/utt2spk
  
  utils/utt2spk_to_spk2utt.pl <$destdir/utt2spk >$destdir/spk2utt
  
  if [ -f $srcdir/feats.scp ]; then
    utils/apply_map.pl -f 1 $destdir/utt_map <$srcdir/feats.scp >$destdir/feats.scp
  fi
  
  
  if [ -f $srcdir/segments ]; then
    utils/apply_map.pl -f 1 $destdir/utt_map <$srcdir/segments >$srcdir/segments
    cp $srcdir/wav.scp $destdir
    if [ -f $srcdir/reco2file_and_channel ]; then
      cp $srcdir/reco2file_and_channel $destdir/
    fi
  else # no segments->wav indexed by utt.
    if [ -f $srcdir/wav.scp ]; then 
      utils/apply_map.pl -f 1 $destdir/utt_map <$srcdir/wav.scp >$destdir/wav.scp
    fi
  fi
  
  if [ -f $srcdir/text ]; then
    utils/apply_map.pl -f 1 $destdir/utt_map <$srcdir/text >$destdir/text
  fi
  if [ -f $srcdir/spk2gender ]; then
    utils/apply_map.pl -f 1 $destdir/spk_map <$srcdir/spk2gender >$destdir/spk2gender
  fi
  if [ -f $srcdir/cmvn.scp ]; then
    utils/apply_map.pl -f 1 $destdir/spk_map <$srcdir/cmvn.scp >$destdir/cmvn.scp
  fi
  for f in stm glm ctm; do
    if [ -f $srcdir/$f ]; then
      cp $srcdir/$f $destdir
    fi
  done
  
  rm $destdir/spk_map $destdir/utt_map
  
  echo "$0: copied data from $srcdir to $destdir"
  utils/validate_data_dir.sh $destdir