def _wad_do_detection_eval(args, json_dataset, res_file, output_dir): coco_dt = json_dataset.WAD_CVPR2018.loadRes(str(res_file)) coco_gt = json_dataset.WAD_CVPR2018.loadGt(args.range, 'boxes') coco_eval = WAD_eval(args, coco_gt, coco_dt, 'bbox') coco_eval.evaluate() coco_eval.accumulate() _log_detection_eval_metrics(json_dataset, coco_eval) eval_file = os.path.join(output_dir, 'detection_results.pkl') save_object(coco_eval, eval_file) logger.info('Wrote json eval results to: {}'.format(eval_file)) return coco_eval
def _apolloscape_do_segmentation_eval(args, json_dataset, res_file, output_dir): coco_dt = json_dataset.ApolloScape.loadRes(str(res_file)) coco_gt = json_dataset.ApolloScape.loadGt(json_dataset.roidb, args.range, 'segms') coco_eval = WAD_eval(args, coco_gt, coco_dt, 'segm') coco_eval.evaluate() coco_eval.accumulate() _log_detection_eval_metrics(json_dataset, coco_eval) eval_file = os.path.join(output_dir, 'segmentation_results.pkl') save_object(coco_eval, eval_file) logger.info('Wrote json eval results to: {}'.format(eval_file)) return coco_eval