()
| 101 | |
| 102 | |
| 103 | def make_parser(): |
| 104 | parser = argparse.ArgumentParser(description="COCO Tfrecord test") |
| 105 | parser.add_argument( |
| 106 | "-i", |
| 107 | "--iters", |
| 108 | default=None, |
| 109 | type=int, |
| 110 | metavar="N", |
| 111 | help="number of iterations to run (default: whole dataset)", |
| 112 | ) |
| 113 | parser.add_argument( |
| 114 | "-w", |
| 115 | "--num_workers", |
| 116 | default=4, |
| 117 | type=int, |
| 118 | metavar="N", |
| 119 | help="number of worker threads (default: %(default)s)", |
| 120 | ) |
| 121 | |
| 122 | return parser |
| 123 | |
| 124 | |
| 125 | if __name__ == "__main__": |