| 17 | torch.backends.cudnn.benchmark = False |
| 18 | |
| 19 | def parse_args(): |
| 20 | parser = argparse.ArgumentParser(description="Test CenterNet") |
| 21 | parser.add_argument("cfg_file", help="config file", type=str) |
| 22 | parser.add_argument("--testiter", dest="testiter", |
| 23 | help="test at iteration i", |
| 24 | default=None, type=int) |
| 25 | parser.add_argument("--split", dest="split", |
| 26 | help="which split to use", |
| 27 | default="validation", type=str) |
| 28 | parser.add_argument("--suffix", dest="suffix", default=None, type=str) |
| 29 | parser.add_argument("--debug", action="store_true") |
| 30 | |
| 31 | args = parser.parse_args() |
| 32 | return args |
| 33 | |
| 34 | def make_dirs(directories): |
| 35 | for directory in directories: |