| 23 | torch.backends.cudnn.benchmark = True |
| 24 | |
| 25 | def parse_args(): |
| 26 | parser = argparse.ArgumentParser(description="Train CenterNet") |
| 27 | parser.add_argument("cfg_file", help="config file", type=str) |
| 28 | parser.add_argument("--iter", dest="start_iter", |
| 29 | help="train at iteration i", |
| 30 | default=0, type=int) |
| 31 | parser.add_argument("--threads", dest="threads", default=4, type=int) |
| 32 | |
| 33 | #args = parser.parse_args() |
| 34 | args, unparsed = parser.parse_known_args() |
| 35 | return args |
| 36 | |
| 37 | def prefetch_data(db, queue, sample_data, data_aug): |
| 38 | ind = 0 |