Reduce saved information to reduce memory usage

2026-09-24 23:33:08 +08:00 · 2021-07-19 17:11:29 +08:00
parent f7a62d945e
commit bd92c22b61
1 changed files with 16 additions and 11 deletions
@@ -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):