Blame view
egs/wsj/s5/utils/nnet/gen_splice.py
1.74 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 |
#!/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(']') |