e2e_vad.py 29 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624
  1. # -*- encoding: utf-8 -*-
  2. # Copyright FunASR (https://github.com/alibaba-damo-academy/FunASR). All Rights Reserved.
  3. # MIT License (https://opensource.org/licenses/MIT)
  4. from enum import Enum
  5. from typing import List, Tuple, Dict, Any
  6. import math
  7. import numpy as np
  8. class VadStateMachine(Enum):
  9. kVadInStateStartPointNotDetected = 1
  10. kVadInStateInSpeechSegment = 2
  11. kVadInStateEndPointDetected = 3
  12. class FrameState(Enum):
  13. kFrameStateInvalid = -1
  14. kFrameStateSpeech = 1
  15. kFrameStateSil = 0
  16. # final voice/unvoice state per frame
  17. class AudioChangeState(Enum):
  18. kChangeStateSpeech2Speech = 0
  19. kChangeStateSpeech2Sil = 1
  20. kChangeStateSil2Sil = 2
  21. kChangeStateSil2Speech = 3
  22. kChangeStateNoBegin = 4
  23. kChangeStateInvalid = 5
  24. class VadDetectMode(Enum):
  25. kVadSingleUtteranceDetectMode = 0
  26. kVadMutipleUtteranceDetectMode = 1
  27. class VADXOptions:
  28. def __init__(
  29. self,
  30. sample_rate: int = 16000,
  31. detect_mode: int = VadDetectMode.kVadMutipleUtteranceDetectMode.value,
  32. snr_mode: int = 0,
  33. max_end_silence_time: int = 800,
  34. max_start_silence_time: int = 3000,
  35. do_start_point_detection: bool = True,
  36. do_end_point_detection: bool = True,
  37. window_size_ms: int = 200,
  38. sil_to_speech_time_thres: int = 150,
  39. speech_to_sil_time_thres: int = 150,
  40. speech_2_noise_ratio: float = 1.0,
  41. do_extend: int = 1,
  42. lookback_time_start_point: int = 200,
  43. lookahead_time_end_point: int = 100,
  44. max_single_segment_time: int = 60000,
  45. nn_eval_block_size: int = 8,
  46. dcd_block_size: int = 4,
  47. snr_thres: int = -100.0,
  48. noise_frame_num_used_for_snr: int = 100,
  49. decibel_thres: int = -100.0,
  50. speech_noise_thres: float = 0.6,
  51. fe_prior_thres: float = 1e-4,
  52. silence_pdf_num: int = 1,
  53. sil_pdf_ids: List[int] = [0],
  54. speech_noise_thresh_low: float = -0.1,
  55. speech_noise_thresh_high: float = 0.3,
  56. output_frame_probs: bool = False,
  57. frame_in_ms: int = 10,
  58. frame_length_ms: int = 25,
  59. ):
  60. self.sample_rate = sample_rate
  61. self.detect_mode = detect_mode
  62. self.snr_mode = snr_mode
  63. self.max_end_silence_time = max_end_silence_time
  64. self.max_start_silence_time = max_start_silence_time
  65. self.do_start_point_detection = do_start_point_detection
  66. self.do_end_point_detection = do_end_point_detection
  67. self.window_size_ms = window_size_ms
  68. self.sil_to_speech_time_thres = sil_to_speech_time_thres
  69. self.speech_to_sil_time_thres = speech_to_sil_time_thres
  70. self.speech_2_noise_ratio = speech_2_noise_ratio
  71. self.do_extend = do_extend
  72. self.lookback_time_start_point = lookback_time_start_point
  73. self.lookahead_time_end_point = lookahead_time_end_point
  74. self.max_single_segment_time = max_single_segment_time
  75. self.nn_eval_block_size = nn_eval_block_size
  76. self.dcd_block_size = dcd_block_size
  77. self.snr_thres = snr_thres
  78. self.noise_frame_num_used_for_snr = noise_frame_num_used_for_snr
  79. self.decibel_thres = decibel_thres
  80. self.speech_noise_thres = speech_noise_thres
  81. self.fe_prior_thres = fe_prior_thres
  82. self.silence_pdf_num = silence_pdf_num
  83. self.sil_pdf_ids = sil_pdf_ids
  84. self.speech_noise_thresh_low = speech_noise_thresh_low
  85. self.speech_noise_thresh_high = speech_noise_thresh_high
  86. self.output_frame_probs = output_frame_probs
  87. self.frame_in_ms = frame_in_ms
  88. self.frame_length_ms = frame_length_ms
  89. class E2EVadSpeechBufWithDoa(object):
  90. def __init__(self):
  91. self.start_ms = 0
  92. self.end_ms = 0
  93. self.buffer = []
  94. self.contain_seg_start_point = False
  95. self.contain_seg_end_point = False
  96. self.doa = 0
  97. def Reset(self):
  98. self.start_ms = 0
  99. self.end_ms = 0
  100. self.buffer = []
  101. self.contain_seg_start_point = False
  102. self.contain_seg_end_point = False
  103. self.doa = 0
  104. class E2EVadFrameProb(object):
  105. def __init__(self):
  106. self.noise_prob = 0.0
  107. self.speech_prob = 0.0
  108. self.score = 0.0
  109. self.frame_id = 0
  110. self.frm_state = 0
  111. class WindowDetector(object):
  112. def __init__(self, window_size_ms: int, sil_to_speech_time: int,
  113. speech_to_sil_time: int, frame_size_ms: int):
  114. self.window_size_ms = window_size_ms
  115. self.sil_to_speech_time = sil_to_speech_time
  116. self.speech_to_sil_time = speech_to_sil_time
  117. self.frame_size_ms = frame_size_ms
  118. self.win_size_frame = int(window_size_ms / frame_size_ms)
  119. self.win_sum = 0
  120. self.win_state = [0] * self.win_size_frame # 初始化窗
  121. self.cur_win_pos = 0
  122. self.pre_frame_state = FrameState.kFrameStateSil
  123. self.cur_frame_state = FrameState.kFrameStateSil
  124. self.sil_to_speech_frmcnt_thres = int(sil_to_speech_time / frame_size_ms)
  125. self.speech_to_sil_frmcnt_thres = int(speech_to_sil_time / frame_size_ms)
  126. self.voice_last_frame_count = 0
  127. self.noise_last_frame_count = 0
  128. self.hydre_frame_count = 0
  129. def Reset(self) -> None:
  130. self.cur_win_pos = 0
  131. self.win_sum = 0
  132. self.win_state = [0] * self.win_size_frame
  133. self.pre_frame_state = FrameState.kFrameStateSil
  134. self.cur_frame_state = FrameState.kFrameStateSil
  135. self.voice_last_frame_count = 0
  136. self.noise_last_frame_count = 0
  137. self.hydre_frame_count = 0
  138. def GetWinSize(self) -> int:
  139. return int(self.win_size_frame)
  140. def DetectOneFrame(self, frameState: FrameState, frame_count: int) -> AudioChangeState:
  141. cur_frame_state = FrameState.kFrameStateSil
  142. if frameState == FrameState.kFrameStateSpeech:
  143. cur_frame_state = 1
  144. elif frameState == FrameState.kFrameStateSil:
  145. cur_frame_state = 0
  146. else:
  147. return AudioChangeState.kChangeStateInvalid
  148. self.win_sum -= self.win_state[self.cur_win_pos]
  149. self.win_sum += cur_frame_state
  150. self.win_state[self.cur_win_pos] = cur_frame_state
  151. self.cur_win_pos = (self.cur_win_pos + 1) % self.win_size_frame
  152. if self.pre_frame_state == FrameState.kFrameStateSil and self.win_sum >= self.sil_to_speech_frmcnt_thres:
  153. self.pre_frame_state = FrameState.kFrameStateSpeech
  154. return AudioChangeState.kChangeStateSil2Speech
  155. if self.pre_frame_state == FrameState.kFrameStateSpeech and self.win_sum <= self.speech_to_sil_frmcnt_thres:
  156. self.pre_frame_state = FrameState.kFrameStateSil
  157. return AudioChangeState.kChangeStateSpeech2Sil
  158. if self.pre_frame_state == FrameState.kFrameStateSil:
  159. return AudioChangeState.kChangeStateSil2Sil
  160. if self.pre_frame_state == FrameState.kFrameStateSpeech:
  161. return AudioChangeState.kChangeStateSpeech2Speech
  162. return AudioChangeState.kChangeStateInvalid
  163. def FrameSizeMs(self) -> int:
  164. return int(self.frame_size_ms)
  165. class E2EVadModel():
  166. """
  167. Author: Speech Lab of DAMO Academy, Alibaba Group
  168. Deep-FSMN for Large Vocabulary Continuous Speech Recognition
  169. https://arxiv.org/abs/1803.05030
  170. """
  171. def __init__(self, vad_post_args: Dict[str, Any]):
  172. super(E2EVadModel, self).__init__()
  173. self.vad_opts = VADXOptions(**vad_post_args)
  174. self.windows_detector = WindowDetector(self.vad_opts.window_size_ms,
  175. self.vad_opts.sil_to_speech_time_thres,
  176. self.vad_opts.speech_to_sil_time_thres,
  177. self.vad_opts.frame_in_ms)
  178. # self.encoder = encoder
  179. # init variables
  180. self.is_final = False
  181. self.data_buf_start_frame = 0
  182. self.frm_cnt = 0
  183. self.latest_confirmed_speech_frame = 0
  184. self.lastest_confirmed_silence_frame = -1
  185. self.continous_silence_frame_count = 0
  186. self.vad_state_machine = VadStateMachine.kVadInStateStartPointNotDetected
  187. self.confirmed_start_frame = -1
  188. self.confirmed_end_frame = -1
  189. self.number_end_time_detected = 0
  190. self.sil_frame = 0
  191. self.sil_pdf_ids = self.vad_opts.sil_pdf_ids
  192. self.noise_average_decibel = -100.0
  193. self.pre_end_silence_detected = False
  194. self.next_seg = True
  195. self.output_data_buf = []
  196. self.output_data_buf_offset = 0
  197. self.frame_probs = []
  198. self.max_end_sil_frame_cnt_thresh = self.vad_opts.max_end_silence_time - self.vad_opts.speech_to_sil_time_thres
  199. self.speech_noise_thres = self.vad_opts.speech_noise_thres
  200. self.scores = None
  201. self.idx_pre_chunk = 0
  202. self.max_time_out = False
  203. self.decibel = []
  204. self.data_buf_size = 0
  205. self.data_buf_all_size = 0
  206. self.waveform = None
  207. self.ResetDetection()
  208. def AllResetDetection(self):
  209. self.is_final = False
  210. self.data_buf_start_frame = 0
  211. self.frm_cnt = 0
  212. self.latest_confirmed_speech_frame = 0
  213. self.lastest_confirmed_silence_frame = -1
  214. self.continous_silence_frame_count = 0
  215. self.vad_state_machine = VadStateMachine.kVadInStateStartPointNotDetected
  216. self.confirmed_start_frame = -1
  217. self.confirmed_end_frame = -1
  218. self.number_end_time_detected = 0
  219. self.sil_frame = 0
  220. self.sil_pdf_ids = self.vad_opts.sil_pdf_ids
  221. self.noise_average_decibel = -100.0
  222. self.pre_end_silence_detected = False
  223. self.next_seg = True
  224. self.output_data_buf = []
  225. self.output_data_buf_offset = 0
  226. self.frame_probs = []
  227. self.max_end_sil_frame_cnt_thresh = self.vad_opts.max_end_silence_time - self.vad_opts.speech_to_sil_time_thres
  228. self.speech_noise_thres = self.vad_opts.speech_noise_thres
  229. self.scores = None
  230. self.idx_pre_chunk = 0
  231. self.max_time_out = False
  232. self.decibel = []
  233. self.data_buf_size = 0
  234. self.data_buf_all_size = 0
  235. self.waveform = None
  236. self.ResetDetection()
  237. def ResetDetection(self):
  238. self.continous_silence_frame_count = 0
  239. self.latest_confirmed_speech_frame = 0
  240. self.lastest_confirmed_silence_frame = -1
  241. self.confirmed_start_frame = -1
  242. self.confirmed_end_frame = -1
  243. self.vad_state_machine = VadStateMachine.kVadInStateStartPointNotDetected
  244. self.windows_detector.Reset()
  245. self.sil_frame = 0
  246. self.frame_probs = []
  247. def ComputeDecibel(self) -> None:
  248. frame_sample_length = int(self.vad_opts.frame_length_ms * self.vad_opts.sample_rate / 1000)
  249. frame_shift_length = int(self.vad_opts.frame_in_ms * self.vad_opts.sample_rate / 1000)
  250. if self.data_buf_all_size == 0:
  251. self.data_buf_all_size = len(self.waveform[0])
  252. self.data_buf_size = self.data_buf_all_size
  253. else:
  254. self.data_buf_all_size += len(self.waveform[0])
  255. for offset in range(0, self.waveform.shape[1] - frame_sample_length + 1, frame_shift_length):
  256. self.decibel.append(
  257. 10 * math.log10(np.square((self.waveform[0][offset: offset + frame_sample_length])).sum() + \
  258. 0.000001))
  259. def ComputeScores(self, scores: np.ndarray) -> None:
  260. # scores = self.encoder(feats, in_cache) # return B * T * D
  261. self.vad_opts.nn_eval_block_size = scores.shape[1]
  262. self.frm_cnt += scores.shape[1] # count total frames
  263. self.scores=scores
  264. def PopDataBufTillFrame(self, frame_idx: int) -> None: # need check again
  265. while self.data_buf_start_frame < frame_idx:
  266. if self.data_buf_size >= int(self.vad_opts.frame_in_ms * self.vad_opts.sample_rate / 1000):
  267. self.data_buf_start_frame += 1
  268. self.data_buf_size = self.data_buf_all_size-self.data_buf_start_frame * int(
  269. self.vad_opts.frame_in_ms * self.vad_opts.sample_rate / 1000)
  270. def PopDataToOutputBuf(self, start_frm: int, frm_cnt: int, first_frm_is_start_point: bool,
  271. last_frm_is_end_point: bool, end_point_is_sent_end: bool) -> None:
  272. self.PopDataBufTillFrame(start_frm)
  273. expected_sample_number = int(frm_cnt * self.vad_opts.sample_rate * self.vad_opts.frame_in_ms / 1000)
  274. if last_frm_is_end_point:
  275. extra_sample = max(0, int(self.vad_opts.frame_length_ms * self.vad_opts.sample_rate / 1000 - \
  276. self.vad_opts.sample_rate * self.vad_opts.frame_in_ms / 1000))
  277. expected_sample_number += int(extra_sample)
  278. if end_point_is_sent_end:
  279. expected_sample_number = max(expected_sample_number, self.data_buf_size)
  280. if self.data_buf_size < expected_sample_number:
  281. print('error in calling pop data_buf\n')
  282. if len(self.output_data_buf) == 0 or first_frm_is_start_point:
  283. self.output_data_buf.append(E2EVadSpeechBufWithDoa())
  284. self.output_data_buf[-1].Reset()
  285. self.output_data_buf[-1].start_ms = start_frm * self.vad_opts.frame_in_ms
  286. self.output_data_buf[-1].end_ms = self.output_data_buf[-1].start_ms
  287. self.output_data_buf[-1].doa = 0
  288. cur_seg = self.output_data_buf[-1]
  289. if cur_seg.end_ms != start_frm * self.vad_opts.frame_in_ms:
  290. print('warning\n')
  291. out_pos = len(cur_seg.buffer) # cur_seg.buff现在没做任何操作
  292. data_to_pop = 0
  293. if end_point_is_sent_end:
  294. data_to_pop = expected_sample_number
  295. else:
  296. data_to_pop = int(frm_cnt * self.vad_opts.frame_in_ms * self.vad_opts.sample_rate / 1000)
  297. if data_to_pop > self.data_buf_size:
  298. print('VAD data_to_pop is bigger than self.data_buf_size!!!\n')
  299. data_to_pop = self.data_buf_size
  300. expected_sample_number = self.data_buf_size
  301. cur_seg.doa = 0
  302. for sample_cpy_out in range(0, data_to_pop):
  303. # cur_seg.buffer[out_pos ++] = data_buf_.back();
  304. out_pos += 1
  305. for sample_cpy_out in range(data_to_pop, expected_sample_number):
  306. # cur_seg.buffer[out_pos++] = data_buf_.back()
  307. out_pos += 1
  308. if cur_seg.end_ms != start_frm * self.vad_opts.frame_in_ms:
  309. print('Something wrong with the VAD algorithm\n')
  310. self.data_buf_start_frame += frm_cnt
  311. cur_seg.end_ms = (start_frm + frm_cnt) * self.vad_opts.frame_in_ms
  312. if first_frm_is_start_point:
  313. cur_seg.contain_seg_start_point = True
  314. if last_frm_is_end_point:
  315. cur_seg.contain_seg_end_point = True
  316. def OnSilenceDetected(self, valid_frame: int):
  317. self.lastest_confirmed_silence_frame = valid_frame
  318. if self.vad_state_machine == VadStateMachine.kVadInStateStartPointNotDetected:
  319. self.PopDataBufTillFrame(valid_frame)
  320. # silence_detected_callback_
  321. # pass
  322. def OnVoiceDetected(self, valid_frame: int) -> None:
  323. self.latest_confirmed_speech_frame = valid_frame
  324. self.PopDataToOutputBuf(valid_frame, 1, False, False, False)
  325. def OnVoiceStart(self, start_frame: int, fake_result: bool = False) -> None:
  326. if self.vad_opts.do_start_point_detection:
  327. pass
  328. if self.confirmed_start_frame != -1:
  329. print('not reset vad properly\n')
  330. else:
  331. self.confirmed_start_frame = start_frame
  332. if not fake_result and self.vad_state_machine == VadStateMachine.kVadInStateStartPointNotDetected:
  333. self.PopDataToOutputBuf(self.confirmed_start_frame, 1, True, False, False)
  334. def OnVoiceEnd(self, end_frame: int, fake_result: bool, is_last_frame: bool) -> None:
  335. for t in range(self.latest_confirmed_speech_frame + 1, end_frame):
  336. self.OnVoiceDetected(t)
  337. if self.vad_opts.do_end_point_detection:
  338. pass
  339. if self.confirmed_end_frame != -1:
  340. print('not reset vad properly\n')
  341. else:
  342. self.confirmed_end_frame = end_frame
  343. if not fake_result:
  344. self.sil_frame = 0
  345. self.PopDataToOutputBuf(self.confirmed_end_frame, 1, False, True, is_last_frame)
  346. self.number_end_time_detected += 1
  347. def MaybeOnVoiceEndIfLastFrame(self, is_final_frame: bool, cur_frm_idx: int) -> None:
  348. if is_final_frame:
  349. self.OnVoiceEnd(cur_frm_idx, False, True)
  350. self.vad_state_machine = VadStateMachine.kVadInStateEndPointDetected
  351. def GetLatency(self) -> int:
  352. return int(self.LatencyFrmNumAtStartPoint() * self.vad_opts.frame_in_ms)
  353. def LatencyFrmNumAtStartPoint(self) -> int:
  354. vad_latency = self.windows_detector.GetWinSize()
  355. if self.vad_opts.do_extend:
  356. vad_latency += int(self.vad_opts.lookback_time_start_point / self.vad_opts.frame_in_ms)
  357. return vad_latency
  358. def GetFrameState(self, t: int) -> FrameState:
  359. frame_state = FrameState.kFrameStateInvalid
  360. cur_decibel = self.decibel[t]
  361. cur_snr = cur_decibel - self.noise_average_decibel
  362. # for each frame, calc log posterior probability of each state
  363. if cur_decibel < self.vad_opts.decibel_thres:
  364. frame_state = FrameState.kFrameStateSil
  365. self.DetectOneFrame(frame_state, t, False)
  366. return frame_state
  367. sum_score = 0.0
  368. noise_prob = 0.0
  369. assert len(self.sil_pdf_ids) == self.vad_opts.silence_pdf_num
  370. if len(self.sil_pdf_ids) > 0:
  371. assert len(self.scores) == 1 # 只支持batch_size = 1的测试
  372. sil_pdf_scores = [self.scores[0][t - self.idx_pre_chunk][sil_pdf_id] for sil_pdf_id in self.sil_pdf_ids]
  373. sum_score = sum(sil_pdf_scores)
  374. noise_prob = math.log(sum_score) * self.vad_opts.speech_2_noise_ratio
  375. total_score = 1.0
  376. sum_score = total_score - sum_score
  377. speech_prob = math.log(sum_score)
  378. if self.vad_opts.output_frame_probs:
  379. frame_prob = E2EVadFrameProb()
  380. frame_prob.noise_prob = noise_prob
  381. frame_prob.speech_prob = speech_prob
  382. frame_prob.score = sum_score
  383. frame_prob.frame_id = t
  384. self.frame_probs.append(frame_prob)
  385. if math.exp(speech_prob) >= math.exp(noise_prob) + self.speech_noise_thres:
  386. if cur_snr >= self.vad_opts.snr_thres and cur_decibel >= self.vad_opts.decibel_thres:
  387. frame_state = FrameState.kFrameStateSpeech
  388. else:
  389. frame_state = FrameState.kFrameStateSil
  390. else:
  391. frame_state = FrameState.kFrameStateSil
  392. if self.noise_average_decibel < -99.9:
  393. self.noise_average_decibel = cur_decibel
  394. else:
  395. self.noise_average_decibel = (cur_decibel + self.noise_average_decibel * (
  396. self.vad_opts.noise_frame_num_used_for_snr
  397. - 1)) / self.vad_opts.noise_frame_num_used_for_snr
  398. return frame_state
  399. def __call__(self, score: np.ndarray, waveform: np.ndarray,
  400. is_final: bool = False, max_end_sil: int = 800, online: bool = False
  401. ):
  402. self.max_end_sil_frame_cnt_thresh = max_end_sil - self.vad_opts.speech_to_sil_time_thres
  403. self.waveform = waveform # compute decibel for each frame
  404. self.ComputeDecibel()
  405. self.ComputeScores(score)
  406. if not is_final:
  407. self.DetectCommonFrames()
  408. else:
  409. self.DetectLastFrames()
  410. segments = []
  411. for batch_num in range(0, score.shape[0]): # only support batch_size = 1 now
  412. segment_batch = []
  413. if len(self.output_data_buf) > 0:
  414. for i in range(self.output_data_buf_offset, len(self.output_data_buf)):
  415. if online:
  416. if not self.output_data_buf[i].contain_seg_start_point:
  417. continue
  418. if not self.next_seg and not self.output_data_buf[i].contain_seg_end_point:
  419. continue
  420. start_ms = self.output_data_buf[i].start_ms if self.next_seg else -1
  421. if self.output_data_buf[i].contain_seg_end_point:
  422. end_ms = self.output_data_buf[i].end_ms
  423. self.next_seg = True
  424. self.output_data_buf_offset += 1
  425. else:
  426. end_ms = -1
  427. self.next_seg = False
  428. else:
  429. if not is_final and (not self.output_data_buf[i].contain_seg_start_point or not self.output_data_buf[
  430. i].contain_seg_end_point):
  431. continue
  432. start_ms = self.output_data_buf[i].start_ms
  433. end_ms = self.output_data_buf[i].end_ms
  434. self.output_data_buf_offset += 1
  435. segment = [start_ms, end_ms]
  436. segment_batch.append(segment)
  437. if segment_batch:
  438. segments.append(segment_batch)
  439. if is_final:
  440. # reset class variables and clear the dict for the next query
  441. self.AllResetDetection()
  442. return segments
  443. def DetectCommonFrames(self) -> int:
  444. if self.vad_state_machine == VadStateMachine.kVadInStateEndPointDetected:
  445. return 0
  446. for i in range(self.vad_opts.nn_eval_block_size - 1, -1, -1):
  447. frame_state = FrameState.kFrameStateInvalid
  448. frame_state = self.GetFrameState(self.frm_cnt - 1 - i)
  449. self.DetectOneFrame(frame_state, self.frm_cnt - 1 - i, False)
  450. self.idx_pre_chunk += self.scores.shape[1]
  451. return 0
  452. def DetectLastFrames(self) -> int:
  453. if self.vad_state_machine == VadStateMachine.kVadInStateEndPointDetected:
  454. return 0
  455. for i in range(self.vad_opts.nn_eval_block_size - 1, -1, -1):
  456. frame_state = FrameState.kFrameStateInvalid
  457. frame_state = self.GetFrameState(self.frm_cnt - 1 - i)
  458. if i != 0:
  459. self.DetectOneFrame(frame_state, self.frm_cnt - 1 - i, False)
  460. else:
  461. self.DetectOneFrame(frame_state, self.frm_cnt - 1, True)
  462. return 0
  463. def DetectOneFrame(self, cur_frm_state: FrameState, cur_frm_idx: int, is_final_frame: bool) -> None:
  464. tmp_cur_frm_state = FrameState.kFrameStateInvalid
  465. if cur_frm_state == FrameState.kFrameStateSpeech:
  466. if math.fabs(1.0) > self.vad_opts.fe_prior_thres:
  467. tmp_cur_frm_state = FrameState.kFrameStateSpeech
  468. else:
  469. tmp_cur_frm_state = FrameState.kFrameStateSil
  470. elif cur_frm_state == FrameState.kFrameStateSil:
  471. tmp_cur_frm_state = FrameState.kFrameStateSil
  472. state_change = self.windows_detector.DetectOneFrame(tmp_cur_frm_state, cur_frm_idx)
  473. frm_shift_in_ms = self.vad_opts.frame_in_ms
  474. if AudioChangeState.kChangeStateSil2Speech == state_change:
  475. silence_frame_count = self.continous_silence_frame_count
  476. self.continous_silence_frame_count = 0
  477. self.pre_end_silence_detected = False
  478. start_frame = 0
  479. if self.vad_state_machine == VadStateMachine.kVadInStateStartPointNotDetected:
  480. start_frame = max(self.data_buf_start_frame, cur_frm_idx - self.LatencyFrmNumAtStartPoint())
  481. self.OnVoiceStart(start_frame)
  482. self.vad_state_machine = VadStateMachine.kVadInStateInSpeechSegment
  483. for t in range(start_frame + 1, cur_frm_idx + 1):
  484. self.OnVoiceDetected(t)
  485. elif self.vad_state_machine == VadStateMachine.kVadInStateInSpeechSegment:
  486. for t in range(self.latest_confirmed_speech_frame + 1, cur_frm_idx):
  487. self.OnVoiceDetected(t)
  488. if cur_frm_idx - self.confirmed_start_frame + 1 > \
  489. self.vad_opts.max_single_segment_time / frm_shift_in_ms:
  490. self.OnVoiceEnd(cur_frm_idx, False, False)
  491. self.vad_state_machine = VadStateMachine.kVadInStateEndPointDetected
  492. elif not is_final_frame:
  493. self.OnVoiceDetected(cur_frm_idx)
  494. else:
  495. self.MaybeOnVoiceEndIfLastFrame(is_final_frame, cur_frm_idx)
  496. else:
  497. pass
  498. elif AudioChangeState.kChangeStateSpeech2Sil == state_change:
  499. self.continous_silence_frame_count = 0
  500. if self.vad_state_machine == VadStateMachine.kVadInStateStartPointNotDetected:
  501. pass
  502. elif self.vad_state_machine == VadStateMachine.kVadInStateInSpeechSegment:
  503. if cur_frm_idx - self.confirmed_start_frame + 1 > \
  504. self.vad_opts.max_single_segment_time / frm_shift_in_ms:
  505. self.OnVoiceEnd(cur_frm_idx, False, False)
  506. self.vad_state_machine = VadStateMachine.kVadInStateEndPointDetected
  507. elif not is_final_frame:
  508. self.OnVoiceDetected(cur_frm_idx)
  509. else:
  510. self.MaybeOnVoiceEndIfLastFrame(is_final_frame, cur_frm_idx)
  511. else:
  512. pass
  513. elif AudioChangeState.kChangeStateSpeech2Speech == state_change:
  514. self.continous_silence_frame_count = 0
  515. if self.vad_state_machine == VadStateMachine.kVadInStateInSpeechSegment:
  516. if cur_frm_idx - self.confirmed_start_frame + 1 > \
  517. self.vad_opts.max_single_segment_time / frm_shift_in_ms:
  518. self.max_time_out = True
  519. self.OnVoiceEnd(cur_frm_idx, False, False)
  520. self.vad_state_machine = VadStateMachine.kVadInStateEndPointDetected
  521. elif not is_final_frame:
  522. self.OnVoiceDetected(cur_frm_idx)
  523. else:
  524. self.MaybeOnVoiceEndIfLastFrame(is_final_frame, cur_frm_idx)
  525. else:
  526. pass
  527. elif AudioChangeState.kChangeStateSil2Sil == state_change:
  528. self.continous_silence_frame_count += 1
  529. if self.vad_state_machine == VadStateMachine.kVadInStateStartPointNotDetected:
  530. # silence timeout, return zero length decision
  531. if ((self.vad_opts.detect_mode == VadDetectMode.kVadSingleUtteranceDetectMode.value) and (
  532. self.continous_silence_frame_count * frm_shift_in_ms > self.vad_opts.max_start_silence_time)) \
  533. or (is_final_frame and self.number_end_time_detected == 0):
  534. for t in range(self.lastest_confirmed_silence_frame + 1, cur_frm_idx):
  535. self.OnSilenceDetected(t)
  536. self.OnVoiceStart(0, True)
  537. self.OnVoiceEnd(0, True, False);
  538. self.vad_state_machine = VadStateMachine.kVadInStateEndPointDetected
  539. else:
  540. if cur_frm_idx >= self.LatencyFrmNumAtStartPoint():
  541. self.OnSilenceDetected(cur_frm_idx - self.LatencyFrmNumAtStartPoint())
  542. elif self.vad_state_machine == VadStateMachine.kVadInStateInSpeechSegment:
  543. if self.continous_silence_frame_count * frm_shift_in_ms >= self.max_end_sil_frame_cnt_thresh:
  544. lookback_frame = int(self.max_end_sil_frame_cnt_thresh / frm_shift_in_ms)
  545. if self.vad_opts.do_extend:
  546. lookback_frame -= int(self.vad_opts.lookahead_time_end_point / frm_shift_in_ms)
  547. lookback_frame -= 1
  548. lookback_frame = max(0, lookback_frame)
  549. self.OnVoiceEnd(cur_frm_idx - lookback_frame, False, False)
  550. self.vad_state_machine = VadStateMachine.kVadInStateEndPointDetected
  551. elif cur_frm_idx - self.confirmed_start_frame + 1 > \
  552. self.vad_opts.max_single_segment_time / frm_shift_in_ms:
  553. self.OnVoiceEnd(cur_frm_idx, False, False)
  554. self.vad_state_machine = VadStateMachine.kVadInStateEndPointDetected
  555. elif self.vad_opts.do_extend and not is_final_frame:
  556. if self.continous_silence_frame_count <= int(
  557. self.vad_opts.lookahead_time_end_point / frm_shift_in_ms):
  558. self.OnVoiceDetected(cur_frm_idx)
  559. else:
  560. self.MaybeOnVoiceEndIfLastFrame(is_final_frame, cur_frm_idx)
  561. else:
  562. pass
  563. if self.vad_state_machine == VadStateMachine.kVadInStateEndPointDetected and \
  564. self.vad_opts.detect_mode == VadDetectMode.kVadMutipleUtteranceDetectMode.value:
  565. self.ResetDetection()