2023-04-01 15:02:53 +08:00
|
|
|
import ffmpeg
|
|
|
|
|
import numpy as np
|
2023-03-31 17:54:38 +08:00
|
|
|
def load_audio(file,sr):
|
|
|
|
|
try:
|
|
|
|
|
# https://github.com/openai/whisper/blob/main/whisper/audio.py#L26
|
|
|
|
|
# This launches a subprocess to decode audio while down-mixing and resampling as necessary.
|
|
|
|
|
# Requires the ffmpeg CLI and `ffmpeg-python` package to be installed.
|
|
|
|
|
out, _ = (
|
|
|
|
|
ffmpeg.input(file, threads=0)
|
|
|
|
|
.output("-", format="s16le", acodec="pcm_s16le", ac=1, ar=sr)
|
2023-04-01 15:10:46 +08:00
|
|
|
.run(cmd=["ffmpeg", "-nostdin"], capture_stdout=True, capture_stderr=True)
|
2023-03-31 17:54:38 +08:00
|
|
|
)
|
|
|
|
|
except ffmpeg.Error as e:
|
|
|
|
|
raise RuntimeError(f"Failed to load audio: {e.stderr.decode()}") from e
|
|
|
|
|
|
|
|
|
|
return np.frombuffer(out, np.int16).flatten().astype(np.float32) / 32768.0
|
|
|
|
|
|
|
|
|
|
if __name__=='__main__' :
|
|
|
|
|
print(load_audio(r"C:\CloudMusic\宮野幸子,森下唯 - 月夜に謳う君 -LUNA-.mp3",16000).shape)
|
