save_log=config.getboolean("model", "save_history"),
log_name=config.get("model", "history_filename"))
logging.info("Saving weights...")
model.save(args.weights)
elif args.mode == "separate":
logging.info("Preparing to separate vocals from instrumentals...")
mixture = Song(logging, "a mixture", config)
mixture.load_file(args.file)
mixture.compute_stft(keep_spectrogram=True)
dump_data = True if args.dump_data.lower() in ("yes", "true", "y", "t",
"1") else False
save_accompaniment = True if args.save_accompaniment.lower() in (
"yes", "true", "y", "t", "1") else False
if dump_data is True:
mixture.dump_amplitude("original")
mixture.dump_spectrogram("original")
model = Model(logging, config)
model.build()
if os.path.isfile(args.weights):
model.load(args.weights)
else:
logging.critical("Couldn't find a weights file.")
sys.exit(11)
if dump_data is True:
model.isolate(mixture,
args.output,
save_accompaniment=save_accompaniment,
save_original_mask=True,
save_original_probabilities=True)
mixture.dump_spectrogram("processed")
else:
