def test_generate_tsv_from_jams():
list_jams = [os.path.join(absolute_dir_path, "material", "5.jams")]
generated_tsv = os.path.join(absolute_dir_path, "generated",
"generated_from_jams.tsv")
generate_tsv_from_jams(list_jams, generated_tsv)
df_gen = pd.read_csv(generated_tsv, sep="\t")
df_mat = pd.read_csv(os.path.join(absolute_dir_path, "material",
"generated_from_jams.tsv"),
sep="\t")
assert (
df_gen == df_mat).all().all() # all on DataFrame and then on Series
else:
out_folder_jams_train = None
out_train_tsv = osp.join(base_folder, 'metadata', 'train', 'soundscapes',
'synthetic.tsv')
list_jams_train = glob.glob(osp.join(train_folder, "*.jams"))
fg_path_train = osp.join(base_folder, "audio", "train", "soundbank",
"foreground")
bg_path_train = osp.join(base_folder, 'audio', "train", "soundbank",
"background")
generate_files_from_jams(list_jams_train,
out_train_folder,
fg_path=fg_path_train,
bg_path=bg_path_train,
out_folder_jams=out_folder_jams_train)
generate_tsv_from_jams(list_jams_train, out_train_tsv)
# ########
# Eval
# ########
# In the evaluation part, there multiple subsets which allows to check robustness of systems
eval_folder = osp.join(base_folder, 'metadata', 'eval', "soundscapes")
list_folders = [
osp.join(eval_folder, dI) for dI in os.listdir(eval_folder)
if osp.isdir(osp.join(eval_folder, dI))
]
fg_path_eval = osp.join(base_folder, "audio", "eval", "soundbank",
"foreground")
bg_path_eval = osp.join(base_folder, "audio", "eval", "soundbank",
"background")
soundbank_dir = args.soundbank
fg_path_train = osp.join(soundbank_dir, "foreground")
bg_path_train = osp.join(soundbank_dir, "background")
out_audio_dir = args.out_audio_dir
if out_audio_dir is None:
out_audio_dir = jams_folder
if args.save_jams:
out_folder_jams = out_audio_dir
else:
out_folder_jams = None
out_tsv = args.out_tsv
if out_tsv is None:
out_tsv = out_audio_dir.replace("audio", "metadata") + ".tsv"
save_isolated = args.save_isolated
# ########
# Generate
# #######
generate_files_from_jams(list_jams,
out_audio_dir,
out_folder_jams=out_folder_jams,
fg_path=fg_path_train,
bg_path=bg_path_train,
save_isolated_events=save_isolated)
if out_tsv:
generate_tsv_from_jams(list_jams, out_tsv)