deepgeek/silero-vad
2
0
Fork 0
mirror of https://github.com/snakers4/silero-vad.git synced 2026-02-04 17:39:22 +08:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #270 from kafan1986/master

Browse Source 
Resolves Windows inference issue INVALID_ARGUMENT : Unexpected input data type.
This commit is contained in:
Dimitrii Voronin
2022-12-06 11:38:41 +02:00
committed by GitHub
parent 42565d5baa d3882c9ebf
commit 5814e548db
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
utils_vad.py
View File

@@ -61,7 +61,7 @@ class OnnxWrapper():
self.reset_states(batch_size)
if sr in [8000, 16000]:
ort_inputs = {'input': x.numpy(), 'h': self._h, 'c': self._c, 'sr': np.array(sr)}
ort_inputs = {'input': x.numpy(), 'h': self._h, 'c': self._c, 'sr': np.array(sr, dtype='int64')}
ort_outs = self.session.run(None, ort_inputs)
out, self._h, self._c = ort_outs
else: