| 22 | |
| 23 | |
| 24 | def default_argument_parser(): |
| 25 | parser = argparse.ArgumentParser( |
| 26 | description="PyTorch OTS Testing" |
| 27 | ) |
| 28 | parser.add_argument( |
| 29 | "--imgs", |
| 30 | default='/home/xxx/Documents/data/places7/val/', |
| 31 | type=str, |
| 32 | help="dataset" |
| 33 | ) |
| 34 | parser.add_argument( |
| 35 | "--cfg", |
| 36 | default="config/resnet.yaml", |
| 37 | metavar="FILE", |
| 38 | help="path to config file", |
| 39 | type=str, |
| 40 | ) |
| 41 | parser.add_argument( |
| 42 | "--cls-file", |
| 43 | default="data/places365_7.txt", |
| 44 | metavar="FILE", |
| 45 | help="path to class file", |
| 46 | type=str, |
| 47 | ) |
| 48 | parser.add_argument( |
| 49 | "--ckpt", |
| 50 | default="ckpt/resnet50_7.pth.tar", |
| 51 | metavar="FILE", |
| 52 | help="path to checkpoint", |
| 53 | type=str, |
| 54 | ) |
| 55 | parser.add_argument( |
| 56 | "opts", |
| 57 | help="Modify config options using the command-line", |
| 58 | default=None, |
| 59 | nargs=argparse.REMAINDER, |
| 60 | ) |
| 61 | |
| 62 | parser.add_argument( |
| 63 | "--local-object", |
| 64 | default="", |
| 65 | help="path to local object features", |
| 66 | type=str, |
| 67 | ) |
| 68 | |
| 69 | return parser |
| 70 | |
| 71 | |
| 72 | def load_dict(filename): |