Add files via upload

This commit is contained in:
Anjok07 2021-05-05 02:30:04 -05:00 committed by GitHub
parent 6357ebcb0c
commit 7fe2353774
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
4 changed files with 75 additions and 74 deletions

View File

@ -98,9 +98,9 @@ def main():
p.add_argument('--pretrained_modelB', '-B', type=str, default='models/MGM-v5-4Band-44100-BETA2.pth') ##DON'T CHNGE
p.add_argument('--pretrained_modelC', '-C', type=str, default='models/HighPrecison_4band_1.pth') ##DON'T CHNGE
p.add_argument('--pretrained_modelD', '-Da', type=str, default='models/HighPrecison_4band_2.pth') ##DON'T CHNGE
p.add_argument('--pretrained_modelE', '-E', type=str, default='models/BigLayer_4band_1.pth') ##DON'T CHNGE
p.add_argument('--pretrained_modelF', '-F', type=str, default='models/BigLayer_4band_2.pth') ##DON'T CHNGE
p.add_argument('--pretrained_modelG', '-G', type=str, default='models/BigLayer_4band_3.pth') ##DON'T CHNGE
p.add_argument('--pretrained_modelE', '-E', type=str, default='models/NewLayer_4band_1.pth') ##DON'T CHNGE
p.add_argument('--pretrained_modelF', '-F', type=str, default='models/NewLayer_4band_2.pth') ##DON'T CHNGE
p.add_argument('--pretrained_modelG', '-G', type=str, default='models/NewLayer_4band_3.pth') ##DON'T CHNGE
p.add_argument('--deepextraction', '-D', action='store_true')
p.add_argument('--saveindivsep', '-s', action='store_true')
p.add_argument('--input', '-i', required=True)
@ -113,7 +113,7 @@ def main():
p.add_argument('--postprocess', '-p', action='store_true')
p.add_argument('--tta', '-t', action='store_true')
p.add_argument('--high_end_process', '-H', type=str, choices=['none', 'bypass', 'correlation'], default='none')
p.add_argument('--aggressiveness', '-A', type=float, default=0.05)
p.add_argument('--aggressiveness', '-A', type=float, default=0.09)
args = p.parse_args()
####################################################-INERATION1-####################################################
@ -591,7 +591,7 @@ def main():
start_time = time.time()
print('loading BigLayer_4band_1...', end=' ')
print('loading NewLayer_4band_1...', end=' ')
device = torch.device('cpu')
model = nets.CascadedASPPNet(mp.param['bins'] * 2)
@ -706,7 +706,7 @@ def main():
start_time = time.time()
print('loading BigLayer_4band_2...', end=' ')
print('loading NewLayer_4band_2...', end=' ')
device = torch.device('cpu')
model = nets.CascadedASPPNet(mp.param['bins'] * 2)
@ -821,7 +821,7 @@ def main():
start_time = time.time()
print('loading BigLayer_4band_3...', end=' ')
print('loading NewLayer_4band_3...', end=' ')
device = torch.device('cpu')
model = nets.CascadedASPPNet(mp.param['bins'] * 2)
@ -931,23 +931,23 @@ def main():
print('Ensembling Instrumentals...')
os.system("python lib/spec_utils.py -a min_mag -m modelparams/1band_sr44100_hl512.json ensembled/temp/1_MGM-v5-4Band-44100-BETA1_Instruments.wav ensembled/temp/2_MGM-v5-4Band-44100-BETA2_Instruments.wav -o ensembled/temp/1E2E_ensam1")
os.system("python lib/spec_utils.py -a min_mag -m modelparams/1band_sr44100_hl512.json ensembled/temp/3_HighPrecison_4band_1_Instruments.wav ensembled/temp/4_HighPrecison_4band_2_Instruments.wav -o ensembled/temp/3E4E_ensam1")
os.system("python lib/spec_utils.py -a min_mag -m modelparams/1band_sr44100_hl512.json ensembled/temp/5_BigLayer_4band_1_Instruments.wav ensembled/temp/6_BigLayer_4band_2_Instruments.wav -o ensembled/temp/5E6E_ensam3")
os.system("python lib/spec_utils.py -a min_mag -m modelparams/1band_sr44100_hl512.json ensembled/temp/5_NewLayer_4band_1_Instruments.wav ensembled/temp/6_NewLayer_4band_2_Instruments.wav -o ensembled/temp/5E6E_ensam3")
os.system("python lib/spec_utils.py -a min_mag -m modelparams/1band_sr44100_hl512.json ensembled/temp/1E2E_ensam1_v.wav ensembled/temp/3E4E_ensam1_v.wav -o ensembled/temp/1E2E3E4E_ensam4")
os.system("python lib/spec_utils.py -a min_mag -m modelparams/1band_sr44100_hl512.json ensembled/temp/1E2E3E4E_ensam4_v.wav ensembled/temp/5E6E_ensam3_v.wav -o ensembled/temp/A6_ensam5")
os.system("python lib/spec_utils.py -a min_mag -m modelparams/1band_sr44100_hl512.json ensembled/temp/7_BigLayer_4band_3_Instruments.wav ensembled/temp/A6_ensam5_v.wav -o ensembled/temp/Complete")
os.system("python lib/spec_utils.py -a min_mag -m modelparams/1band_sr44100_hl512.json ensembled/temp/7_NewLayer_4band_3_Instruments.wav ensembled/temp/A6_ensam5_v.wav -o ensembled/temp/Complete")
print('Ensembling Vocals...')
os.system("python lib/spec_utils.py -a max_mag -m modelparams/1band_sr44100_hl512.json ensembled/temp/1_MGM-v5-4Band-44100-BETA1_Vocals.wav ensembled/temp/2_MGM-v5-4Band-44100-BETA2_Vocals.wav -o ensembled/temp/1E2EV_ensam1")
os.system("python lib/spec_utils.py -a max_mag -m modelparams/1band_sr44100_hl512.json ensembled/temp/3_HighPrecison_4band_1_Vocals.wav ensembled/temp/4_HighPrecison_4band_2_Vocals.wav -o ensembled/temp/3E4EV_ensam1")
os.system("python lib/spec_utils.py -a max_mag -m modelparams/1band_sr44100_hl512.json ensembled/temp/5_BigLayer_4band_1_Vocals.wav ensembled/temp/6_BigLayer_4band_2_Vocals.wav -o ensembled/temp/5E6EV_ensam3")
os.system("python lib/spec_utils.py -a max_mag -m modelparams/1band_sr44100_hl512.json ensembled/temp/5_NewLayer_4band_1_Vocals.wav ensembled/temp/6_NewLayer_4band_2_Vocals.wav -o ensembled/temp/5E6EV_ensam3")
os.system("python lib/spec_utils.py -a max_mag -m modelparams/1band_sr44100_hl512.json ensembled/temp/1E2EV_ensam1_v.wav ensembled/temp/3E4EV_ensam1_v.wav -o ensembled/temp/1E2E3E4EV_ensam4")
os.system("python lib/spec_utils.py -a max_mag -m modelparams/1band_sr44100_hl512.json ensembled/temp/1E2E3E4EV_ensam4_v.wav ensembled/temp/5E6EV_ensam3_v.wav -o ensembled/temp/A6V_ensam5")
os.system("python lib/spec_utils.py -a max_mag -m modelparams/1band_sr44100_hl512.json ensembled/temp/7_BigLayer_4band_3_Vocals.wav ensembled/temp/A6V_ensam5_v.wav -o ensembled/temp/CompleteV")
os.system("python lib/spec_utils.py -a max_mag -m modelparams/1band_sr44100_hl512.json ensembled/temp/7_NewLayer_4band_3_Vocals.wav ensembled/temp/A6V_ensam5_v.wav -o ensembled/temp/CompleteV")
print('Performing Deep Extraction...')
os.system("python lib/spec_utils.py -a min_mag -m modelparams/1band_sr44100_hl512.json ensembled/temp/Complete_v.wav ensembled/temp/CompleteV_v.wav -o ensembled/temp/difftemp")
os.system("python lib/diffext.py ensembled/temp/Complete_v.wav ensembled/temp/difftemp_v.wav ensembled/temp/aligned-difftemp_v.wav ensembled/temp/subtracted-difftemp_v.wav")
os.rename('ensembled/temp/subtracted-difftemp_v.wav', 'ensembled/{}_4BAND_Ensembled_DeepExtraction_Instrumental.wav'.format(basename))
os.system("python lib/spec_utils.py -a invertB -m modelparams/1band_sr44100_hl512.json ensembled/temp/Complete_v.wav ensembled/temp/difftemp_v.wav -o ensembled/temp/difftempC")
os.rename('ensembled/temp/difftempC_v.wav', 'ensembled/{}_4BAND_Ensembled_DeepExtraction_Instrumental.wav'.format(basename))
os.rename('ensembled/temp/Complete_v.wav', 'ensembled/{}_4BAND_Ensembled_Instrumental.wav'.format(basename))
os.rename('ensembled/temp/CompleteV_v.wav', 'ensembled/{}_4BAND_Ensembled_Vocals.wav'.format(basename))
print('Deep Extraction Complete!')
@ -955,19 +955,19 @@ def main():
print('Ensembling Instrumentals...')
os.system("python lib/spec_utils.py -a min_mag -m modelparams/1band_sr44100_hl512.json ensembled/temp/1_MGM-v5-4Band-44100-BETA1_Instruments.wav ensembled/temp/2_MGM-v5-4Band-44100-BETA2_Instruments.wav -o ensembled/temp/1E2E_ensam1")
os.system("python lib/spec_utils.py -a min_mag -m modelparams/1band_sr44100_hl512.json ensembled/temp/3_HighPrecison_4band_1_Instruments.wav ensembled/temp/4_HighPrecison_4band_2_Instruments.wav -o ensembled/temp/3E4E_ensam1")
os.system("python lib/spec_utils.py -a min_mag -m modelparams/1band_sr44100_hl512.json ensembled/temp/5_BigLayer_4band_1_Instruments.wav ensembled/temp/6_BigLayer_4band_2_Instruments.wav -o ensembled/temp/5E6E_ensam3")
os.system("python lib/spec_utils.py -a min_mag -m modelparams/1band_sr44100_hl512.json ensembled/temp/5_NewLayer_4band_1_Instruments.wav ensembled/temp/6_NewLayer_4band_2_Instruments.wav -o ensembled/temp/5E6E_ensam3")
os.system("python lib/spec_utils.py -a min_mag -m modelparams/1band_sr44100_hl512.json ensembled/temp/1E2E_ensam1_v.wav ensembled/temp/3E4E_ensam1_v.wav -o ensembled/temp/1E2E3E4E_ensam4")
os.system("python lib/spec_utils.py -a min_mag -m modelparams/1band_sr44100_hl512.json ensembled/temp/1E2E3E4E_ensam4_v.wav ensembled/temp/5E6E_ensam3_v.wav -o ensembled/temp/A6_ensam5")
os.system("python lib/spec_utils.py -a min_mag -m modelparams/1band_sr44100_hl512.json ensembled/temp/7_BigLayer_4band_3_Instruments.wav ensembled/temp/A6_ensam5_v.wav -o ensembled/temp/Complete")
os.system("python lib/spec_utils.py -a min_mag -m modelparams/1band_sr44100_hl512.json ensembled/temp/7_NewLayer_4band_3_Instruments.wav ensembled/temp/A6_ensam5_v.wav -o ensembled/temp/Complete")
os.rename('ensembled/temp/Complete_v.wav', 'ensembled/{}_4BAND_Ensembled_Instrumental.wav'.format(basename))
print('Ensembling Vocals...')
os.system("python lib/spec_utils.py -a max_mag -m modelparams/1band_sr44100_hl512.json ensembled/temp/1_MGM-v5-4Band-44100-BETA1_Vocals.wav ensembled/temp/2_MGM-v5-4Band-44100-BETA2_Vocals.wav -o ensembled/temp/1E2EV_ensam1")
os.system("python lib/spec_utils.py -a max_mag -m modelparams/1band_sr44100_hl512.json ensembled/temp/3_HighPrecison_4band_1_Vocals.wav ensembled/temp/4_HighPrecison_4band_2_Vocals.wav -o ensembled/temp/3E4EV_ensam1")
os.system("python lib/spec_utils.py -a max_mag -m modelparams/1band_sr44100_hl512.json ensembled/temp/5_BigLayer_4band_1_Vocals.wav ensembled/temp/6_BigLayer_4band_2_Vocals.wav -o ensembled/temp/5E6EV_ensam3")
os.system("python lib/spec_utils.py -a max_mag -m modelparams/1band_sr44100_hl512.json ensembled/temp/5_NewLayer_4band_1_Vocals.wav ensembled/temp/6_NewLayer_4band_2_Vocals.wav -o ensembled/temp/5E6EV_ensam3")
os.system("python lib/spec_utils.py -a max_mag -m modelparams/1band_sr44100_hl512.json ensembled/temp/1E2EV_ensam1_v.wav ensembled/temp/3E4EV_ensam1_v.wav -o ensembled/temp/1E2E3E4EV_ensam4")
os.system("python lib/spec_utils.py -a max_mag -m modelparams/1band_sr44100_hl512.json ensembled/temp/1E2E3E4EV_ensam4_v.wav ensembled/temp/5E6EV_ensam3_v.wav -o ensembled/temp/A6V_ensam5")
os.system("python lib/spec_utils.py -a max_mag -m modelparams/1band_sr44100_hl512.json ensembled/temp/7_BigLayer_4band_3_Vocals.wav ensembled/temp/A6V_ensam5_v.wav -o ensembled/temp/CompleteV")
os.system("python lib/spec_utils.py -a max_mag -m modelparams/1band_sr44100_hl512.json ensembled/temp/7_NewLayer_4band_3_Vocals.wav ensembled/temp/A6V_ensam5_v.wav -o ensembled/temp/CompleteV")
os.rename('ensembled/temp/CompleteV_v.wav', 'ensembled/{}_4BAND_Ensembled_Vocals.wav'.format(basename))
if args.saveindivsep:
@ -976,16 +976,16 @@ def main():
os.rename('ensembled/temp/2_MGM-v5-4Band-44100-BETA2_Instruments.wav', 'separated/{}_MGM-v5-4Band-44100-BETA2_Instruments.wav'.format(basename))
os.rename('ensembled/temp/3_HighPrecison_4band_1_Instruments.wav', 'separated/{}_HighPrecison_4band_1_Instruments.wav'.format(basename))
os.rename('ensembled/temp/4_HighPrecison_4band_2_Instruments.wav', 'separated/{}_HighPrecison_4band_2_Instruments.wav'.format(basename))
os.rename('ensembled/temp/5_BigLayer_4band_1_Instruments.wav', 'separated/{}_BigLayer_4band_1_Instruments.wav'.format(basename))
os.rename('ensembled/temp/6_BigLayer_4band_2_Instruments.wav', 'separated/{}_BigLayer_4band_2_Instruments.wav'.format(basename))
os.rename('ensembled/temp/7_BigLayer_4band_3_Instruments.wav', 'separated/{}_BigLayer_4band_3_Instruments.wav'.format(basename))
os.rename('ensembled/temp/5_NewLayer_4band_1_Instruments.wav', 'separated/{}_NewLayer_4band_1_Instruments.wav'.format(basename))
os.rename('ensembled/temp/6_NewLayer_4band_2_Instruments.wav', 'separated/{}_NewLayer_4band_2_Instruments.wav'.format(basename))
os.rename('ensembled/temp/7_NewLayer_4band_3_Instruments.wav', 'separated/{}_NewLayer_4band_3_Instruments.wav'.format(basename))
os.rename('ensembled/temp/1_MGM-v5-4Band-44100-BETA1_Vocals.wav', 'separated/{}_MGM-v5-4Band-44100-BETA1_Vocals.wav'.format(basename))
os.rename('ensembled/temp/2_MGM-v5-4Band-44100-BETA2_Vocals.wav', 'separated/{}_MGM-v5-4Band-44100-BETA2_Vocals.wav'.format(basename))
os.rename('ensembled/temp/3_HighPrecison_4band_1_Vocals.wav', 'separated/{}_HighPrecison_4band_1_Vocals.wav'.format(basename))
os.rename('ensembled/temp/4_HighPrecison_4band_2_Vocals.wav', 'separated/{}_HighPrecison_4band_2_Vocals.wav'.format(basename))
os.rename('ensembled/temp/5_BigLayer_4band_1_Vocals.wav', 'separated/{}_BigLayer_4band_1_Vocals.wav'.format(basename))
os.rename('ensembled/temp/6_BigLayer_4band_2_Vocals.wav', 'separated/{}_BigLayer_4band_2_Vocals.wav'.format(basename))
os.rename('ensembled/temp/7_BigLayer_4band_3_Vocals.wav', 'separated/{}_BigLayer_4band_3_Vocals.wav'.format(basename))
os.rename('ensembled/temp/5_NewLayer_4band_1_Vocals.wav', 'separated/{}_NewLayer_4band_1_Vocals.wav'.format(basename))
os.rename('ensembled/temp/6_NewLayer_4band_2_Vocals.wav', 'separated/{}_NewLayer_4band_2_Vocals.wav'.format(basename))
os.rename('ensembled/temp/7_NewLayer_4band_3_Vocals.wav', 'separated/{}_NewLayer_4band_3_Vocals.wav'.format(basename))
os.remove("ensembled/temp/A6V_ensam5_v.wav")
os.remove("ensembled/temp/1E2E3E4EV_ensam4_v.wav")
os.remove("ensembled/temp/5E6EV_ensam3_v.wav")

