summaryrefslogtreecommitdiff
path: root/speech_utils
diff options
context:
space:
mode:
authorDeterminant <[email protected]>2015-08-13 15:57:58 +0800
committerDeterminant <[email protected]>2015-08-13 15:57:58 +0800
commitc177a7549bd90670af4b29fa813ddea32cfe0f78 (patch)
tree63c2c477d638d53baaeaaafc1a099d513b4e51c7 /speech_utils
parent184a2710192950fb62269726243a240ff18294eb (diff)
add frm_trim option to global_transf
Diffstat (limited to 'speech_utils')
-rw-r--r--speech_utils/init.lua7
1 files changed, 4 insertions, 3 deletions
diff --git a/speech_utils/init.lua b/speech_utils/init.lua
index 3d1a470..9690e71 100644
--- a/speech_utils/init.lua
+++ b/speech_utils/init.lua
@@ -1,6 +1,7 @@
nerv.speech_utils = {}
-function nerv.speech_utils.global_transf(feat_utter, global_transf, frm_ext, gconf)
+function nerv.speech_utils.global_transf(feat_utter, global_transf,
+ frm_ext, frm_trim, gconf)
local rearranged
feat_utter = gconf.cumat_type.new_from_host(feat_utter)
if frm_ext > 0 then
@@ -21,8 +22,8 @@ function nerv.speech_utils.global_transf(feat_utter, global_transf, frm_ext, gco
global_transf:init(input[1]:nrow())
global_transf:propagate(input, output)
-- trim frames
- expanded = gconf.mmat_type(output[1]:nrow() - frm_ext * 2, output[1]:ncol())
- output[1]:copy_toh(expanded, frm_ext, feat_utter:nrow() - frm_ext)
+ expanded = gconf.mmat_type(output[1]:nrow() - frm_trim * 2, output[1]:ncol())
+ output[1]:copy_toh(expanded, frm_trim, feat_utter:nrow() - frm_trim)
collectgarbage("collect")
return expanded
end