Parse arguments.
()
| 107 | |
| 108 | |
| 109 | def parse_args(): |
| 110 | """Parse arguments.""" |
| 111 | parser = argparse.ArgumentParser(description='Splitting images') |
| 112 | add_parser(parser) |
| 113 | args = parser.parse_args() |
| 114 | |
| 115 | if args.base_json is not None: |
| 116 | with open(args.base_json, 'r') as f: |
| 117 | prior_config = json.load(f) |
| 118 | |
| 119 | for action in parser._actions: |
| 120 | if action.dest not in prior_config or \ |
| 121 | not hasattr(action, 'default'): |
| 122 | continue |
| 123 | action.default = prior_config[action.dest] |
| 124 | args = parser.parse_args() |
| 125 | |
| 126 | # assert arguments |
| 127 | print(args) |
| 128 | assert args.img_ann_dirs is not None, "argument img_ann_dirs can't be None" |
| 129 | assert len(args.sizes) == len(args.gaps) |
| 130 | assert len(args.sizes) == 1 or len(args.rates) == 1 |
| 131 | assert args.save_ext in ['.png', '.jpg', 'bmp', '.tif'] |
| 132 | assert args.iof_thr >= 0 and args.iof_thr < 1 |
| 133 | assert args.iof_thr >= 0 and args.iof_thr <= 1 |
| 134 | assert not osp.exists(args.save_dir), \ |
| 135 | f'{osp.join(args.save_dir)} already exists' |
| 136 | return args |
| 137 | |
| 138 | |
| 139 | def get_sliding_window(info, sizes, gaps, img_rate_thr): |