RVC/Retrieval-based-Voice-Conversion-WebUI
1
0
Fork 0
mirror of https://github.com/RVC-Project/Retrieval-based-Voice-Conversion-WebUI.git synced 2026-01-20 02:51:09 +00:00
Code Issues Packages Projects Releases Wiki Activity

Format code (#455)

Browse Source 
Co-authored-by: github-actions[bot] <github-actions[bot]@users.noreply.github.com>
This commit is contained in:
github-actions[bot]
2023-06-06 14:35:35 +00:00
committed by GitHub
parent 6f1bc7d683
commit 52c97ed464
6 changed files with 63 additions and 47 deletions
Download Patch File Download Diff File Expand all files Collapse all files

64
infer_uvr5.py
View File

@@ -123,7 +123,7 @@ class _audio_pre_:
else:
wav_instrument = spec_utils.cmb_spectrogram_to_wave(y_spec_m, self.mp)
print("%s instruments done" % name)
if(format in ["wav","flac"]):
if format in ["wav", "flac"]:
sf.write(
os.path.join(
ins_root,
@@ -133,18 +133,18 @@ class _audio_pre_:
self.mp.param["sr"],
) #
else:
path=os.path.join(
ins_root, "instrument_{}_{}.wav".format(name, self.data["agg"])
)
path = os.path.join(
ins_root, "instrument_{}_{}.wav".format(name, self.data["agg"])
)
sf.write(
path ,
path,
(np.array(wav_instrument) * 32768).astype("int16"),
self.mp.param["sr"],
)
if(os.path.exists(path)):
if os.path.exists(path):
os.system(
"ffmpeg -i %s -vn %s -q:a 2 -y"
% (path, path[:-4]+".%s"%format)
% (path, path[:-4] + ".%s" % format)
)
if vocal_root is not None:
if self.data["high_end_process"].startswith("mirroring"):
@@ -157,29 +157,31 @@ class _audio_pre_:
else:
wav_vocals = spec_utils.cmb_spectrogram_to_wave(v_spec_m, self.mp)
print("%s vocals done" % name)
if(format in ["wav","flac"]):
if format in ["wav", "flac"]:
sf.write(
os.path.join(
vocal_root, "vocal_{}_{}.{}".format(name, self.data["agg"], format)
vocal_root,
"vocal_{}_{}.{}".format(name, self.data["agg"], format),
),
(np.array(wav_vocals) * 32768).astype("int16"),
self.mp.param["sr"],
)
else:
path=os.path.join(
vocal_root, "vocal_{}_{}.wav".format(name, self.data["agg"])
)
path = os.path.join(
vocal_root, "vocal_{}_{}.wav".format(name, self.data["agg"])
)
sf.write(
path ,
path,
(np.array(wav_vocals) * 32768).astype("int16"),
self.mp.param["sr"],
)
if(os.path.exists(path)):
if os.path.exists(path):
os.system(
"ffmpeg -i %s -vn %s -q:a 2 -y"
% (path, path[:-4]+".%s"%format)
% (path, path[:-4] + ".%s" % format)
)
class _audio_pre_new:
def __init__(self, agg, model_path, device, is_half):
self.model_path = model_path
@@ -288,7 +290,7 @@ class _audio_pre_new:
else:
wav_instrument = spec_utils.cmb_spectrogram_to_wave(y_spec_m, self.mp)
print("%s instruments done" % name)
if(format in ["wav","flac"]):
if format in ["wav", "flac"]:
sf.write(
os.path.join(
ins_root,
@@ -298,18 +300,18 @@ class _audio_pre_new:
self.mp.param["sr"],
) #
else:
path=os.path.join(
ins_root, "instrument_{}_{}.wav".format(name, self.data["agg"])
)
path = os.path.join(
ins_root, "instrument_{}_{}.wav".format(name, self.data["agg"])
)
sf.write(
path ,
path,
(np.array(wav_instrument) * 32768).astype("int16"),
self.mp.param["sr"],
)
if(os.path.exists(path)):
if os.path.exists(path):
os.system(
"ffmpeg -i %s -vn %s -q:a 2 -y"
% (path, path[:-4]+".%s"%format)
% (path, path[:-4] + ".%s" % format)
)
if vocal_root is not None:
if self.data["high_end_process"].startswith("mirroring"):
@@ -322,29 +324,31 @@ class _audio_pre_new:
else:
wav_vocals = spec_utils.cmb_spectrogram_to_wave(v_spec_m, self.mp)
print("%s vocals done" % name)
if(format in ["wav","flac"]):
if format in ["wav", "flac"]:
sf.write(
os.path.join(
vocal_root, "vocal_{}_{}.{}".format(name, self.data["agg"], format)
vocal_root,
"vocal_{}_{}.{}".format(name, self.data["agg"], format),
),
(np.array(wav_vocals) * 32768).astype("int16"),
self.mp.param["sr"],
)
else:
path=os.path.join(
vocal_root, "vocal_{}_{}.wav".format(name, self.data["agg"])
)
path = os.path.join(
vocal_root, "vocal_{}_{}.wav".format(name, self.data["agg"])
)
sf.write(
path ,
path,
(np.array(wav_vocals) * 32768).astype("int16"),
self.mp.param["sr"],
)
if(os.path.exists(path)):
if os.path.exists(path):
os.system(
"ffmpeg -i %s -vn %s -q:a 2 -y"
% (path, path[:-4]+".%s"%format)
% (path, path[:-4] + ".%s" % format)
)
if __name__ == "__main__":
device = "cuda"
is_half = True