From 785e70d269a554b2ba29daf137354103221f479e Mon Sep 17 00:00:00 2001 From: canwang Date: Mon, 3 Jun 2019 14:12:56 +0800 Subject: [PATCH] fix a bug in AP --- crowdpose-api/PythonAPI/crowdposetools/cocoeval.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/crowdpose-api/PythonAPI/crowdposetools/cocoeval.py b/crowdpose-api/PythonAPI/crowdposetools/cocoeval.py index 239e9b5..8586a4f 100644 --- a/crowdpose-api/PythonAPI/crowdposetools/cocoeval.py +++ b/crowdpose-api/PythonAPI/crowdposetools/cocoeval.py @@ -539,11 +539,11 @@ def _summarizeKps(): stats[6] = _summarize(0, maxDets=20, iouThr=.5) stats[7] = _summarize(0, maxDets=20, iouThr=.75) type_result = self.get_type_result(first=0.2, second=0.8) - stats[8] = _summarize(-1, maxDets=20, + stats[8] = _summarize(1, maxDets=20, areaRng='easy', type_score=type_result[0]) - stats[9] = _summarize(-1, maxDets=20, + stats[9] = _summarize(1, maxDets=20, areaRng='medium', type_score=type_result[1]) - stats[10]=_summarize(-1, maxDets=20, + stats[10]=_summarize(1, maxDets=20, areaRng='hard', type_score=type_result[2]) return stats if not self.eval: