瀏覽代碼

refine seg evaluation

will-jl944 4 年之前
父節點
當前提交
8ab8cbc020
共有 1 個文件被更改,包括 4 次插入6 次删除
  1. 4 6
      dygraph/paddlex/cv/models/segmenter.py

+ 4 - 6
dygraph/paddlex/cv/models/segmenter.py

@@ -364,12 +364,10 @@ class BaseSegmenter(BaseModel):
                         label_area_list = label_area_list[:valid]
                         conf_mat_list = conf_mat_list[:valid]
 
-                    for i in range(len(intersect_area_list)):
-                        intersect_area_all = intersect_area_all + intersect_area_list[
-                            i]
-                        pred_area_all = pred_area_all + pred_area_list[i]
-                        label_area_all = label_area_all + label_area_list[i]
-                    conf_mat_all = conf_mat_all + conf_mat_list
+                    intersect_area_all += sum(intersect_area_list)
+                    pred_area_all += sum(pred_area_list)
+                    label_area_all += sum(label_area_list)
+                    conf_mat_all.extend(conf_mat_list)
 
                 else:
                     intersect_area_all = intersect_area_all + intersect_area