()
| 2 | from pathlib import Path |
| 3 | |
| 4 | def get_base_parser(): |
| 5 | parser = ArgumentParser() |
| 6 | parser.add_argument('-out_path', type=Path, default='./output', help='output folder, default is \'./output\'') |
| 7 | parser.add_argument('-image_format', type=str, default='png', choices=['png', 'jpg'], help='image format for saving outputs') |
| 8 | return parser |
| 9 | |
| 10 | def add_network_args(parser: ArgumentParser): |
| 11 | parser.add_argument('-size', type=str, required=True, choices=['small', 'large', 'defocus_deblur']) |
no outgoing calls
no test coverage detected