RVC_HF / infer /lib /audio.py
r3gm's picture
Upload 288 files
7bc29af
raw
history blame
6.58 kB
import librosa
import numpy as np
import av
from io import BytesIO
import ffmpeg
import os
import sys
import random
from infer.lib.csvutil import CSVutil
#import csv
platform_stft_mapping = {
'linux': 'stftpitchshift',
'darwin': 'stftpitchshift',
'win32': 'stftpitchshift.exe',
}
stft = platform_stft_mapping.get(sys.platform)
def wav2(i, o, format):
inp = av.open(i, 'rb')
if format == "m4a": format = "mp4"
out = av.open(o, 'wb', format=format)
if format == "ogg": format = "libvorbis"
if format == "mp4": format = "aac"
ostream = out.add_stream(format)
for frame in inp.decode(audio=0):
for p in ostream.encode(frame): out.mux(p)
for p in ostream.encode(None): out.mux(p)
out.close()
inp.close()
def audio2(i, o, format, sr):
inp = av.open(i, 'rb')
out = av.open(o, 'wb', format=format)
if format == "ogg": format = "libvorbis"
if format == "f32le": format = "pcm_f32le"
ostream = out.add_stream(format, channels=1)
ostream.sample_rate = sr
for frame in inp.decode(audio=0):
for p in ostream.encode(frame): out.mux(p)
out.close()
inp.close()
def load_audion(file, sr):
try:
file = (
file.strip(" ").strip('"').strip("\n").strip('"').strip(" ")
) # 防止小白拷路径头尾带了空格和"和回车
with open(file, "rb") as f:
with BytesIO() as out:
audio2(f, out, "f32le", sr)
return np.frombuffer(out.getvalue(), np.float32).flatten()
except AttributeError:
audio = file[1] / 32768.0
if len(audio.shape) == 2:
audio = np.mean(audio, -1)
return librosa.resample(audio, orig_sr=file[0], target_sr=16000)
except Exception as e:
raise RuntimeError(f"Failed to load audio: {e}")
def load_audio(file, sr, DoFormant=False, Quefrency=1.0, Timbre=1.0):
converted = False
DoFormant, Quefrency, Timbre = CSVutil("csvdb/formanting.csv", "r", "formanting")
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.
file = (
file.strip(" ").strip('"').strip("\n").strip('"').strip(" ")
) # 防止小白拷路径头尾带了空格和"和回车
file_formanted = file.strip(" ").strip('"').strip("\n").strip('"').strip(" ")
# print(f"dofor={bool(DoFormant)} timbr={Timbre} quef={Quefrency}\n")
if (
lambda DoFormant: True
if DoFormant.lower() == "true"
else (False if DoFormant.lower() == "false" else DoFormant)
)(DoFormant):
numerator = round(random.uniform(1, 4), 4)
# os.system(f"stftpitchshift -i {file} -q {Quefrency} -t {Timbre} -o {file_formanted}")
# print('stftpitchshift -i "%s" -p 1.0 --rms -w 128 -v 8 -q %s -t %s -o "%s"' % (file, Quefrency, Timbre, file_formanted))
if not file.endswith(".wav"):
if not os.path.isfile(f"{file_formanted}.wav"):
converted = True
# print(f"\nfile = {file}\n")
# print(f"\nfile_formanted = {file_formanted}\n")
converting = (
ffmpeg.input(file_formanted, threads=0)
.output(f"{file_formanted}.wav")
.run(
cmd=["ffmpeg", "-nostdin"],
capture_stdout=True,
capture_stderr=True,
)
)
else:
pass
file_formanted = (
f"{file_formanted}.wav"
if not file_formanted.endswith(".wav")
else file_formanted
)
print(f" · Formanting {file_formanted}...\n")
os.system(
'%s -i "%s" -q "%s" -t "%s" -o "%sFORMANTED_%s.wav"'
% (
stft,
file_formanted,
Quefrency,
Timbre,
file_formanted,
str(numerator),
)
)
print(f" · Formanted {file_formanted}!\n")
# filepraat = (os.path.abspath(os.getcwd()) + '\\' + file).replace('/','\\')
# file_formantedpraat = ('"' + os.path.abspath(os.getcwd()) + '/' + 'formanted'.join(file_formanted) + '"').replace('/','\\')
# print("%sFORMANTED_%s.wav" % (file_formanted, str(numerator)))
out, _ = (
ffmpeg.input(
"%sFORMANTED_%s.wav" % (file_formanted, str(numerator)), threads=0
)
.output("-", format="f32le", acodec="pcm_f32le", ac=1, ar=sr)
.run(
cmd=["ffmpeg", "-nostdin"], capture_stdout=True, capture_stderr=True
)
)
try:
os.remove("%sFORMANTED_%s.wav" % (file_formanted, str(numerator)))
except Exception:
pass
print("couldn't remove formanted type of file")
else:
out, _ = (
ffmpeg.input(file, threads=0)
.output("-", format="f32le", acodec="pcm_f32le", ac=1, ar=sr)
.run(
cmd=["ffmpeg", "-nostdin"], capture_stdout=True, capture_stderr=True
)
)
except Exception as e:
raise RuntimeError(f"Failed to load audio: {e}")
if converted:
try:
os.remove(file_formanted)
except Exception:
pass
print("couldn't remove converted type of file")
converted = False
return np.frombuffer(out, np.float32).flatten()
def check_audio_duration(file):
try:
file = file.strip(" ").strip('"').strip("\n").strip('"').strip(" ")
probe = ffmpeg.probe(file)
duration = float(probe['streams'][0]['duration'])
if duration < 0.76:
print(
f"\n------------\n"
f"Audio file, {file.split('/')[-1]}, under ~0.76s detected - file is too short. Target at least 1-2s for best results."
f"\n------------\n\n"
)
return False
return True
except Exception as e:
raise RuntimeError(f"Failed to check audio duration: {e}")