increasing timestamps fixed
but the code needs to be simplified and cleaned before merging
This commit is contained in:
parent
6b968c6e29
commit
c8123344c6
2 changed files with 9 additions and 5 deletions
|
|
@ -235,16 +235,19 @@ class OnlineASRProcessor:
|
||||||
|
|
||||||
self.buffer_trimming_way, self.buffer_trimming_sec = buffer_trimming
|
self.buffer_trimming_way, self.buffer_trimming_sec = buffer_trimming
|
||||||
|
|
||||||
def init(self):
|
def init(self, keep_offset=False):
|
||||||
"""run this when starting or restarting processing"""
|
"""run this when starting or restarting processing"""
|
||||||
self.audio_buffer = np.array([],dtype=np.float32)
|
self.audio_buffer = np.array([],dtype=np.float32)
|
||||||
self.buffer_time_offset = 0
|
|
||||||
|
|
||||||
self.transcript_buffer = HypothesisBuffer(logfile=self.logfile)
|
self.transcript_buffer = HypothesisBuffer(logfile=self.logfile)
|
||||||
|
if not keep_offset:
|
||||||
|
self.buffer_time_offset = 0
|
||||||
|
self.transcript_buffer.last_commited_time = 0
|
||||||
|
else:
|
||||||
|
self.transcript_buffer.last_commited_time = self.buffer_time_offset
|
||||||
|
|
||||||
self.commited = []
|
self.commited = []
|
||||||
self.last_chunked_at = 0
|
self.last_chunked_at = 0
|
||||||
|
|
||||||
self.silence_iters = 0
|
|
||||||
|
|
||||||
def insert_audio_chunk(self, audio):
|
def insert_audio_chunk(self, audio):
|
||||||
self.audio_buffer = np.append(self.audio_buffer, audio)
|
self.audio_buffer = np.append(self.audio_buffer, audio)
|
||||||
|
|
@ -400,6 +403,7 @@ class OnlineASRProcessor:
|
||||||
o = self.transcript_buffer.complete()
|
o = self.transcript_buffer.complete()
|
||||||
f = self.to_flush(o)
|
f = self.to_flush(o)
|
||||||
print("last, noncommited:",f,file=self.logfile)
|
print("last, noncommited:",f,file=self.logfile)
|
||||||
|
self.buffer_time_offset += len(self.audio_buffer)/16000
|
||||||
return f
|
return f
|
||||||
|
|
||||||
|
|
||||||
|
|
|
||||||
|
|
@ -45,7 +45,7 @@ class VACOnlineASRProcessor(OnlineASRProcessor):
|
||||||
|
|
||||||
def finish(self):
|
def finish(self):
|
||||||
ret = self.online.finish()
|
ret = self.online.finish()
|
||||||
self.online.init()
|
self.online.init(keep_offset=True)
|
||||||
self.current_online_chunk_buffer_size = 0
|
self.current_online_chunk_buffer_size = 0
|
||||||
return ret
|
return ret
|
||||||
|
|
||||||
|
|
|
||||||
Loading…
Reference in a new issue