Reduce saved information to reduce memory usage
@@ -168,22 +168,27 @@ class DetectionIoUEvaluator(object):
numGlobalCareGt += numGtCare
numGlobalCareDet += numDetCare
+ # perSampleMetrics = {
+ # 'precision': precision,
+ # 'recall': recall,
+ # 'hmean': hmean,
+ # 'pairs': pairs,
+ # 'iouMat': [] if len(detPols) > 100 else iouMat.tolist(),
+ # 'gtPolPoints': gtPolPoints,
+ # 'detPolPoints': detPolPoints,
+ # 'gtCare': numGtCare,
+ # 'detCare': numDetCare,
+ # 'gtDontCare': gtDontCarePolsNum,
+ # 'detDontCare': detDontCarePolsNum,
+ # 'detMatched': detMatched,
+ # 'evaluationLog': evaluationLog
+ # }
+
perSampleMetrics = {
- 'precision': precision,
- 'recall': recall,
- 'hmean': hmean,
- 'pairs': pairs,
- 'iouMat': [] if len(detPols) > 100 else iouMat.tolist(),
- 'gtPolPoints': gtPolPoints,
- 'detPolPoints': detPolPoints,
'gtCare': numGtCare,
'detCare': numDetCare,
- 'gtDontCare': gtDontCarePolsNum,
- 'detDontCare': detDontCarePolsNum,
'detMatched': detMatched,
- 'evaluationLog': evaluationLog
}
-
return perSampleMetrics
def combine_results(self, results):
