()
| 9 | |
| 10 | |
| 11 | def parse_args(): |
| 12 | parser = argparse.ArgumentParser( |
| 13 | description='MMDet3D test (and eval) a model') |
| 14 | parser.add_argument('results_file', help='the results pkl file') |
| 15 | parser.add_argument('ann_file', help='annoations json file') |
| 16 | |
| 17 | parser.add_argument('--iou_thr', |
| 18 | type=list, |
| 19 | default=[0.25, 0.5], |
| 20 | help='the IoU threshold during evaluation') |
| 21 | |
| 22 | args = parser.parse_args() |
| 23 | return args |
| 24 | |
| 25 | |
| 26 | def ground_eval(gt_annos, det_annos, iou_thr): |