diff --git a/PythonAPI/pycocotools/cocoeval.py b/PythonAPI/pycocotools/cocoeval.py index f4e3decb..fd91f347 100644 --- a/PythonAPI/pycocotools/cocoeval.py +++ b/PythonAPI/pycocotools/cocoeval.py @@ -419,7 +419,7 @@ def accumulate(self, p = None): toc = time.time() print('DONE (t={:0.2f}s).'.format( toc-tic)) - def summarize(self): + def summarize(self, main_iou): ''' Compute and display summary metrics for evaluation results. Note this functin can *only* be applied on the default parameter setting @@ -458,7 +458,7 @@ def _summarize( ap=1, iouThr=None, areaRng='all', maxDets=100 ): def _summarizeDets(): stats = np.zeros((12,)) stats[0] = _summarize(1) - stats[1] = _summarize(1, iouThr=.5, maxDets=self.params.maxDets[2]) + stats[1] = _summarize(1, iouThr=main_iou, maxDets=self.params.maxDets[2]) stats[2] = _summarize(1, iouThr=.75, maxDets=self.params.maxDets[2]) stats[3] = _summarize(1, areaRng='small', maxDets=self.params.maxDets[2]) stats[4] = _summarize(1, areaRng='medium', maxDets=self.params.maxDets[2])