third_party/SGMNet/datadump/dump.py

import argparse
import yaml


def str2bool(v):
    return v.lower() in ("true", "1")


# Parse command line arguments.
parser = argparse.ArgumentParser(description="dump eval data.")
parser.add_argument("--config_path", type=str, default="configs/yfcc.yaml")


def get_dumper(name):
    mod = __import__("dumper.{}".format(name), fromlist=[""])
    return getattr(mod, name)


if __name__ == "__main__":
    args = parser.parse_args()
    with open(args.config_path, "r") as f:
        config = yaml.load(f)

    dataset = get_dumper(config["data_name"])(config)

    dataset.initialize()
    if config["extractor"]["extract"]:
        dataset.dump_feature()
    dataset.format_dump_data()