浏览代码

refine confusion matrix computation

will-jl944 4 年之前
父节点
当前提交
71ed1d640e
共有 1 个文件被更改,包括 1 次插入1 次删除
  1. 1 1
      dygraph/paddlex/cv/models/segmenter.py

+ 1 - 1
dygraph/paddlex/cv/models/segmenter.py

@@ -390,7 +390,7 @@ class BaseSegmenter(BaseModel):
             ], [miou, class_iou, oacc, class_acc, kappa, category_f1score]))
 
         if return_details:
-            conf_mat = sum(conf_mat_all).numpy()
+            conf_mat = sum(conf_mat_all)
             eval_details = {'confusion_matrix': conf_mat.tolist()}
             return eval_metrics, eval_details
         return eval_metrics