Blame view

egs/wsj/s5/utils/nnet/gen_splice.py 1.74 KB
8dcb6dfcb   Yannick Estève   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
  #!/usr/bin/env python
  
  # Copyright 2012  Brno University of Technology (author: Karel Vesely)
  
  # Licensed under the Apache License, Version 2.0 (the "License");
  # you may not use this file except in compliance with the License.
  # You may obtain a copy of the License at
  #
  #  http://www.apache.org/licenses/LICENSE-2.0
  #
  # THIS CODE IS PROVIDED *AS IS* BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
  # KIND, EITHER EXPRESS OR IMPLIED, INCLUDING WITHOUT LIMITATION ANY IMPLIED
  # WARRANTIES OR CONDITIONS OF TITLE, FITNESS FOR A PARTICULAR PURPOSE,
  # MERCHANTABLITY OR NON-INFRINGEMENT.
  # See the Apache 2 License for the specific language governing permissions and
  # limitations under the License.
  
  # ./gen_splice.py
  # generates <splice> Component
  
  from __future__ import print_function
  from math import *
  import sys
  
  
  from optparse import OptionParser
  
  def print_on_same_line(text):
      print(text, end=' ')
  
  parser = OptionParser()
  parser.add_option('--fea-dim', dest='dim_in', help='feature dimension')
  parser.add_option('--splice', dest='splice', help='number of frames to concatenate with the central frame')
  parser.add_option('--splice-step', dest='splice_step', help='splicing step (frames dont need to be consecutive, --splice 3 --splice-step 2 will select offsets: -6 -4 -2 0 2 4 6)', default='1' )
  (options, args) = parser.parse_args()
  
  if(options.dim_in == None):
      parser.print_help()
      sys.exit(1)
  
  dim_in=int(options.dim_in)
  splice=int(options.splice)
  splice_step=int(options.splice_step)
  
  dim_out=(2*splice+1)*dim_in
  
  print('<splice> {0} {1}'.format(dim_out, dim_in))
  print_on_same_line('[')
  
  splice_vec = list(range(-splice*splice_step, splice*splice_step+1, splice_step))
  for idx in range(len(splice_vec)):
      print_on_same_line(splice_vec[idx])
  
  print(']')