Spaces:
Running
Running
File size: 690 Bytes
a80d6bb c74a070 a80d6bb c74a070 a80d6bb c74a070 a80d6bb c74a070 a80d6bb c74a070 a80d6bb c74a070 a80d6bb c74a070 |
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 28 29 30 |
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()
|