Spaces:
Running
Running
| # Save loudness normalized (-14 LUFS) musdb-XL audio files for delimiter evaluation | |
| import os | |
| import argparse | |
| import tqdm | |
| import musdb | |
| import soundfile as sf | |
| import librosa | |
| import pyloudnorm as pyln | |
| from utils import db2linear, str2bool | |
| tqdm.monitor_interval = 0 | |
| def main(): | |
| parser = argparse.ArgumentParser(description="model test.py") | |
| parser.add_argument( | |
| "--target", | |
| type=str, | |
| default="vocals", | |
| help="target source. all, vocals, drums, bass, other", | |
| ) | |
| parser.add_argument("--data_root", type=str, default="/path/to/musdb_XL") | |
| parser.add_argument( | |
| "--data_root_hq", | |
| type=str, | |
| default="/data1/Music/musdb18hq", | |
| help="this is used when saving loud-norm stem of musdb-XL") | |
| parser.add_argument( | |
| "--output_directory", | |
| type=str, | |
| default="/path/to/results", | |
| ) | |
| parser.add_argument("--exp_name", type=str, default="delimit_6_s") | |
| parser.add_argument( | |
| "--save_16k_mono", | |
| type=str2bool, | |
| default=False, | |
| help="Save 16k mono wav files for FAD evaluation.", | |
| ) | |
| args, _ = parser.parse_known_args() | |
| os.makedirs(args.output_directory, exist_ok=True) | |
| meter = pyln.Meter(44100) | |
| args.test_output_dir = f"{args.output_directory}/test/{args.exp_name}" | |
| test_tracks = musdb.DB(root=args.data_root, subsets="test", is_wav=True) | |
| if args.target != "mixture": # In this file, args.target should not be "mixture" | |
| hq_tracks = musdb.DB(root=args.data_root_hq, subsets='test', is_wav=True) | |
| for idx, track in tqdm.tqdm(enumerate(test_tracks)): | |
| track_name = track.name | |
| if ( | |
| os.path.basename(args.data_root) == "musdb18hq" | |
| and track_name == "PR - Oh No" | |
| ): # We have to consider this exception because 'PR - Oh No' mixture.wav is left-panned. We will use the linear mixture instead. | |
| # Please refer https://github.com/jeonchangbin49/musdb-XL/blob/main/make_L_and_XL.py | |
| track_audio = ( | |
| track.targets["vocals"].audio | |
| + track.targets["drums"].audio | |
| + track.targets["bass"].audio | |
| + track.targets["other"].audio | |
| ) | |
| else: | |
| track_audio = track.audio | |
| delimiter_track = librosa.load(f"{args.test_output_dir}/{track_name}/all.wav", sr=44100, mono=False)[0].T | |
| print(track_name) | |
| if args.target != "mixture": | |
| hq_track = hq_tracks[idx] | |
| hq_audio = hq_track.audio | |
| hq_stem = hq_track.targets[args.target].audio | |
| hq_samplewise_gain = track_audio / (hq_audio + 1e-8) | |
| XL_stem = hq_samplewise_gain * hq_stem | |
| XL_samplewise_gain = delimiter_track / (track_audio + 1e-8) | |
| delimiter_stem = XL_samplewise_gain * XL_stem | |
| sf.write( | |
| f"{args.test_output_dir}/{track_name}/{args.target}.wav", delimiter_stem, 44100 | |
| ) | |
| if __name__ == "__main__": | |
| main() | |