generate_dcase_task4_2022.py

import argparse
import glob
import os
import shutil
import time
import warnings
from pathlib import Path
from pprint import pformat

import desed


def create_folder(folder, exist_ok=True, delete_if_exists=False):
    """Create folder (and parent folders) if not exists.

    Args:
        folder: str, path of folder(s) to create.
        delete_if_exists: bool, True if you want to delete the folder when exists

    Returns:
        None
    """
    if not folder == "":
        if delete_if_exists:
            if os.path.exists(folder):
                shutil.rmtree(folder)
                os.mkdir(folder)

        os.makedirs(folder, exist_ok=exist_ok)


def _create_symlink(src, dest, **kwargs):
    if os.path.exists(dest):
        warnings.warn(f"Symlink already exists : {dest}, skipping.\n")
    else:
        os.makedirs(os.path.dirname(dest), exist_ok=True)
        os.symlink(os.path.abspath(src), dest, **kwargs)


def create_synth_dcase(synth_path, destination_folder):
    """Create symbolic links for synethtic part of the dataset

    Args:
        synth_path (str): synthetic folder path
        destination_folder (str): destination folder path
    """
    print("Creating symlinks for synthetic data")
    split_sets = ["train", "validation"]
    if os.path.exists(os.path.join(synth_path, "audio", "eval")):
        split_sets.append("eval")

    for split_set in split_sets:
        # AUDIO
        split_audio_folder = os.path.join(synth_path, "audio", split_set)
        audio_subfolders = [
            d
            for d in os.listdir(split_audio_folder)
            if os.path.isdir(os.path.join(split_audio_folder, d))
        ]
        # Manage the validation case which changed from 2020
        if split_set == "validation" and not len(audio_subfolders):
            split_audio_folder = os.path.join(synth_path, "audio")
            audio_subfolders = ["validation"]

        for subfolder in audio_subfolders:
            abs_src_folder = os.path.abspath(
                os.path.join(split_audio_folder, subfolder)
            )
            dest_folder = os.path.join(
                destination_folder, "audio", split_set, subfolder
            )
            _create_symlink(abs_src_folder, dest_folder)

        # META
        split_meta_folder = os.path.join(
            synth_path, "metadata", split_set, f"synthetic21_{split_set}"
        )
        meta_files = glob.glob(os.path.join(split_meta_folder, "*.tsv"))
        for meta_file in meta_files:
            create_folder(destination_folder)
            dest_file = os.path.join(
                destination_folder,
                "metadata",
                split_set,
                f"synthetic21_{split_set}",
                os.path.basename(meta_file),
            )
            _create_symlink(meta_file, dest_file)


if __name__ == "__main__":
    t = time.time()
    parser = argparse.ArgumentParser()
    parser.add_argument(
        "--basedir",
        type=str,
        default="../../data",
        help="The base data folder in which we'll create the different datasets."
        "Useful when you don't have any dataset, provide this one and the output folder",
    )
    parser.add_argument(
        "--out_dir",
        type=str,
        default=None,
        help="Output basefolder in which to put the created 2021 dataset (with real and soundscapes)",
    )
    parser.add_argument(
        "--only_real",
        action="store_true",
        help="True if only the real part of the dataset need to be downloaded",
    )

    parser.add_argument(
        "--only_synth",
        action="store_true",
        help="True if only the synthetic part of the dataset need to be downloaded",
    )

    parser.add_argument(
        "--only_strong",
        action="store_true",
        help="True if only the strongly annotated part of the Audioset dataset need to be downloaded",
    )

    args = parser.parse_args()
    pformat(vars(args))

    # #########
    # Paths
    # #########
    bdir = args.basedir
    dcase_dataset_folder = args.out_dir
    only_real = args.only_real
    only_synth = args.only_synth
    only_strong = args.only_strong
    missing_files = None

    download_all = (only_real and only_synth and only_strong) or (
        not only_real and not only_synth and not only_strong
    )
    print(f"Download all: {download_all}")

    # Default paths if not defined (using basedir)
    if dcase_dataset_folder is None:
        dcase_dataset_folder = os.path.join(bdir, "dcase", "dataset")

    # #########
    # Download the different datasets if they do not exist
    # #########

    # download real dataset
    if only_real or download_all:
        print("Downloading audioset dataset")
        missing_files = desed.download_audioset_data(
            dcase_dataset_folder, n_jobs=3, chunk_size=10
        )

    # download strong-label Audioset dataset
    if only_strong or download_all:
        url_strong = (
            "https://zenodo.org/record/6444477/files/audioset_strong.tsv?download=1"
        )
        basedir_missing_files = "missing_files"
        desed.utils.create_folder(basedir_missing_files)

        strong_label_metadata_path = os.path.join(
            dcase_dataset_folder, "metadata", "train", "audioset_strong.tsv"
        )
        sl_path = Path(strong_label_metadata_path)
        if not sl_path.is_file():
            desed.utils.download_file_from_url(url_strong, strong_label_metadata_path)
            print(f"File saved in {strong_label_metadata_path}")

            print("Downloading strong-label Audioset dataset...")
            path_missing_files_audioset = os.path.join(
                basedir_missing_files, "missing_files_" + "strong_label_real" + ".tsv"
            )
            desed.download.download_audioset_files_from_csv(
                strong_label_metadata_path,
                os.path.join(
                    dcase_dataset_folder, "audio", "train", "strong_label_real"
                ),
                missing_files_tsv=path_missing_files_audioset,
            )

        else:
            print(f"The file {sl_path} already exists.")

    # download synthetic dataset
    if only_synth or download_all:
        print(f"Downloading synthetic part of the dataset")
        url_synth = "https://zenodo.org/record/6026841/files/dcase_synth.zip?download=1"
        synth_folder = str(os.path.basename(url_synth)).split(".")[0]
        desed.download.download_and_unpack_archive(
            url_synth, dcase_dataset_folder, archive_format="zip"
        )
        synth_folder = os.path.join(bdir, "dcase", "dataset", synth_folder)
        create_synth_dcase(synth_folder, dcase_dataset_folder)

    print(f"Time of the program: {time.time() - t} s")
    print(
        f"The dcase dataset has been saved in the following path: {dcase_dataset_folder}"
    )
    if missing_files is not None:
        warnings.warn(
            f"You have missing files.\n\n"
            f"Please try to redownload desed_real again: \n"
            f"import desed\n"
            f"desed.download_audioset_data('{dcase_dataset_folder}', n_jobs=3, chunk_size=10)\n\n"
            f"Please, send your missing_files_xx.tsv to the task organisers to get your missing files.\n"
        )