mirror of
https://github.com/Anjok07/ultimatevocalremovergui.git
synced 2024-11-24 07:20:10 +01:00
Add files via upload
This commit is contained in:
Anjok07
GitHub
parent
04f29d3bb7
commit
df51d61602
65
UVR.py
65
UVR.py
@ -112,7 +112,7 @@ DEFAULT_DATA = {
|
||||
'demucs_only': False,
|
||||
'split_mode': True,
|
||||
#MDX-Net
|
||||
'demucsmodel': True,
|
||||
'demucsmodel': False,
|
||||
'demucsmodelVR': False,
|
||||
'non_red': False,
|
||||
'noise_reduc': True,
|
||||
@ -124,6 +124,7 @@ DEFAULT_DATA = {
|
||||
'audfile': True,
|
||||
'autocompensate': True,
|
||||
'chunks': 'Auto',
|
||||
'chunks_d': 'Full',
|
||||
'n_fft_scale': 6144,
|
||||
'segment': 'None',
|
||||
'dim_f': 2048,
|
||||
@ -427,6 +428,7 @@ class MainWindow(TkinterDnD.Tk):
|
||||
self.nophaseinst_var = tk.BooleanVar(value=data['nophaseinst'])
|
||||
self.noisereduc_var = tk.BooleanVar(value=data['noise_reduc'])
|
||||
self.chunks_var = tk.StringVar(value=data['chunks'])
|
||||
self.chunks_d_var = tk.StringVar(value=data['chunks_d'])
|
||||
self.noisereduc_s_var = tk.StringVar(value=data['noisereduc_s'])
|
||||
self.mixing_var = tk.StringVar(value=data['mixing']) #dropdown
|
||||
# Models
|
||||
@ -724,6 +726,17 @@ class MainWindow(TkinterDnD.Tk):
|
||||
'55', '60', '65', '70', '75', '80',
|
||||
'85', '90', '95', 'Full')
|
||||
|
||||
# Demucs-Segment
|
||||
self.options_segment_Label = tk.Label(master=self.options_Frame,
|
||||
text='Segment',
|
||||
background='#0e0e0f', font=self.font, foreground='#13a4c9')
|
||||
self.options_segment_Optionmenu = ttk.OptionMenu(self.options_Frame,
|
||||
self.segment_var,
|
||||
None, 'None', '1', '5', '10', '15', '20',
|
||||
'25', '30', '35', '40', '45', '50',
|
||||
'55', '60', '65', '70', '75', '80',
|
||||
'85', '90', '95', '100')
|
||||
|
||||
|
||||
# Overlap
|
||||
self.options_overlap_b_Label = tk.Label(master=self.options_Frame,
|
||||
@ -920,6 +933,12 @@ class MainWindow(TkinterDnD.Tk):
|
||||
self.options_chunks_Optionmenu.place(x=71, y=-2, width=-118, height=7,
|
||||
relx=2/3, rely=3/self.COL2_ROWS, relwidth=1/3, relheight=1/self.COL2_ROWS)
|
||||
|
||||
# Demucs-Segment
|
||||
self.options_segment_Label.place(x=12, y=0, width=0, height=-10,
|
||||
relx=2/3, rely=2/self.COL2_ROWS, relwidth=1/3, relheight=1/self.COL2_ROWS)
|
||||
self.options_segment_Optionmenu.place(x=71, y=-2, width=-118, height=7,
|
||||
relx=2/3, rely=3/self.COL2_ROWS, relwidth=1/3, relheight=1/self.COL2_ROWS)
|
||||
|
||||
# Overlap
|
||||
self.options_overlap_b_Label.place(x=13, y=0, width=0, height=-10,
|
||||
relx=2/3, rely=4/self.COL2_ROWS, relwidth=1/3, relheight=1/self.COL2_ROWS)
|
||||
@ -1242,6 +1261,7 @@ class MainWindow(TkinterDnD.Tk):
|
||||
'audfile': self.audfile_var.get(),
|
||||
'autocompensate': self.autocompensate_var.get(),
|
||||
'chunks': chunks,
|
||||
'chunks_d': self.chunks_d_var.get(),
|
||||
'noisereduc_s': noisereduc_s,
|
||||
'mixing': mixing,
|
||||
'n_fft_scale': self.n_fft_scale_var.get(),
|
||||
@ -1512,6 +1532,8 @@ class MainWindow(TkinterDnD.Tk):
|
||||
self.options_shifts_b_Optionmenu.place_forget()
|
||||
self.options_split_Checkbutton.configure(state=tk.DISABLED)
|
||||
self.options_split_Checkbutton.place_forget()
|
||||
self.options_segment_Label.place_forget()
|
||||
self.options_segment_Optionmenu.place_forget()
|
||||
|
||||
|
||||
elif self.aiModel_var.get() == 'VR Architecture':
|
||||
@ -1594,6 +1616,8 @@ class MainWindow(TkinterDnD.Tk):
|
||||
self.options_shifts_b_Optionmenu.place_forget()
|
||||
self.options_split_Checkbutton.configure(state=tk.DISABLED)
|
||||
self.options_split_Checkbutton.place_forget()
|
||||
self.options_segment_Label.place_forget()
|
||||
self.options_segment_Optionmenu.place_forget()
|
||||
|
||||
elif self.aiModel_var.get() == 'Demucs v3':
|
||||
#Keep for Ensemble & VR Architecture Mode
|
||||
@ -1609,10 +1633,10 @@ class MainWindow(TkinterDnD.Tk):
|
||||
self.options_demucs_stems_Optionmenu.place(x=55, y=-2, width=-85, height=7,
|
||||
relx=1/3, rely=3/self.COL2_ROWS, relwidth=1/3, relheight=1/self.COL2_ROWS)
|
||||
|
||||
# Chunks
|
||||
self.options_chunks_Label.place(x=12, y=0, width=0, height=-10,
|
||||
# Segment
|
||||
self.options_segment_Label.place(x=12, y=0, width=0, height=-10,
|
||||
relx=2/3, rely=2/self.COL2_ROWS, relwidth=1/3, relheight=1/self.COL2_ROWS)
|
||||
self.options_chunks_Optionmenu.place(x=55, y=-2, width=-85, height=7,
|
||||
self.options_segment_Optionmenu.place(x=55, y=-2, width=-85, height=7,
|
||||
relx=2/3, rely=3/self.COL2_ROWS, relwidth=1/3, relheight=1/self.COL2_ROWS)
|
||||
|
||||
# Shifts
|
||||
@ -1679,6 +1703,8 @@ class MainWindow(TkinterDnD.Tk):
|
||||
self.options_algo_Label.place_forget()
|
||||
self.options_algo_Optionmenu.place_forget()
|
||||
self.options_modelFolder_Checkbutton.place_forget()
|
||||
self.options_chunks_Label.place_forget()
|
||||
self.options_chunks_Optionmenu.place_forget()
|
||||
|
||||
elif self.aiModel_var.get() == 'Ensemble Mode':
|
||||
if self.ensChoose_var.get() == 'Manual Ensemble':
|
||||
@ -1740,6 +1766,8 @@ class MainWindow(TkinterDnD.Tk):
|
||||
self.options_shifts_b_Optionmenu.place_forget()
|
||||
self.options_split_Checkbutton.configure(state=tk.DISABLED)
|
||||
self.options_split_Checkbutton.place_forget()
|
||||
self.options_segment_Label.place_forget()
|
||||
self.options_segment_Optionmenu.place_forget()
|
||||
|
||||
|
||||
elif self.ensChoose_var.get() == 'Multi-AI Ensemble':
|
||||
@ -1820,6 +1848,8 @@ class MainWindow(TkinterDnD.Tk):
|
||||
self.options_shifts_b_Optionmenu.place_forget()
|
||||
self.options_split_Checkbutton.configure(state=tk.DISABLED)
|
||||
self.options_split_Checkbutton.place_forget()
|
||||
self.options_segment_Label.place_forget()
|
||||
self.options_segment_Optionmenu.place_forget()
|
||||
else:
|
||||
# Choose Ensemble
|
||||
self.options_ensChoose_Label.place(x=0, y=19, width=0, height=-10,
|
||||
@ -1898,6 +1928,8 @@ class MainWindow(TkinterDnD.Tk):
|
||||
self.options_shifts_b_Optionmenu.place_forget()
|
||||
self.options_split_Checkbutton.configure(state=tk.DISABLED)
|
||||
self.options_split_Checkbutton.place_forget()
|
||||
self.options_segment_Label.place_forget()
|
||||
self.options_segment_Optionmenu.place_forget()
|
||||
|
||||
|
||||
if self.inst_only_var.get() == True:
|
||||
@ -2261,8 +2293,8 @@ class MainWindow(TkinterDnD.Tk):
|
||||
"""
|
||||
top= Toplevel(self)
|
||||
|
||||
top.geometry("670x500")
|
||||
window_height = 670
|
||||
top.geometry("750x500")
|
||||
window_height = 750
|
||||
window_width = 500
|
||||
|
||||
top.title("Advanced Demucs Options")
|
||||
@ -2304,7 +2336,7 @@ class MainWindow(TkinterDnD.Tk):
|
||||
l0=tk.Label(frame0, text='Chunks (Set Manually)', font=("Century Gothic", "9"), foreground='#13a4c9')
|
||||
l0.grid(row=1,column=0,padx=0,pady=10)
|
||||
|
||||
l0=ttk.Entry(frame0, textvariable=self.chunks_var, justify='center')
|
||||
l0=ttk.Entry(frame0, textvariable=self.chunks_d_var, justify='center')
|
||||
l0.grid(row=2,column=0,padx=0,pady=0)
|
||||
|
||||
l0=tk.Label(frame0, text='Chunk Margin', font=("Century Gothic", "9"), foreground='#13a4c9')
|
||||
@ -2325,26 +2357,32 @@ class MainWindow(TkinterDnD.Tk):
|
||||
l0=ttk.Entry(frame0, textvariable=self.overlap_b_var, justify='center')
|
||||
l0.grid(row=8,column=0,padx=0,pady=0)
|
||||
|
||||
l0=ttk.Checkbutton(frame0, text='Save Stems to Model Name Directory', variable=self.audfile_var)
|
||||
l0.grid(row=9,column=0,padx=0,pady=0)
|
||||
l0=tk.Label(frame0, text='Segment', font=("Century Gothic", "9"), foreground='#13a4c9')
|
||||
l0.grid(row=9,column=0,padx=0,pady=10)
|
||||
|
||||
l0=ttk.Entry(frame0, textvariable=self.segment_var, justify='center')
|
||||
l0.grid(row=10,column=0,padx=0,pady=0)
|
||||
|
||||
l0=ttk.Checkbutton(frame0, text='Save Stems to Model & Track Name Directory', variable=self.audfile_var)
|
||||
l0.grid(row=11,column=0,padx=0,pady=0)
|
||||
|
||||
l0=ttk.Checkbutton(frame0, text='Settings Test Mode', variable=self.modelFolder_var)
|
||||
l0.grid(row=10,column=0,padx=0,pady=0)
|
||||
l0.grid(row=12,column=0,padx=0,pady=0)
|
||||
|
||||
# l0=ttk.Checkbutton(frame0, text='Basic Prediction', variable=self.audfile_var)
|
||||
# l0.grid(row=10,column=0,padx=0,pady=0)
|
||||
|
||||
l0=ttk.Button(frame0,text='Open Demucs Model Folder', command=self.open_Modelfolder_de)
|
||||
l0.grid(row=11,column=0,padx=0,pady=0)
|
||||
l0.grid(row=13,column=0,padx=0,pady=0)
|
||||
|
||||
l0=ttk.Button(frame0,text='Back to Main Menu', command=close_win)
|
||||
l0.grid(row=12,column=0,padx=0,pady=10)
|
||||
l0.grid(row=14,column=0,padx=0,pady=10)
|
||||
|
||||
def close_win_self():
|
||||
top.destroy()
|
||||
|
||||
l0=ttk.Button(frame0,text='Close Window', command=close_win_self)
|
||||
l0.grid(row=13,column=0,padx=0,pady=0)
|
||||
l0.grid(row=15,column=0,padx=0,pady=0)
|
||||
|
||||
|
||||
def advanced_mdx_options(self):
|
||||
@ -3435,6 +3473,7 @@ class MainWindow(TkinterDnD.Tk):
|
||||
'audfile': self.audfile_var.get(),
|
||||
'autocompensate': self.autocompensate_var.get(),
|
||||
'chunks': chunks,
|
||||
'chunks_d': self.chunks_d_var.get(),
|
||||
'n_fft_scale': self.n_fft_scale_var.get(),
|
||||
'segment': self.segment_var.get(),
|
||||
'dim_f': self.dim_f_var.get(),
|
||||
|
||||
@ -867,7 +867,7 @@ data = {
|
||||
'export_path': None,
|
||||
'saveFormat': 'Wav',
|
||||
# Processing Options
|
||||
'demucsmodel': True,
|
||||
'demucsmodel': False,
|
||||
'gpu': -1,
|
||||
'chunks': 10,
|
||||
'non_red': False,
|
||||
|
||||
@ -46,23 +46,52 @@ class Predictor():
|
||||
if data['gpu'] == -1:
|
||||
device = torch.device('cpu')
|
||||
|
||||
if data['demucsmodel']:
|
||||
self.demucs = HDemucs(sources=["drums", "bass", "other", "vocals"])
|
||||
widget_text.write(base_text + 'Loading Demucs model... ')
|
||||
update_progress(**progress_kwargs,
|
||||
step=0.05)
|
||||
path_d = Path('models/Demucs_Models')
|
||||
print('What Demucs model was chosen? ', data['DemucsModel'])
|
||||
self.demucs = _gm(name=data['DemucsModel'], repo=path_d)
|
||||
widget_text.write('Done!\n')
|
||||
if 'UVR' in data['DemucsModel']:
|
||||
widget_text.write(base_text + "2 stem model selected.\n")
|
||||
self.demucs = HDemucs(sources=["drums", "bass", "other", "vocals"])
|
||||
widget_text.write(base_text + 'Loading Demucs model... ')
|
||||
update_progress(**progress_kwargs,
|
||||
step=0.05)
|
||||
path_d = Path('models/Demucs_Models')
|
||||
print('What Demucs model was chosen? ', data['DemucsModel'])
|
||||
self.demucs = _gm(name=data['DemucsModel'], repo=path_d)
|
||||
widget_text.write('Done!\n')
|
||||
if 'UVR' in data['DemucsModel']:
|
||||
widget_text.write(base_text + "2 stem model selected.\n")
|
||||
if isinstance(self.demucs, BagOfModels):
|
||||
widget_text.write(base_text + f"Selected model is a bag of {len(self.demucs.models)} models.\n")
|
||||
|
||||
if data['segment'] == 'None':
|
||||
segment = None
|
||||
if isinstance(self.demucs, BagOfModels):
|
||||
widget_text.write(base_text + f"Selected model is a bag of {len(self.demucs.models)} models.\n")
|
||||
if segment is not None:
|
||||
for sub in self.demucs.models:
|
||||
sub.segment = segment
|
||||
else:
|
||||
if segment is not None:
|
||||
sub.segment = segment
|
||||
else:
|
||||
try:
|
||||
segment = int(data['segment'])
|
||||
if isinstance(self.demucs, BagOfModels):
|
||||
if segment is not None:
|
||||
for sub in self.demucs.models:
|
||||
sub.segment = segment
|
||||
else:
|
||||
if segment is not None:
|
||||
sub.segment = segment
|
||||
widget_text.write(base_text + "Segments set to "f"{segment}.\n")
|
||||
except:
|
||||
segment = None
|
||||
if isinstance(self.demucs, BagOfModels):
|
||||
if segment is not None:
|
||||
for sub in self.demucs.models:
|
||||
sub.segment = segment
|
||||
else:
|
||||
if segment is not None:
|
||||
sub.segment = segment
|
||||
|
||||
self.demucs.to(device)
|
||||
self.demucs.eval()
|
||||
|
||||
|
||||
update_progress(**progress_kwargs,
|
||||
step=0.1)
|
||||
|
||||
@ -536,12 +565,16 @@ class Predictor():
|
||||
|
||||
# 1 = demucs only
|
||||
# 0 = onnx only
|
||||
if data['chunks'] == 'Full':
|
||||
chunk_set = 0
|
||||
if data['chunks_d'] == 'Full':
|
||||
if split_mode == True:
|
||||
chunk_set = 0
|
||||
else:
|
||||
widget_text.write(base_text + "Chunk size set to full... \n")
|
||||
chunk_set = 0
|
||||
else:
|
||||
chunk_set = data['chunks']
|
||||
|
||||
if data['chunks'] == 'Auto':
|
||||
if data['chunks_d'] == 'Auto':
|
||||
if split_mode == True:
|
||||
widget_text.write(base_text + "Split Mode is on (Chunks disabled).\n")
|
||||
chunk_set = 0
|
||||
@ -578,16 +611,13 @@ class Predictor():
|
||||
if int(sys_mem) >= int(33):
|
||||
chunk_set = int(0)
|
||||
widget_text.write(base_text + 'Chunk size auto-set to Full... \n')
|
||||
elif data['chunks'] == 'Full':
|
||||
chunk_set = 0
|
||||
widget_text.write(base_text + "Chunk size set to full... \n")
|
||||
else:
|
||||
if split_mode == True:
|
||||
widget_text.write(base_text + "Split Mode is on (Chunks disabled).\n")
|
||||
chunk_set = 0
|
||||
else:
|
||||
widget_text.write(base_text + "Split Mode is off (Chunks enabled).\n")
|
||||
chunk_set = int(data['chunks'])
|
||||
chunk_set = int(data['chunks_d'])
|
||||
widget_text.write(base_text + "Chunk size user-set to "f"{chunk_set}... \n")
|
||||
|
||||
samples = mix.shape[-1]
|
||||
@ -662,12 +692,13 @@ data = {
|
||||
# Processing Options
|
||||
'demucsmodel': True,
|
||||
'gpu': -1,
|
||||
'chunks': 'Full',
|
||||
'chunks_d': 'Full',
|
||||
'modelFolder': False,
|
||||
'voc_only_b': False,
|
||||
'inst_only_b': False,
|
||||
'overlap_b': 0.25,
|
||||
'shifts_b': 2,
|
||||
'segment': 'None',
|
||||
'margin': 44100,
|
||||
'split_mode': False,
|
||||
'compensate': 1.03597672895,
|
||||
@ -675,7 +706,7 @@ data = {
|
||||
'DemucsModel': 'mdx_extra',
|
||||
'audfile': True,
|
||||
}
|
||||
default_chunks = data['chunks']
|
||||
default_chunks = data['chunks_d']
|
||||
|
||||
def update_progress(progress_var, total_files, file_num, step: float = 1):
|
||||
"""Calculate the progress for the progress widget in the GUI"""
|
||||
@ -724,7 +755,7 @@ def main(window: tk.Wm, text_widget: tk.Text, button_widget: tk.Button, progress
|
||||
global split_mode
|
||||
|
||||
# Update default settings
|
||||
default_chunks = data['chunks']
|
||||
default_chunks = data['chunks_d']
|
||||
|
||||
widget_text = text_widget
|
||||
gui_progress_bar = progress_var
|
||||
@ -808,6 +839,23 @@ def main(window: tk.Wm, text_widget: tk.Text, button_widget: tk.Button, progress
|
||||
shift_set = int(data['shifts_b'])
|
||||
split_mode = data['split_mode']
|
||||
|
||||
def determinemusicfileFolderName():
|
||||
"""
|
||||
Determine the name that is used for the folder and appended
|
||||
to the back of the music files
|
||||
"""
|
||||
songFolderName = ''
|
||||
|
||||
if str(music_file):
|
||||
songFolderName += os.path.splitext(os.path.basename(music_file))[0]
|
||||
|
||||
if songFolderName:
|
||||
|
||||
songFolderName = '/' + songFolderName
|
||||
|
||||
|
||||
return songFolderName
|
||||
|
||||
def determinemodelFolderName():
|
||||
"""
|
||||
Determine the name that is used for the folder and appended
|
||||
@ -828,20 +876,26 @@ def main(window: tk.Wm, text_widget: tk.Text, button_widget: tk.Button, progress
|
||||
|
||||
if data['audfile'] == True:
|
||||
modelFolderName = determinemodelFolderName()
|
||||
songFolderName = determinemusicfileFolderName()
|
||||
|
||||
if modelFolderName:
|
||||
folder_path = f'{data["export_path"]}{modelFolderName}'
|
||||
if not os.path.isdir(folder_path):
|
||||
os.mkdir(folder_path)
|
||||
|
||||
if songFolderName:
|
||||
folder_path = f'{data["export_path"]}{modelFolderName}{songFolderName}'
|
||||
if not os.path.isdir(folder_path):
|
||||
os.mkdir(folder_path)
|
||||
|
||||
_mixture = f'{data["input_paths"]}'
|
||||
if data['modelFolder']:
|
||||
try:
|
||||
_basename = f'{data["export_path"]}{modelFolderName}/{file_num}_{str(timestampnum)}_{os.path.splitext(os.path.basename(music_file))[0]}'
|
||||
_basename = f'{data["export_path"]}{modelFolderName}{songFolderName}/{file_num}_{str(timestampnum)}_{os.path.splitext(os.path.basename(music_file))[0]}'
|
||||
except:
|
||||
_basename = f'{data["export_path"]}{modelFolderName}/{file_num}_{str(randomnum)}_{os.path.splitext(os.path.basename(music_file))[0]}'
|
||||
_basename = f'{data["export_path"]}{modelFolderName}{songFolderName}/{file_num}_{str(randomnum)}_{os.path.splitext(os.path.basename(music_file))[0]}'
|
||||
else:
|
||||
_basename = f'{data["export_path"]}{modelFolderName}/{file_num}_{os.path.splitext(os.path.basename(music_file))[0]}'
|
||||
_basename = f'{data["export_path"]}{modelFolderName}{songFolderName}/{file_num}_{os.path.splitext(os.path.basename(music_file))[0]}'
|
||||
else:
|
||||
_mixture = f'{data["input_paths"]}'
|
||||
if data['modelFolder']:
|
||||
|
||||
@ -407,11 +407,11 @@ class Predictor():
|
||||
if 'UVR' in demucs_model_set:
|
||||
sources[3] = (spec_effects(wave=[demucs_out[1],base_out[0]],
|
||||
algorithm=data['mixing'],
|
||||
value=b[3])*float(data['compensate'])) # compensation
|
||||
value=b[3])*float(compensate)) # compensation
|
||||
else:
|
||||
sources[3] = (spec_effects(wave=[demucs_out[3],base_out[0]],
|
||||
algorithm=data['mixing'],
|
||||
value=b[3])*float(data['compensate'])) # compensation
|
||||
value=b[3])*float(compensate)) # compensation
|
||||
|
||||
return sources
|
||||
|
||||
@ -629,7 +629,7 @@ data = {
|
||||
'output_image': True,
|
||||
'voc_only': False,
|
||||
'inst_only': False,
|
||||
'demucsmodel': True,
|
||||
'demucsmodel': False,
|
||||
'chunks': 'auto',
|
||||
'non_red': False,
|
||||
'noisereduc_s': 3,
|
||||
@ -643,6 +643,7 @@ data = {
|
||||
'margin': 44100,
|
||||
'split_mode': False,
|
||||
'compensate': 1.03597672895,
|
||||
'autocompensate': True,
|
||||
'demucs_only': False,
|
||||
'mixing': 'Default',
|
||||
'DemucsModel_MDX': 'UVR_Demucs_Model_1',
|
||||
@ -691,7 +692,8 @@ def main(window: tk.Wm, text_widget: tk.Text, button_widget: tk.Button, progress
|
||||
global model_set
|
||||
global model_set_name
|
||||
global ModelName_2
|
||||
global mdx_model_hash
|
||||
global compensate
|
||||
global autocompensate
|
||||
|
||||
global demucs_model_set
|
||||
|
||||
@ -711,6 +713,7 @@ def main(window: tk.Wm, text_widget: tk.Text, button_widget: tk.Button, progress
|
||||
# Update default settings
|
||||
default_chunks = data['chunks']
|
||||
default_noisereduc_s = data['noisereduc_s']
|
||||
autocompensate = data['autocompensate']
|
||||
|
||||
widget_text = text_widget
|
||||
gui_progress_bar = progress_var
|
||||
@ -736,8 +739,6 @@ def main(window: tk.Wm, text_widget: tk.Text, button_widget: tk.Button, progress
|
||||
except:
|
||||
pass
|
||||
|
||||
n_fft_scale_set=6144
|
||||
dim_f_set=2048
|
||||
|
||||
global nn_arch_sizes
|
||||
global nn_architecture
|
||||
@ -1322,6 +1323,8 @@ def main(window: tk.Wm, text_widget: tk.Text, button_widget: tk.Button, progress
|
||||
mdx_ensem = 'UVR_MDXNET_9662'
|
||||
if data['mdx_ensem'] == 'UVR-MDX-NET Karaoke':
|
||||
mdx_ensem = 'UVR_MDXNET_KARA'
|
||||
if data['mdx_ensem'] == 'UVR-MDX-NET Main':
|
||||
mdx_ensem = 'UVR_MDXNET_Main'
|
||||
if data['mdx_ensem'] == 'Demucs UVR Model 1':
|
||||
mdx_ensem = 'UVR_Demucs_Model_1'
|
||||
if data['mdx_ensem'] == 'Demucs UVR Model 2':
|
||||
@ -1332,15 +1335,26 @@ def main(window: tk.Wm, text_widget: tk.Text, button_widget: tk.Button, progress
|
||||
mdx_ensem = 'mdx_extra_q'
|
||||
|
||||
#MDX-Net Model 2
|
||||
|
||||
|
||||
if data['mdx_ensem_b'] == 'UVR-MDX-NET 1':
|
||||
mdx_ensem_b = 'UVR_MDXNET_1_9703'
|
||||
if os.path.isfile('models/MDX_Net_Models/UVR_MDXNET_1_9703.onnx'):
|
||||
mdx_ensem_b = 'UVR_MDXNET_1_9703'
|
||||
else:
|
||||
mdx_ensem_b = 'UVR_MDXNET_9703'
|
||||
if data['mdx_ensem_b'] == 'UVR-MDX-NET 2':
|
||||
mdx_ensem_b = 'UVR_MDXNET_2_9682'
|
||||
if os.path.isfile('models/MDX_Net_Models/UVR_MDXNET_2_9682.onnx'):
|
||||
mdx_ensem_b = 'UVR_MDXNET_2_9682'
|
||||
else:
|
||||
mdx_ensem_b = 'UVR_MDXNET_9682'
|
||||
if data['mdx_ensem_b'] == 'UVR-MDX-NET 3':
|
||||
mdx_ensem_b = 'UVR_MDXNET_3_9662'
|
||||
if os.path.isfile('models/MDX_Net_Models/UVR_MDXNET_3_9662.onnx'):
|
||||
mdx_ensem_b = 'UVR_MDXNET_3_9662'
|
||||
else:
|
||||
mdx_ensem_b = 'UVR_MDXNET_9662'
|
||||
if data['mdx_ensem_b'] == 'UVR-MDX-NET Karaoke':
|
||||
mdx_ensem_b = 'UVR_MDXNET_KARA'
|
||||
if data['mdx_ensem_b'] == 'UVR-MDX-NET Main':
|
||||
mdx_ensem_b = 'UVR_MDXNET_Main'
|
||||
if data['mdx_ensem_b'] == 'Demucs UVR Model 1':
|
||||
mdx_ensem_b = 'UVR_Demucs_Model_1'
|
||||
if data['mdx_ensem_b'] == 'Demucs UVR Model 2':
|
||||
@ -1351,6 +1365,8 @@ def main(window: tk.Wm, text_widget: tk.Text, button_widget: tk.Button, progress
|
||||
mdx_ensem_b = 'mdx_extra_q'
|
||||
if data['mdx_ensem_b'] == 'No Model':
|
||||
mdx_ensem_b = 'pass'
|
||||
|
||||
|
||||
|
||||
if data['vr_ensem'] == 'No Model' and data['vr_ensem_mdx_a'] == 'No Model' and data['vr_ensem_mdx_b'] == 'No Model' and data['vr_ensem_mdx_c'] == 'No Model':
|
||||
mdx_vr = [
|
||||
@ -1544,7 +1560,7 @@ def main(window: tk.Wm, text_widget: tk.Text, button_widget: tk.Button, progress
|
||||
|
||||
if data['ensChoose'] == 'Basic VR Ensemble':
|
||||
loops = Basic_Ensem
|
||||
ensefolder = 'Basic_Ensemble_Outputs'
|
||||
ensefolder = 'Basic_VR_Outputs'
|
||||
if data['vr_ensem_c'] == 'No Model' and data['vr_ensem_d'] == 'No Model' and data['vr_ensem_e'] == 'No Model':
|
||||
ensemode = 'Basic_Ensemble' + '_' + vr_ensem_a_name + '_' + vr_ensem_b_name
|
||||
elif data['vr_ensem_c'] == 'No Model' and data['vr_ensem_d'] == 'No Model':
|
||||
@ -1575,7 +1591,7 @@ def main(window: tk.Wm, text_widget: tk.Text, button_widget: tk.Button, progress
|
||||
ensemode = 'Vocal_Models'
|
||||
if data['ensChoose'] == 'Multi-AI Ensemble':
|
||||
loops = mdx_vr
|
||||
ensefolder = 'MDX_VR_Ensemble_Outputs'
|
||||
ensefolder = 'Multi_AI_Ensemble_Outputs'
|
||||
if data['vr_ensem'] == 'No Model' and data['vr_ensem_mdx_a'] == 'No Model' and data['vr_ensem_mdx_b'] == 'No Model' and data['vr_ensem_mdx_c'] == 'No Model':
|
||||
ensemode = 'MDX-Net_Models'
|
||||
elif data['vr_ensem_mdx_a'] == 'No Model' and data['vr_ensem_mdx_b'] == 'No Model' and data['vr_ensem_mdx_c'] == 'No Model':
|
||||
@ -1607,6 +1623,13 @@ def main(window: tk.Wm, text_widget: tk.Text, button_widget: tk.Button, progress
|
||||
step=0)
|
||||
|
||||
try:
|
||||
|
||||
if float(data['noisereduc_s']) >= 11:
|
||||
text_widget.write('Error: Noise Reduction only supports values between 0-10.\nPlease set a value between 0-10 (with or without decimals) and try again.')
|
||||
progress_var.set(0)
|
||||
button_widget.configure(state=tk.NORMAL) # Enable Button
|
||||
return
|
||||
|
||||
total, used, free = shutil.disk_usage("/")
|
||||
|
||||
total_space = int(total/1.074e+9)
|
||||
@ -2064,12 +2087,24 @@ def main(window: tk.Wm, text_widget: tk.Text, button_widget: tk.Button, progress
|
||||
modeltype = 'v'
|
||||
demucs_model_set = data['DemucsModel_MDX']
|
||||
noise_pro = 'MDX-NET_Noise_Profile_14_kHz'
|
||||
if autocompensate == True:
|
||||
compensate = 1.03597672895
|
||||
else:
|
||||
compensate = data['compensate']
|
||||
n_fft_scale_set=6144
|
||||
dim_f_set=2048
|
||||
if mdx_name == 'UVR_MDXNET_2_9682':
|
||||
demucs_only = 'off'
|
||||
model_set = 'UVR_MDXNET_2_9682.onnx'
|
||||
model_set_name = 'UVR_MDXNET_2_9682'
|
||||
modeltype = 'v'
|
||||
noise_pro = 'MDX-NET_Noise_Profile_14_kHz'
|
||||
if autocompensate == True:
|
||||
compensate = 1.03597672895
|
||||
else:
|
||||
compensate = data['compensate']
|
||||
n_fft_scale_set=6144
|
||||
dim_f_set=2048
|
||||
if mdx_name == 'UVR_MDXNET_3_9662':
|
||||
demucs_only = 'off'
|
||||
model_set = 'UVR_MDXNET_3_9662.onnx'
|
||||
@ -2077,12 +2112,24 @@ def main(window: tk.Wm, text_widget: tk.Text, button_widget: tk.Button, progress
|
||||
modeltype = 'v'
|
||||
demucs_model_set = data['DemucsModel_MDX']
|
||||
noise_pro = 'MDX-NET_Noise_Profile_14_kHz'
|
||||
if autocompensate == True:
|
||||
compensate = 1.03597672895
|
||||
else:
|
||||
compensate = data['compensate']
|
||||
n_fft_scale_set=6144
|
||||
dim_f_set=2048
|
||||
if mdx_name == 'UVR_MDXNET_KARA':
|
||||
demucs_only = 'off'
|
||||
model_set = 'UVR_MDXNET_KARA.onnx'
|
||||
model_set_name = 'UVR_MDXNET_KARA'
|
||||
modeltype = 'v'
|
||||
noise_pro = 'MDX-NET_Noise_Profile_14_kHz'
|
||||
if autocompensate == True:
|
||||
compensate = 1.03597672895
|
||||
else:
|
||||
compensate = data['compensate']
|
||||
n_fft_scale_set=6144
|
||||
dim_f_set=2048
|
||||
if mdx_name == 'UVR_MDXNET_9703':
|
||||
demucs_only = 'off'
|
||||
model_set = 'UVR_MDXNET_9703.onnx'
|
||||
@ -2090,6 +2137,12 @@ def main(window: tk.Wm, text_widget: tk.Text, button_widget: tk.Button, progress
|
||||
modeltype = 'v'
|
||||
demucs_model_set = data['DemucsModel_MDX']
|
||||
noise_pro = 'MDX-NET_Noise_Profile_14_kHz'
|
||||
if autocompensate == True:
|
||||
compensate = 1.03597672895
|
||||
else:
|
||||
compensate = data['compensate']
|
||||
n_fft_scale_set=6144
|
||||
dim_f_set=2048
|
||||
if mdx_name == 'UVR_MDXNET_9682':
|
||||
demucs_only = 'off'
|
||||
model_set = 'UVR_MDXNET_9682.onnx'
|
||||
@ -2097,6 +2150,12 @@ def main(window: tk.Wm, text_widget: tk.Text, button_widget: tk.Button, progress
|
||||
modeltype = 'v'
|
||||
demucs_model_set = data['DemucsModel_MDX']
|
||||
noise_pro = 'MDX-NET_Noise_Profile_14_kHz'
|
||||
if autocompensate == True:
|
||||
compensate = 1.03597672895
|
||||
else:
|
||||
compensate = data['compensate']
|
||||
n_fft_scale_set=6144
|
||||
dim_f_set=2048
|
||||
if mdx_name == 'UVR_MDXNET_9662':
|
||||
demucs_only = 'off'
|
||||
model_set = 'UVR_MDXNET_9662.onnx'
|
||||
@ -2104,6 +2163,12 @@ def main(window: tk.Wm, text_widget: tk.Text, button_widget: tk.Button, progress
|
||||
modeltype = 'v'
|
||||
demucs_model_set = data['DemucsModel_MDX']
|
||||
noise_pro = 'MDX-NET_Noise_Profile_14_kHz'
|
||||
if autocompensate == True:
|
||||
compensate = 1.03597672895
|
||||
else:
|
||||
compensate = data['compensate']
|
||||
n_fft_scale_set=6144
|
||||
dim_f_set=2048
|
||||
if mdx_name == 'UVR_MDXNET_KARA':
|
||||
demucs_only = 'off'
|
||||
model_set = 'UVR_MDXNET_KARA.onnx'
|
||||
@ -2111,6 +2176,25 @@ def main(window: tk.Wm, text_widget: tk.Text, button_widget: tk.Button, progress
|
||||
modeltype = 'v'
|
||||
demucs_model_set = data['DemucsModel_MDX']
|
||||
noise_pro = 'MDX-NET_Noise_Profile_14_kHz'
|
||||
if autocompensate == True:
|
||||
compensate = 1.03597672895
|
||||
else:
|
||||
compensate = data['compensate']
|
||||
n_fft_scale_set=6144
|
||||
dim_f_set=2048
|
||||
if mdx_name == 'UVR_MDXNET_Main':
|
||||
demucs_only = 'off'
|
||||
model_set = 'UVR_MDXNET_Main.onnx'
|
||||
model_set_name = 'UVR_MDXNET_Main'
|
||||
modeltype = 'v'
|
||||
demucs_model_set = data['DemucsModel_MDX']
|
||||
noise_pro = 'MDX-NET_Noise_Profile_17_kHz'
|
||||
if autocompensate == True:
|
||||
compensate = 1.075
|
||||
else:
|
||||
compensate = data['compensate']
|
||||
n_fft_scale_set=7680
|
||||
dim_f_set=3072
|
||||
if 'Demucs' in mdx_name:
|
||||
demucs_only = 'on'
|
||||
demucs_switch = 'on'
|
||||
@ -2527,7 +2611,7 @@ def main(window: tk.Wm, text_widget: tk.Text, button_widget: tk.Button, progress
|
||||
{
|
||||
'algorithm':'min_mag',
|
||||
'model_params':'lib_v5/modelparams/1band_sr44100_hl512.json',
|
||||
'output':'{}_User_Ensembled_(Min Spec)'.format(trackname1),
|
||||
'output':'{}_Manual_Ensemble_(Min Spec)'.format(trackname1),
|
||||
'type': 'Instrumentals'
|
||||
}
|
||||
]
|
||||
@ -2536,7 +2620,7 @@ def main(window: tk.Wm, text_widget: tk.Text, button_widget: tk.Button, progress
|
||||
{
|
||||
'algorithm':'max_mag',
|
||||
'model_params':'lib_v5/modelparams/1band_sr44100_hl512.json',
|
||||
'output': '{}_User_Ensembled_(Max Spec)'.format(trackname1),
|
||||
'output': '{}_Manual_Ensemble_(Max Spec)'.format(trackname1),
|
||||
'type': 'Vocals'
|
||||
}
|
||||
]
|
||||
|
||||
Loading…
Reference in New Issue
Block a user