Fix for mono input

Signed-off-by: Jael Gu <mengjia.gu@zilliz.com>
3 years ago · 4159c67497
1 changed files with 3 additions and 1 deletions
--- a/nn_fingerprint.py
+++ b/nn_fingerprint.py
@ -115,7 +115,9 @@ class NNFingerprint(NNOperator):
            audio = numpy.vstack(frames).transpose()
        else:
            audio = numpy.hstack(frames)
-            audio = audio[None, :]
+            if len(audio.shape) == 1:
+                audio = audio[None, :]
+            assert len(audio.shape) == 2
        audio = self.int2float(audio)

        if sr != self.params['sample_rate']: