vad fix
This commit is contained in:
+3
-5
@@ -18,8 +18,7 @@ class SileroVADOnnx:
|
|||||||
self._reset_state()
|
self._reset_state()
|
||||||
|
|
||||||
def _reset_state(self):
|
def _reset_state(self):
|
||||||
self._h = np.zeros((2, 1, 64), dtype=np.float32)
|
self._state = np.zeros((2, 1, 128), dtype=np.float32)
|
||||||
self._c = np.zeros((2, 1, 64), dtype=np.float32)
|
|
||||||
|
|
||||||
def reset_states(self):
|
def reset_states(self):
|
||||||
self._reset_state()
|
self._reset_state()
|
||||||
@@ -32,10 +31,9 @@ class SileroVADOnnx:
|
|||||||
ort_inputs = {
|
ort_inputs = {
|
||||||
"input": input_data,
|
"input": input_data,
|
||||||
"sr": sr,
|
"sr": sr,
|
||||||
"h": self._h,
|
"state": self._state,
|
||||||
"c": self._c,
|
|
||||||
}
|
}
|
||||||
out, self._h, self._c = self.session.run(None, ort_inputs)
|
out, self._state = self.session.run(None, ort_inputs)
|
||||||
return float(out.squeeze())
|
return float(out.squeeze())
|
||||||
|
|
||||||
|
|
||||||
|
|||||||
Reference in New Issue
Block a user