From f7130944c05c33face7799dd84c289fa4a12cde9 Mon Sep 17 00:00:00 2001 From: KamioRinn Date: Sat, 27 Jan 2024 06:00:48 +0800 Subject: [PATCH] fix uvr5 hp3 output --- tools/uvr5/vr.py | 3 +++ tools/uvr5/webui.py | 4 ++-- 2 files changed, 5 insertions(+), 2 deletions(-) diff --git a/tools/uvr5/vr.py b/tools/uvr5/vr.py index 448e57f..86fbac4 100644 --- a/tools/uvr5/vr.py +++ b/tools/uvr5/vr.py @@ -110,6 +110,9 @@ class AudioPre: y_spec_m = pred * X_phase v_spec_m = X_spec_m - y_spec_m + if is_hp3 == True: + ins_root,vocal_root = vocal_root,ins_root + if ins_root is not None: if self.data["high_end_process"].startswith("mirroring"): input_high_end_ = spec_utils.mirroring( diff --git a/tools/uvr5/webui.py b/tools/uvr5/webui.py index 97170bf..e09830d 100644 --- a/tools/uvr5/webui.py +++ b/tools/uvr5/webui.py @@ -81,7 +81,7 @@ def uvr(model_name, inp_root, save_root_vocal, paths, save_root_ins, agg, format try: if done == 0: pre_fun._path_audio_( - inp_path, save_root_ins, save_root_vocal, format0 + inp_path, save_root_ins, save_root_vocal, format0, is_hp3=is_hp3 ) infos.append("%s->Success" % (os.path.basename(inp_path))) yield "\n".join(infos) @@ -89,7 +89,7 @@ def uvr(model_name, inp_root, save_root_vocal, paths, save_root_ins, agg, format try: if done == 0: pre_fun._path_audio_( - inp_path, save_root_ins, save_root_vocal, format0 + inp_path, save_root_ins, save_root_vocal, format0, is_hp3=is_hp3 ) infos.append("%s->Success" % (os.path.basename(inp_path))) yield "\n".join(infos)