View File

@ -33,9 +33,9 @@ ECHO l. LOFI_2band_iter5_2
ECHO m. LOFI_2band_iter5_2
ECHO n. HighPrecison_4band_1
ECHO o. HighPrecison_4band_2
ECHO p. BigLayer_4band_1.pth
ECHO q. BigLayer_4band_2.pth
ECHO r. BigLayer_4band_3.pth
ECHO p. NewLayer_4band_1.pth
ECHO q. NewLayer_4band_2.pth
ECHO r. NewLayer_4band_3.pth
ECHO s. Ensemble 4Band Models (7 Models) - Final Outputs Save to Ensembled Folder!
ECHO t. Ensemble All Models (12 Models) - Final Outputs Save to Ensembled Folder!
set choice=
@ -453,15 +453,15 @@ ECHO "%choice%" is not valid, try again
ECHO.
goto start
:deep
ECHO Running Model BigLayer_4band_1
set model=BigLayer_4band_1
ECHO Running Model NewLayer_4band_1
set model=NewLayer_4band_1
cd /d %~dp0
python inference.py -g 0 -m modelparams\4band_44100.json -w 352 -D -n 129605KB -P models\%model%.pth -t -i %1
goto start
:default
ECHO Running Model BigLayer_4band_1
set model=BigLayer_4band_1
ECHO Running Model NewLayer_4band_1
set model=NewLayer_4band_1
cd /d %~dp0
python inference.py -g 0 -m modelparams\4band_44100.json -w 352 -n 129605KB -P models\%model%.pth -t -i %1
@ -480,16 +480,16 @@ ECHO.
goto start
:deep
ECHO Running Model BigLayer_4band_2
set model=BigLayer_4band_2
ECHO Running Model NewLayer_4band_2
set model=NewLayer_4band_2
cd /d %~dp0
python inference.py -g 0 -m modelparams\4band_44100.json -w 352 -D -n 129605KB -P models\%model%.pth -t -i %1
goto end
:default
ECHO Running Model BigLayer_4band_2
set model=BigLayer_4band_2
ECHO Running Model NewLayer_4band_2
set model=NewLayer_4band_2
cd /d %~dp0
python inference.py -g 0 -m modelparams\4band_44100.json -w 352 -n 129605KB -P models\%model%.pth -t -i %1
@ -509,7 +509,7 @@ goto start
:deep
ECHO Running Model
set model=BigLayer_4band_3
set model=NewLayer_4band_3
cd /d %~dp0
python inference.py -g 0 -m modelparams\4band_44100.json -w 352 -D -n 129605KB -P models\%model%.pth -t -i %1
@ -517,7 +517,7 @@ goto end
:default
ECHO Running Model
set model=BigLayer_4band_3
set model=NewLayer_4band_3
cd /d %~dp0
python inference.py -g 0 -m modelparams\4band_44100.json -w 352 -n 129605KB -P models\%model%.pth -t -i %1
@ -594,9 +594,9 @@ ECHO l. LOFI_2band_iter5_2
ECHO m. LOFI_2band_iter5_2
ECHO n. HighPrecison_4band_1
ECHO o. HighPrecison_4band_2
ECHO p. BigLayer_4band_1
ECHO q. BigLayer_4band_2
ECHO r. BigLayer_4band_3
ECHO p. NewLayer_4band_1
ECHO q. NewLayer_4band_2
ECHO r. NewLayer_4band_3
ECHO s. Ensemble 4Band Models (7 Models) - Final Outputs Save to Ensembled Folder!
ECHO t. Ensemble All Models (12 Models) - Final Outputs Save to Ensembled Folder!
set choice=
@ -1014,15 +1014,15 @@ ECHO "%choice%" is not valid, try again
ECHO.
goto start
:deep
ECHO Running Model BigLayer_4band_1
set model=BigLayer_4band_1
ECHO Running Model NewLayer_4band_1
set model=NewLayer_4band_1
cd /d %~dp0
python inference.py -g 0 -m modelparams\4band_44100.json -w 352 -D -n 129605KB -P models\%model%.pth -i %1
goto start
:default
ECHO Running Model BigLayer_4band_1
set model=BigLayer_4band_1
ECHO Running Model NewLayer_4band_1
set model=NewLayer_4band_1
cd /d %~dp0
python inference.py -g 0 -m modelparams\4band_44100.json -w 352 -n 129605KB -P models\%model%.pth -i %1
@ -1041,16 +1041,16 @@ ECHO.
goto start
:deep
ECHO Running Model BigLayer_4band_2
set model=BigLayer_4band_2
ECHO Running Model NewLayer_4band_2
set model=NewLayer_4band_2
cd /d %~dp0
python inference.py -g 0 -m modelparams\4band_44100.json -w 352 -D -n 129605KB -P models\%model%.pth -i %1
goto end
:default
ECHO Running Model BigLayer_4band_2
set model=BigLayer_4band_2
ECHO Running Model NewLayer_4band_2
set model=NewLayer_4band_2
cd /d %~dp0
python inference.py -g 0 -m modelparams\4band_44100.json -w 352 -n 129605KB -P models\%model%.pth -i %1
@ -1070,7 +1070,7 @@ goto start
:deep
ECHO Running Model
set model=BigLayer_4band_3
set model=NewLayer_4band_3
cd /d %~dp0
python inference.py -g 0 -m modelparams\4band_44100.json -w 352 -D -n 129605KB -P models\%model%.pth -i %1
@ -1078,7 +1078,7 @@ goto end
:default
ECHO Running Model
set model=BigLayer_4band_3
set model=NewLayer_4band_3
cd /d %~dp0
python inference.py -g 0 -m modelparams\4band_44100.json -w 352 -n 129605KB -P models\%model%.pth -i %1
@ -1134,5 +1134,6 @@ ECHO Ensemble All 44100-Models
cd /d %~dp0
python allmodels_ens_inference.py -g 0 -w 352 -i %1
goto end
:end
pause

