###################################################################################### ###################打印IOU为0.5时的每类bbox大于minsize的Precision&&Recall############### import json import numpy as np from tqdm import tqdm from collections import defaultdict from copy import deepcopy class Dt: def __init__(self, img_id, score, bbox, cat): """ single dt :param img_id: :param score: a list of score :param bbox: bbox """ self.img_id = img_id self.score = score self.bbox = bbox self.cat = cat self.is_match = False class Gt: def __init__(self, img_id, bbox, cat): """ :param img_id: :param bbox: """ self.img_id = img_id self.bbox = bbox self.cat = cat self.is_match = False def compute_iou(bbox1, bbox2): x1, y1, w1, h1 = bbox1 x2, y2, w2, h2 = bbox2 s_sum = w1 * h1 + w2 * h2 left = max(x1, x2) right = min(x1 + w1, x2 + w2) top = max(y1, y2) bottom = min(y1 + h1, y2 + h2) if left >= right or top >= bottom: return 0 intersect = (right
目标检测中评估每类目标在IOU为0.5的Precision&&Recall
于 2022-04-21 18:45:52 首次发布