cvtools.evaluation.f1_score module

cvtools.evaluation.f1_score.f1_score(det_results, gt_bboxes, gt_labels, gt_ignore=None, scale_ranges=None, iou_thr=0.5, dataset=None, print_summary=True)

Evaluate F1 score of a dataset.

参数:

• det_results (list) -- a list of list, [[cls1_det, cls2_det, ...], ...]
• gt_bboxes (list) -- ground truth bboxes of each image, a list of K*4 array.
• gt_labels (list) -- ground truth labels of each image, a list of K array
• gt_ignore (list) -- gt ignore indicators of each image, a list of K array
• scale_ranges (list, optional) -- [(min1, max1), (min2, max2), ...]
• iou_thr (float) -- IoU threshold
• dataset (None or str or list) -- dataset name or dataset classes, there are minor differences in metrics for different datsets, e.g. "voc07", "imagenet_det", etc.
• print_summary (bool) -- whether to print the mAP summary

返回:

(mAP, [dict, dict, ...])

返回类型:

tuple

cvtools.evaluation.f1_score.get_coco_anns(file, imgs=None)