AI/wekws
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Compare commits

base: AI:main
Branches Tags
AI:main
AI:diwu-wekws
..
compare: AI:diwu-wekws
Branches Tags
AI:main
AI:diwu-wekws

1 Commits
main ... diwu-wekws

Author SHA1 Message Date
di.wu
d618951e87 [wekws] fix log (only one process print model) 2023-10-07 15:46:22 +08:00
1 changed files with 0 additions and 9 deletions
Show Stats Expand all files Collapse all files

9
wekws/bin/stream_kws_ctc.py
View File

@ -481,15 +481,6 @@ class KeyWordSpotter(torch.nn.Module):
# update frame offset # update frame offset
self.total_frames += len(probs) * self.downsampling self.total_frames += len(probs) * self.downsampling
# For streaming kws, the cur_hyps should be reset if the time of
# a possible keyword last over the max_frames value you set.
# see this issue:https://github.com/duj12/kws_demo/issues/2
if len(self.cur_hyps) > 0 and len(self.cur_hyps[0][0]) > 0:
keyword_may_start = int(self.cur_hyps[0][1][2][0]['frame'])
if (self.total_frames - keyword_may_start) > self.max_frames:
self.reset()
return self.result return self.result
def reset(self): def reset(self):