소스 검색

startup function `wavfile_read_mono`: support mono

Fabian Peter Hammerle 4 년 전
부모
커밋
e3063f2a55
1개의 변경된 파일2개의 추가작업 그리고 0개의 파일을 삭제
  1. 2 0
      profile_default/startup/init.py

+ 2 - 0
profile_default/startup/init.py

@@ -51,6 +51,8 @@ def wavfile_read_mono(
 ) -> typing.Tuple[int, numpy.ndarray]:
     # https://docs.scipy.org/doc/scipy/reference/generated/scipy.io.wavfile.read.html
     rate, data = scipy.io.wavfile.read(path)
+    if len(data.shape) == 1:
+        return rate, data
     data_first_channel = data[:, 0]
     for channel_index in range(1, data.shape[1]):
         assert (data_first_channel == data[:, channel_index]).all()