|
import argparse |
|
import yaml |
|
|
|
|
|
def str2bool(v): |
|
return v.lower() in ("true", "1") |
|
|
|
|
|
|
|
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() |
|
|
