| 11 | |
| 12 | |
| 13 | def parse_args(): |
| 14 | parser = argparse.ArgumentParser( |
| 15 | description='Convert HRF dataset to mmsegmentation format') |
| 16 | parser.add_argument('healthy_path', help='the path of healthy.zip') |
| 17 | parser.add_argument( |
| 18 | 'healthy_manualsegm_path', help='the path of healthy_manualsegm.zip') |
| 19 | parser.add_argument('glaucoma_path', help='the path of glaucoma.zip') |
| 20 | parser.add_argument( |
| 21 | 'glaucoma_manualsegm_path', help='the path of glaucoma_manualsegm.zip') |
| 22 | parser.add_argument( |
| 23 | 'diabetic_retinopathy_path', |
| 24 | help='the path of diabetic_retinopathy.zip') |
| 25 | parser.add_argument( |
| 26 | 'diabetic_retinopathy_manualsegm_path', |
| 27 | help='the path of diabetic_retinopathy_manualsegm.zip') |
| 28 | parser.add_argument('--tmp_dir', help='path of the temporary directory') |
| 29 | parser.add_argument('-o', '--out_dir', help='output path') |
| 30 | args = parser.parse_args() |
| 31 | return args |
| 32 | |
| 33 | |
| 34 | def main(): |