View File

@ -98,9 +98,9 @@ def main():
p.add_argument('--pretrained_modelB', '-B', type=str, default='models/MGM-v5-4Band-44100-BETA2.pth') ##DON'T CHNGE INERATION-2
p.add_argument('--pretrained_modelC', '-C', type=str, default='models/HighPrecison_4band_1.pth') ##DON'T CHNGE INERATION-3
p.add_argument('--pretrained_modelD', '-Da', type=str, default='models/HighPrecison_4band_2.pth') ##DON'T CHNGE INERATION-4
p.add_argument('--pretrained_modelE', '-E', type=str, default='models/BigLayer_4band_1.pth') ##DON'T CHNGE INERATION-5
p.add_argument('--pretrained_modelF', '-F', type=str, default='models/BigLayer_4band_2.pth') ##DON'T CHNGE INERATION-6
p.add_argument('--pretrained_modelG', '-G', type=str, default='models/BigLayer_4band_3.pth') ##DON'T CHNGE INERATION-7
p.add_argument('--pretrained_modelE', '-E', type=str, default='models/NewLayer_4band_1.pth') ##DON'T CHNGE INERATION-5
p.add_argument('--pretrained_modelF', '-F', type=str, default='models/NewLayer_4band_2.pth') ##DON'T CHNGE INERATION-6
p.add_argument('--pretrained_modelG', '-G', type=str, default='models/NewLayer_4band_3.pth') ##DON'T CHNGE INERATION-7
p.add_argument('--pretrained_modelH', '-Hm', type=str, default='models/MGM-v5-MIDSIDE-44100-BETA1.pth') ##DON'T CHNGE INERATION-8
p.add_argument('--pretrained_modelI', '-Im', type=str, default='models/MGM-v5-MIDSIDE-44100-BETA2.pth') ##DON'T CHNGE INERATION-9
p.add_argument('--pretrained_modelJ', '-J', type=str, default='models/MGM-v5-3Band-44100-BETA.pth') ##DON'T CHNGE INERATION-10
@ -603,7 +603,7 @@ def main():
start_time = time.time()
print('loading BigLayer_4band_1...', end=' ')
print('loading NewLayer_4band_1...', end=' ')
device = torch.device('cpu')
model = nets.CascadedASPPNet(mp.param['bins'] * 2)
@ -718,7 +718,7 @@ def main():
start_time = time.time()
print('loading BigLayer_4band_2...', end=' ')
print('loading NewLayer_4band_2...', end=' ')
device = torch.device('cpu')
model = nets.CascadedASPPNet(mp.param['bins'] * 2)
@ -833,7 +833,7 @@ def main():
start_time = time.time()
print('loading BigLayer_4band_3...', end=' ')
print('loading NewLayer_4band_3...', end=' ')
device = torch.device('cpu')
model = nets.CascadedASPPNet(mp.param['bins'] * 2)
@ -1749,10 +1749,10 @@ def main():
print('Ensembling Instrumentals...')
os.system("python lib/spec_utils.py -a min_mag -m modelparams/1band_sr44100_hl512.json ensembled/temp/1_MGM-v5-4Band-44100-BETA1_Instruments.wav ensembled/temp/2_MGM-v5-4Band-44100-BETA2_Instruments.wav -o ensembled/temp/1E2E_ensam1")
os.system("python lib/spec_utils.py -a min_mag -m modelparams/1band_sr44100_hl512.json ensembled/temp/3_HighPrecison_4band_1_Instruments.wav ensembled/temp/4_HighPrecison_4band_2_Instruments.wav -o ensembled/temp/3E4E_ensam1")
os.system("python lib/spec_utils.py -a min_mag -m modelparams/1band_sr44100_hl512.json ensembled/temp/5_BigLayer_4band_1_Instruments.wav ensembled/temp/6_BigLayer_4band_2_Instruments.wav -o ensembled/temp/5E6E_ensam3")
os.system("python lib/spec_utils.py -a min_mag -m modelparams/1band_sr44100_hl512.json ensembled/temp/5_NewLayer_4band_1_Instruments.wav ensembled/temp/6_NewLayer_4band_2_Instruments.wav -o ensembled/temp/5E6E_ensam3")
os.system("python lib/spec_utils.py -a min_mag -m modelparams/1band_sr44100_hl512.json ensembled/temp/1E2E_ensam1_v.wav ensembled/temp/3E4E_ensam1_v.wav -o ensembled/temp/1E2E3E4E_ensam4")
os.system("python lib/spec_utils.py -a min_mag -m modelparams/1band_sr44100_hl512.json ensembled/temp/1E2E3E4E_ensam4_v.wav ensembled/temp/5E6E_ensam3_v.wav -o ensembled/temp/A6_ensam5")
os.system("python lib/spec_utils.py -a min_mag -m modelparams/1band_sr44100_hl512.json ensembled/temp/7_BigLayer_4band_3_Instruments.wav ensembled/temp/A6_ensam5_v.wav -o ensembled/temp/1STHALF")
os.system("python lib/spec_utils.py -a min_mag -m modelparams/1band_sr44100_hl512.json ensembled/temp/7_NewLayer_4band_3_Instruments.wav ensembled/temp/A6_ensam5_v.wav -o ensembled/temp/1STHALF")
os.system("python lib/spec_utils.py -a min_mag -m modelparams/1band_sr44100_hl512.json ensembled/temp/8_MGM-v5-MIDSIDE-44100-BETA1_Instruments.wav ensembled/temp/9_MGM-v5-MIDSIDE-44100-BETA2_Instruments.wav -o ensembled/temp/8E9E_ensam1")
os.system("python lib/spec_utils.py -a min_mag -m modelparams/1band_sr44100_hl512.json ensembled/temp/10_MGM-v5-3Band-44100-BETA_Instruments.wav ensembled/temp/13_LOFI_2band-1_33966KB_Instruments.wav -o ensembled/temp/10E13E_ensam3")
os.system("python lib/spec_utils.py -a min_mag -m modelparams/1band_sr44100_hl512.json ensembled/temp/8E9E_ensam1_v.wav ensembled/temp/10E13E_ensam3_v.wav -o ensembled/temp/8E9E10E13E_ensam4")
@ -1762,10 +1762,10 @@ def main():
print('Ensembling Vocals...')
os.system("python lib/spec_utils.py -a max_mag -m modelparams/1band_sr44100_hl512.json ensembled/temp/1_MGM-v5-4Band-44100-BETA1_Vocals.wav ensembled/temp/2_MGM-v5-4Band-44100-BETA2_Vocals.wav -o ensembled/temp/1E2EV_ensam1")
os.system("python lib/spec_utils.py -a max_mag -m modelparams/1band_sr44100_hl512.json ensembled/temp/3_HighPrecison_4band_1_Vocals.wav ensembled/temp/4_HighPrecison_4band_2_Vocals.wav -o ensembled/temp/3E4EV_ensam1")
os.system("python lib/spec_utils.py -a max_mag -m modelparams/1band_sr44100_hl512.json ensembled/temp/5_BigLayer_4band_1_Vocals.wav ensembled/temp/6_BigLayer_4band_2_Vocals.wav -o ensembled/temp/5E6EV_ensam3")
os.system("python lib/spec_utils.py -a max_mag -m modelparams/1band_sr44100_hl512.json ensembled/temp/5_NewLayer_4band_1_Vocals.wav ensembled/temp/6_NewLayer_4band_2_Vocals.wav -o ensembled/temp/5E6EV_ensam3")
os.system("python lib/spec_utils.py -a max_mag -m modelparams/1band_sr44100_hl512.json ensembled/temp/1E2EV_ensam1_v.wav ensembled/temp/3E4EV_ensam1_v.wav -o ensembled/temp/1E2E3E4EV_ensam4")
os.system("python lib/spec_utils.py -a max_mag -m modelparams/1band_sr44100_hl512.json ensembled/temp/1E2E3E4EV_ensam4_v.wav ensembled/temp/5E6EV_ensam3_v.wav -o ensembled/temp/A6V_ensam5")
os.system("python lib/spec_utils.py -a max_mag -m modelparams/1band_sr44100_hl512.json ensembled/temp/7_BigLayer_4band_3_Vocals.wav ensembled/temp/A6V_ensam5_v.wav -o ensembled/temp/1STHALFV")
os.system("python lib/spec_utils.py -a max_mag -m modelparams/1band_sr44100_hl512.json ensembled/temp/7_NewLayer_4band_3_Vocals.wav ensembled/temp/A6V_ensam5_v.wav -o ensembled/temp/1STHALFV")
os.system("python lib/spec_utils.py -a max_mag -m modelparams/1band_sr44100_hl512.json ensembled/temp/8_MGM-v5-MIDSIDE-44100-BETA1_Vocals.wav ensembled/temp/9_MGM-v5-MIDSIDE-44100-BETA2_Vocals.wav -o ensembled/temp/8E9EV_ensam1")
os.system("python lib/spec_utils.py -a max_mag -m modelparams/1band_sr44100_hl512.json ensembled/temp/10_MGM-v5-3Band-44100-BETA_Vocals.wav ensembled/temp/13_LOFI_2band-1_33966KB_Vocals.wav -o ensembled/temp/10E13EV_ensam3")
os.system("python lib/spec_utils.py -a max_mag -m modelparams/1band_sr44100_hl512.json ensembled/temp/8E9EV_ensam1_v.wav ensembled/temp/10E13EV_ensam3_v.wav -o ensembled/temp/8E9E10E13EV_ensam4")
@ -1774,8 +1774,8 @@ def main():
print('Performing Deep Extraction...')
os.system("python lib/spec_utils.py -a min_mag -m modelparams/1band_sr44100_hl512.json ensembled/temp/Complete_v.wav ensembled/temp/CompleteV_v.wav -o ensembled/temp/difftemp")
os.system("python lib/diffext.py ensembled/temp/Complete_v.wav ensembled/temp/difftemp_v.wav ensembled/temp/aligned-difftemp_v.wav ensembled/temp/subtracted-difftemp_v.wav")
os.rename('ensembled/temp/subtracted-difftemp_v.wav', 'ensembled/{}_ALLMODELS_Ensembled_DeepExtraction_Instrumental.wav'.format(basename))
os.system("python lib/spec_utils.py -a invertB -m modelparams/1band_sr44100_hl512.json ensembled/temp/Complete_v.wav ensembled/temp/difftemp_v.wav -o ensembled/temp/difftempC")
os.rename('ensembled/temp/difftempC_v.wav', 'ensembled/{}_ALLMODELS_Ensembled_DeepExtraction_Instrumental.wav'.format(basename))
os.rename('ensembled/temp/Complete_v.wav', 'ensembled/{}_ALLMODELS_Ensembled_Instrumental.wav'.format(basename))
os.rename('ensembled/temp/CompleteV_v.wav', 'ensembled/{}_ALLMODELS_Ensembled_Vocals.wav'.format(basename))
print('Deep Extraction Complete!')
@ -1783,10 +1783,10 @@ def main():
print('Ensembling Instrumentals...')
os.system("python lib/spec_utils.py -a min_mag -m modelparams/1band_sr44100_hl512.json ensembled/temp/1_MGM-v5-4Band-44100-BETA1_Instruments.wav ensembled/temp/2_MGM-v5-4Band-44100-BETA2_Instruments.wav -o ensembled/temp/1E2E_ensam1")
os.system("python lib/spec_utils.py -a min_mag -m modelparams/1band_sr44100_hl512.json ensembled/temp/3_HighPrecison_4band_1_Instruments.wav ensembled/temp/4_HighPrecison_4band_2_Instruments.wav -o ensembled/temp/3E4E_ensam1")
os.system("python lib/spec_utils.py -a min_mag -m modelparams/1band_sr44100_hl512.json ensembled/temp/5_BigLayer_4band_1_Instruments.wav ensembled/temp/6_BigLayer_4band_2_Instruments.wav -o ensembled/temp/5E6E_ensam3")
os.system("python lib/spec_utils.py -a min_mag -m modelparams/1band_sr44100_hl512.json ensembled/temp/5_NewLayer_4band_1_Instruments.wav ensembled/temp/6_NewLayer_4band_2_Instruments.wav -o ensembled/temp/5E6E_ensam3")
os.system("python lib/spec_utils.py -a min_mag -m modelparams/1band_sr44100_hl512.json ensembled/temp/1E2E_ensam1_v.wav ensembled/temp/3E4E_ensam1_v.wav -o ensembled/temp/1E2E3E4E_ensam4")
os.system("python lib/spec_utils.py -a min_mag -m modelparams/1band_sr44100_hl512.json ensembled/temp/1E2E3E4E_ensam4_v.wav ensembled/temp/5E6E_ensam3_v.wav -o ensembled/temp/A6_ensam5")
os.system("python lib/spec_utils.py -a min_mag -m modelparams/1band_sr44100_hl512.json ensembled/temp/7_BigLayer_4band_3_Instruments.wav ensembled/temp/A6_ensam5_v.wav -o ensembled/temp/1STHALF")
os.system("python lib/spec_utils.py -a min_mag -m modelparams/1band_sr44100_hl512.json ensembled/temp/7_NewLayer_4band_3_Instruments.wav ensembled/temp/A6_ensam5_v.wav -o ensembled/temp/1STHALF")
os.system("python lib/spec_utils.py -a min_mag -m modelparams/1band_sr44100_hl512.json ensembled/temp/8_MGM-v5-MIDSIDE-44100-BETA1_Instruments.wav ensembled/temp/9_MGM-v5-MIDSIDE-44100-BETA2_Instruments.wav -o ensembled/temp/8E9E_ensam1")
os.system("python lib/spec_utils.py -a min_mag -m modelparams/1band_sr44100_hl512.json ensembled/temp/10_MGM-v5-3Band-44100-BETA_Instruments.wav ensembled/temp/13_LOFI_2band-1_33966KB_Instruments.wav -o ensembled/temp/10E13E_ensam3")
os.system("python lib/spec_utils.py -a min_mag -m modelparams/1band_sr44100_hl512.json ensembled/temp/8E9E_ensam1_v.wav ensembled/temp/10E13E_ensam3_v.wav -o ensembled/temp/8E9E10E13E_ensam4")
@ -1797,10 +1797,10 @@ def main():
print('Ensembling Vocals...')
os.system("python lib/spec_utils.py -a max_mag -m modelparams/1band_sr44100_hl512.json ensembled/temp/1_MGM-v5-4Band-44100-BETA1_Vocals.wav ensembled/temp/2_MGM-v5-4Band-44100-BETA2_Vocals.wav -o ensembled/temp/1E2EV_ensam1")
os.system("python lib/spec_utils.py -a max_mag -m modelparams/1band_sr44100_hl512.json ensembled/temp/3_HighPrecison_4band_1_Vocals.wav ensembled/temp/4_HighPrecison_4band_2_Vocals.wav -o ensembled/temp/3E4EV_ensam1")
os.system("python lib/spec_utils.py -a max_mag -m modelparams/1band_sr44100_hl512.json ensembled/temp/5_BigLayer_4band_1_Vocals.wav ensembled/temp/6_BigLayer_4band_2_Vocals.wav -o ensembled/temp/5E6EV_ensam3")
os.system("python lib/spec_utils.py -a max_mag -m modelparams/1band_sr44100_hl512.json ensembled/temp/5_NewLayer_4band_1_Vocals.wav ensembled/temp/6_NewLayer_4band_2_Vocals.wav -o ensembled/temp/5E6EV_ensam3")
os.system("python lib/spec_utils.py -a max_mag -m modelparams/1band_sr44100_hl512.json ensembled/temp/1E2EV_ensam1_v.wav ensembled/temp/3E4EV_ensam1_v.wav -o ensembled/temp/1E2E3E4EV_ensam4")
os.system("python lib/spec_utils.py -a max_mag -m modelparams/1band_sr44100_hl512.json ensembled/temp/1E2E3E4EV_ensam4_v.wav ensembled/temp/5E6EV_ensam3_v.wav -o ensembled/temp/A6V_ensam5")
os.system("python lib/spec_utils.py -a max_mag -m modelparams/1band_sr44100_hl512.json ensembled/temp/7_BigLayer_4band_3_Vocals.wav ensembled/temp/A6V_ensam5_v.wav -o ensembled/temp/1STHALFV")
os.system("python lib/spec_utils.py -a max_mag -m modelparams/1band_sr44100_hl512.json ensembled/temp/7_NewLayer_4band_3_Vocals.wav ensembled/temp/A6V_ensam5_v.wav -o ensembled/temp/1STHALFV")
os.system("python lib/spec_utils.py -a max_mag -m modelparams/1band_sr44100_hl512.json ensembled/temp/8_MGM-v5-MIDSIDE-44100-BETA1_Vocals.wav ensembled/temp/9_MGM-v5-MIDSIDE-44100-BETA2_Vocals.wav -o ensembled/temp/8E9EV_ensam1")
os.system("python lib/spec_utils.py -a max_mag -m modelparams/1band_sr44100_hl512.json ensembled/temp/10_MGM-v5-3Band-44100-BETA_Vocals.wav ensembled/temp/13_LOFI_2band-1_33966KB_Vocals.wav -o ensembled/temp/10E13EV_ensam3")
os.system("python lib/spec_utils.py -a max_mag -m modelparams/1band_sr44100_hl512.json ensembled/temp/8E9EV_ensam1_v.wav ensembled/temp/10E13EV_ensam3_v.wav -o ensembled/temp/8E9E10E13EV_ensam4")
@ -1814,16 +1814,16 @@ def main():
os.rename('ensembled/temp/2_MGM-v5-4Band-44100-BETA2_Instruments.wav', 'separated/{}_MGM-v5-4Band-44100-BETA2_Instruments.wav'.format(basename))
os.rename('ensembled/temp/3_HighPrecison_4band_1_Instruments.wav', 'separated/{}_HighPrecison_4band_1_Instruments.wav'.format(basename))
os.rename('ensembled/temp/4_HighPrecison_4band_2_Instruments.wav', 'separated/{}_HighPrecison_4band_2_Instruments.wav'.format(basename))
os.rename('ensembled/temp/5_BigLayer_4band_1_Instruments.wav', 'separated/{}_BigLayer_4band_1_Instruments.wav'.format(basename))
os.rename('ensembled/temp/6_BigLayer_4band_2_Instruments.wav', 'separated/{}_BigLayer_4band_2_Instruments.wav'.format(basename))
os.rename('ensembled/temp/7_BigLayer_4band_3_Instruments.wav', 'separated/{}_BigLayer_4band_3_Instruments.wav'.format(basename))
os.rename('ensembled/temp/5_NewLayer_4band_1_Instruments.wav', 'separated/{}_NewLayer_4band_1_Instruments.wav'.format(basename))
os.rename('ensembled/temp/6_NewLayer_4band_2_Instruments.wav', 'separated/{}_NewLayer_4band_2_Instruments.wav'.format(basename))
os.rename('ensembled/temp/7_NewLayer_4band_3_Instruments.wav', 'separated/{}_NewLayer_4band_3_Instruments.wav'.format(basename))
os.rename('ensembled/temp/1_MGM-v5-4Band-44100-BETA1_Vocals.wav', 'separated/{}_MGM-v5-4Band-44100-BETA1_Vocals.wav'.format(basename))
os.rename('ensembled/temp/2_MGM-v5-4Band-44100-BETA2_Vocals.wav', 'separated/{}_MGM-v5-4Band-44100-BETA2_Vocals.wav'.format(basename))
os.rename('ensembled/temp/3_HighPrecison_4band_1_Vocals.wav', 'separated/{}_HighPrecison_4band_1_Vocals.wav'.format(basename))
os.rename('ensembled/temp/4_HighPrecison_4band_2_Vocals.wav', 'separated/{}_HighPrecison_4band_2_Vocals.wav'.format(basename))
os.rename('ensembled/temp/5_BigLayer_4band_1_Vocals.wav', 'separated/{}_BigLayer_4band_1_Vocals.wav'.format(basename))
os.rename('ensembled/temp/6_BigLayer_4band_2_Vocals.wav', 'separated/{}_BigLayer_4band_2_Vocals.wav'.format(basename))
os.rename('ensembled/temp/7_BigLayer_4band_3_Vocals.wav', 'separated/{}_BigLayer_4band_3_Vocals.wav'.format(basename))
os.rename('ensembled/temp/5_NewLayer_4band_1_Vocals.wav', 'separated/{}_NewLayer_4band_1_Vocals.wav'.format(basename))
os.rename('ensembled/temp/6_NewLayer_4band_2_Vocals.wav', 'separated/{}_NewLayer_4band_2_Vocals.wav'.format(basename))
os.rename('ensembled/temp/7_NewLayer_4band_3_Vocals.wav', 'separated/{}_NewLayer_4band_3_Vocals.wav'.format(basename))
os.rename('ensembled/temp/8_MGM-v5-MIDSIDE-44100-BETA1_Instruments.wav', 'separated/{}_MGM-v5-MIDSIDE-44100-BETA1_Instruments.wav'.format(basename))
os.rename('ensembled/temp/9_MGM-v5-MIDSIDE-44100-BETA2_Instruments.wav', 'separated/{}_MGM-v5-MIDSIDE-44100-BETA2_Instruments.wav'.format(basename))
os.rename('ensembled/temp/10_MGM-v5-3Band-44100-BETA_Instruments.wav', 'separated/{}_MGM-v5-3Band-44100-BETA_Instruments.wav'.format(basename))

