File size: 685 Bytes
404d2af |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 |
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() |