Browse Source

refine confusion matrix computation

will-jl944 4 years ago
parent
commit
71ed1d640e
1 changed files with 1 additions and 1 deletions
  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