| 7 | |
| 8 | |
| 9 | def parse_args(): |
| 10 | parser = ArgumentParser() |
| 11 | parser.add_argument('img', help='Image file') |
| 12 | parser.add_argument('config', help='Config file') |
| 13 | parser.add_argument('checkpoint', help='Checkpoint file') |
| 14 | parser.add_argument('--out-file', default=None, help='Path to output file') |
| 15 | parser.add_argument( |
| 16 | '--device', default='cuda:0', help='Device used for inference') |
| 17 | parser.add_argument( |
| 18 | '--palette', |
| 19 | default='dota', |
| 20 | choices=['dota', 'sar', 'hrsc', 'hrsc_classwise', 'random'], |
| 21 | help='Color palette used for visualization') |
| 22 | parser.add_argument( |
| 23 | '--score-thr', type=float, default=0.3, help='bbox score threshold') |
| 24 | args = parser.parse_args() |
| 25 | return args |
| 26 | |
| 27 | |
| 28 | def main(args): |