Cool_Tools/Retrieval-based-Voice-Conversion-WebUI
1
0
Fork 0
mirror of synced 2025-01-19 01:24:07 +01:00
Code Releases Wiki Activity

Format code (#1153)

Browse Source 
Co-authored-by: github-actions[bot] <github-actions[bot]@users.noreply.github.com>
This commit is contained in:
github-actions[bot] 2023-09-01 12:03:35 +08:00 committed by GitHub
parent 0c75454ddf
commit d634c2727e
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 35 additions and 16 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

39
gui_v1.py
View File

@ -303,8 +303,16 @@ if __name__ == "__main__":
), ),
], ],
[ [
sg.Checkbox(i18n("输入降噪"), key="I_noise_reduce", enable_events=True), sg.Checkbox(
sg.Checkbox(i18n("输出降噪"), key="O_noise_reduce", enable_events=True), i18n("输入降噪"),
key="I_noise_reduce",
enable_events=True,
),
sg.Checkbox(
i18n("输出降噪"),
key="O_noise_reduce",
enable_events=True,
),
], ],
], ],
title=i18n("性能设置"), title=i18n("性能设置"),
@ -375,9 +383,9 @@ if __name__ == "__main__":
json.dump(settings, j) json.dump(settings, j)
if event == "stop_vc" and self.flag_vc == True: if event == "stop_vc" and self.flag_vc == True:
self.flag_vc = False self.flag_vc = False
# Parameter hot update # Parameter hot update
if event == 'threhold': if event == "threhold":
self.config.threhold = values["threhold"] self.config.threhold = values["threhold"]
elif event == "pitch": elif event == "pitch":
self.config.pitch = values["pitch"] self.config.pitch = values["pitch"]
@ -389,14 +397,14 @@ if __name__ == "__main__":
self.rvc.change_index_rate(values["index_rate"]) self.rvc.change_index_rate(values["index_rate"])
elif event in ["pm", "harvest", "crepe", "rmvpe"]: elif event in ["pm", "harvest", "crepe", "rmvpe"]:
self.config.f0method = event self.config.f0method = event
elif event == 'I_noise_reduce': elif event == "I_noise_reduce":
self.config.I_noise_reduce = values["I_noise_reduce"] self.config.I_noise_reduce = values["I_noise_reduce"]
elif event == 'O_noise_reduce': elif event == "O_noise_reduce":
self.config.O_noise_reduce = values["O_noise_reduce"] self.config.O_noise_reduce = values["O_noise_reduce"]
elif event != "start_vc" and self.flag_vc == True: elif event != "start_vc" and self.flag_vc == True:
# Other parameters do not support hot update # Other parameters do not support hot update
self.flag_vc = False self.flag_vc = False
def set_values(self, values): def set_values(self, values):
if len(values["pth_path"].strip()) == 0: if len(values["pth_path"].strip()) == 0:
sg.popup(i18n("请选择pth文件")) sg.popup(i18n("请选择pth文件"))
@ -502,9 +510,20 @@ if __name__ == "__main__":
self.sola_buffer: torch.Tensor = torch.zeros( self.sola_buffer: torch.Tensor = torch.zeros(
self.crossfade_frame, device=device, dtype=torch.float32 self.crossfade_frame, device=device, dtype=torch.float32
) )
self.fade_in_window: torch.Tensor = torch.sin(0.5 * np.pi * torch.linspace( self.fade_in_window: torch.Tensor = (
0.0, 1.0, steps=self.crossfade_frame, device=device, dtype=torch.float32 torch.sin(
)) ** 2 0.5
* np.pi
* torch.linspace(
0.0,
1.0,
steps=self.crossfade_frame,
device=device,
dtype=torch.float32,
)
)
** 2
)
self.fade_out_window: torch.Tensor = 1 - self.fade_in_window self.fade_out_window: torch.Tensor = 1 - self.fade_in_window
self.resampler = tat.Resample( self.resampler = tat.Resample(
orig_freq=self.config.samplerate, new_freq=16000, dtype=torch.float32 orig_freq=self.config.samplerate, new_freq=16000, dtype=torch.float32

12
tools/rvc_for_realtime.py
View File

@ -112,17 +112,17 @@ class RVC:
self.is_half = config.is_half self.is_half = config.is_half
except: except:
print(traceback.format_exc()) print(traceback.format_exc())
def change_key(self, new_key): def change_key(self, new_key):
self.f0_up_key = new_key self.f0_up_key = new_key
def change_index_rate(self, new_index_rate): def change_index_rate(self, new_index_rate):
if new_index_rate != 0 and self.index_rate == 0: if new_index_rate != 0 and self.index_rate == 0:
self.index = faiss.read_index(self.index_path) self.index = faiss.read_index(self.index_path)
self.big_npy = self.index.reconstruct_n(0, self.index.ntotal) self.big_npy = self.index.reconstruct_n(0, self.index.ntotal)
print("index search enabled") print("index search enabled")
self.index_rate = new_index_rate self.index_rate = new_index_rate
def get_f0_post(self, f0): def get_f0_post(self, f0):
f0_min = self.f0_min f0_min = self.f0_min
f0_max = self.f0_max f0_max = self.f0_max