| 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() |