mirror of
https://github.com/Mangio621/Mangio-RVC-Fork.git
synced 2025-12-16 19:47:42 +01:00
26ddba243d
switched from storing data in txt to sqlite3. no more formant.txt and stop.txt! all of the data from sql database is stored in memory of PC.
74 lines
3.1 KiB
Python
74 lines
3.1 KiB
Python
import ffmpeg
|
|
import numpy as np
|
|
|
|
# import praatio
|
|
# import praatio.praat_scripts
|
|
import os
|
|
|
|
import sqlite3
|
|
|
|
|
|
|
|
# praatEXE = join('.',os.path.abspath(os.getcwd()) + r"\Praat.exe")
|
|
|
|
|
|
def load_audio(file, sr, DoFormant, Quefrency, Timbre):
|
|
try:
|
|
conn = sqlite3.connect('TEMP/db:cachedb?mode=memory&cache=shared', check_same_thread=False)
|
|
cursor = conn.cursor()
|
|
# 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(" ")
|
|
cursor.execute("SELECT Quefrency, Timbre, DoFormant FROM formant_data")
|
|
Quefrency, Timbre, DoFormant = cursor.fetchone()
|
|
print(f"dofor={bool(DoFormant)} timbr={Timbre} quef={Quefrency}\n")
|
|
if bool(DoFormant):
|
|
# 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"):
|
|
converting = (
|
|
ffmpeg.input(file, threads = 0)
|
|
.output(f"{file_formanted}.wav")
|
|
.run(
|
|
cmd=["ffmpeg", "-nostdin"], capture_stdout=True, capture_stderr=True
|
|
)
|
|
)
|
|
print("formanting...")
|
|
os.system(
|
|
'runtime\Scripts\stftpitchshift.exe -i "%s" -q %s -t %s -o "%sFORMANTED"'
|
|
% (file_formanted, Quefrency, Timbre, file_formanted)
|
|
)
|
|
print("formanted!")
|
|
# filepraat = (os.path.abspath(os.getcwd()) + '\\' + file).replace('/','\\')
|
|
# file_formantedpraat = ('"' + os.path.abspath(os.getcwd()) + '/' + 'formanted'.join(file_formanted) + '"').replace('/','\\')
|
|
|
|
out, _ = (
|
|
ffmpeg.input("%sFORMANTED%s" % (file_formanted, ".wav"), threads=0)
|
|
.output("-", format="f32le", acodec="pcm_f32le", ac=1, ar=sr)
|
|
.run(
|
|
cmd=["ffmpeg", "-nostdin"], capture_stdout=True, capture_stderr=True
|
|
)
|
|
)
|
|
|
|
os.remove("%sFORMANTED%s" % (file_formanted, ".wav"))
|
|
os.remove(f"{file_formanted}.wav")
|
|
|
|
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}")
|
|
|
|
conn.close()
|
|
return np.frombuffer(out, np.float32).flatten()
|