()
| 14 | |
| 15 | |
| 16 | def get_args_parser(): |
| 17 | parser = argparse.ArgumentParser() |
| 18 | |
| 19 | parser.add_argument( |
| 20 | "--output_dir", |
| 21 | type=str, |
| 22 | default="", |
| 23 | help="value for outdir", |
| 24 | ) |
| 25 | parser.add_argument( |
| 26 | "--eval_dataset", type=str, default="nyu", choices=list(dataset_metadata.keys()) |
| 27 | ) |
| 28 | parser.add_argument( |
| 29 | "--align", |
| 30 | type=str, |
| 31 | default="scale&shift", |
| 32 | choices=["scale&shift", "scale", "metric"], |
| 33 | ) |
| 34 | return parser |
| 35 | |
| 36 | |
| 37 | def main(args): |