| 13 | |
| 14 | |
| 15 | def para_args(): |
| 16 | parser = argparse.ArgumentParser(description="Network configurations!") |
| 17 | parser.add_argument("--model-path", default="data/results/default_model", metavar="FILE", |
| 18 | help="path to config file", type=str) |
| 19 | parser.add_argument("--dataset", default="PITT", type=str) |
| 20 | parser.add_argument("--epoch", default="62", type=int) |
| 21 | parser.add_argument("--noise", default=0, type=int, help="apply noise during test") |
| 22 | parser.add_argument("--type", default="rot", type=str, choices=["recall", "rot"], help="choose do top recall or rot analysis") |
| 23 | parser.add_argument("--trans-noise", type=int, required=True) |
| 24 | parser.add_argument("--rot-noise", type=int, required=True) |
| 25 | parser.add_argument("--log", type=bool, required=False, default=False) |
| 26 | args = parser.parse_args() |
| 27 | return args |
| 28 | |
| 29 | |
| 30 | def val(config, type, noise, trans_noise, rot_noise, log): |