| 9 | |
| 10 | |
| 11 | def parse_args(): |
| 12 | parser = argparse.ArgumentParser() |
| 13 | parser.add_argument('--base', type=str) |
| 14 | parser.add_argument('--name', type=str) |
| 15 | parser.add_argument('--dst', type=str) |
| 16 | parser.add_argument( |
| 17 | '--start', type=int, |
| 18 | help=('The start index to copy for each masked ratio directory. ' |
| 19 | 'Index range [args.start, args.start + args.num) will be used. ' |
| 20 | 'For different mask datasets, you should avoid overlapping range.') |
| 21 | ) |
| 22 | parser.add_argument('--num', type=int) |
| 23 | parser.add_argument('--min_area', type=int, default=5) |
| 24 | parser.add_argument('--max_area', type=int, default=60) |
| 25 | args = parser.parse_args() |
| 26 | assert args.base is not None |
| 27 | assert args.dst is not None |
| 28 | return args |
| 29 | |
| 30 | |
| 31 | if __name__ == '__main__': |