()
| 261 | |
| 262 | |
| 263 | def parse_args(): |
| 264 | parser = argparse.ArgumentParser( |
| 265 | description='MMDet3D test (and eval) a model') |
| 266 | parser.add_argument('results_file', help='the results pkl file') |
| 267 | parser.add_argument('ann_file', help='annoations json file') |
| 268 | |
| 269 | parser.add_argument('--iou_thr', |
| 270 | type=list, |
| 271 | default=[0.25, 0.5], |
| 272 | help='the IoU threshold during evaluation') |
| 273 | |
| 274 | args = parser.parse_args() |
| 275 | return args |
| 276 | |
| 277 | |
| 278 | def ground_eval(gt_annos, det_annos, iou_thr): |