()
| 9 | |
| 10 | |
| 11 | def parse_args(): |
| 12 | parser = argparse.ArgumentParser( |
| 13 | description='Convert DRIVE dataset to mmsegmentation format') |
| 14 | parser.add_argument( |
| 15 | 'training_path', help='the training part of DRIVE dataset') |
| 16 | parser.add_argument( |
| 17 | 'testing_path', help='the testing part of DRIVE dataset') |
| 18 | parser.add_argument('--tmp_dir', help='path of the temporary directory') |
| 19 | parser.add_argument('-o', '--out_dir', help='output path') |
| 20 | args = parser.parse_args() |
| 21 | return args |
| 22 | |
| 23 | |
| 24 | def main(): |