| 514 | self.useCats = 0 |
| 515 | |
| 516 | def setKpParams(self): |
| 517 | self.imgIds = [] |
| 518 | self.catIds = [] |
| 519 | # np.arange causes trouble. the data point on arange is slightly larger than the true value |
| 520 | self.iouThrs = np.linspace(.5, 0.95, int(np.round((0.95 - .5) / .05)) + 1, endpoint=True) |
| 521 | self.recThrs = np.linspace(.0, 1.00, int(np.round((1.00 - .0) / .01)) + 1, endpoint=True) |
| 522 | self.maxDets = [20] |
| 523 | self.areaRng = [[0 ** 2, 1e5 ** 2], [32 ** 2, 96 ** 2], [96 ** 2, 1e5 ** 2]] |
| 524 | self.areaRngLbl = ['all', 'medium', 'large'] |
| 525 | self.useCats = 1 |
| 526 | self.kpt_oks_sigmas = np.array([.26, .25, .25, .35, .35, .79, .79, .72, .72, .62,.62, 1.07, 1.07, .87, .87, .89, .89])/10.0 |
| 527 | |
| 528 | def __init__(self, iouType='segm'): |
| 529 | if iouType == 'segm' or iouType == 'bbox': |