compute_coco_eval#

compute_coco_eval(outputs, targets, num_class_ids)[source]#

Return mAP averaged over 0.5-0.95 using pycocotools eval.

Note: boxes are in (ymin, xmin, ymax, xmax) format with values ranging

from 0 to h or w.

Parameters
  • outputs

    (list) of length m containing dicts of form {‘boxes’: <tensor with shape (n, 4)>,

    ’class_ids’: <tensor with shape (n,)>, ‘scores’: <tensor with shape (n,)>}

  • targets

    (list) of length m containing dicts of form {‘boxes’: <tensor with shape (n, 4)>,

    ’class_ids’: <tensor with shape (n,)>}