def __init__(self, cocoGt=None, cocoDt=None, iouType='segm', l_thrs=0.5, h_thrs=0.95): assert l_thrs < h_thrs and l_thrs > 0.0 and h_thrs < 1.0 COCOeval.__init__(self, cocoGt=cocoGt, cocoDt=cocoDt, iouType=iouType) self.params.iouThrs = np.linspace(l_thrs, h_thrs, np.round( (h_thrs - l_thrs) / .05) + 1, endpoint=True) # self.params.recThrs = np.linspace(.0, 1.00, np.round((1.00 - .0) / .01) + 1, endpoint=True) self.l_thrs, self.h_thrs = l_thrs, h_thrs
def __init__(self, cocoGt=None, cocoDt=None, iouType='segm'): COCOEval.__init__(self, cocoGt, cocoDt, iouType) self.summary = [['pose', 'pose_heatval', 'oks']]