Example #1
0
 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
Example #2
0
 def __init__(self, cocoGt=None, cocoDt=None, iouType='segm'):
     COCOEval.__init__(self, cocoGt, cocoDt, iouType)
     self.summary = [['pose', 'pose_heatval', 'oks']]