View File

@ -104,7 +104,7 @@ def main():
p.add_argument('--postprocess', '-p', action='store_true')
p.add_argument('--tta', '-t', action='store_true')
p.add_argument('--high_end_process', '-H', type=str, choices=['none', 'bypass', 'correlation'], default='none')
p.add_argument('--aggressiveness', '-A', type=float, default=0.05)
p.add_argument('--aggressiveness', '-A', type=float, default=0.07)
args = p.parse_args()
if args.nn_architecture == 'default':
@ -232,8 +232,8 @@ def main():
print('Performing Deep Extraction...')
os.system("python lib/spec_utils.py -a min_mag -m modelparams/1band_sr44100_hl512.json ensembled/temp/tempI.wav ensembled/temp/tempV.wav -o ensembled/temp/difftemp")
os.system("python lib/diffext.py ensembled/temp/tempI.wav ensembled/temp/difftemp_v.wav ensembled/temp/aligned-difftemp_v.wav ensembled/temp/subtracted-difftemp_v.wav")
os.rename('ensembled/temp/subtracted-difftemp_v.wav', 'separated/{}_{}_DeepExtraction_Instruments.wav'.format(basename, model_name))
os.system("python lib/spec_utils.py -a invertB -m modelparams/1band_sr44100_hl512.json ensembled/temp/tempI.wav ensembled/temp/difftemp_v.wav -o ensembled/temp/difftemp")
os.rename('ensembled/temp/difftemp_v.wav', 'separated/{}_{}_DeepExtraction_Instruments.wav'.format(basename, model_name))
print('Complete!')
print('Total time: {0:.{1}f}s'.format(time.time() - start_time, 1))