Explorar o código

refine confusion matrix computation

will-jl944 %!s(int64=4) %!d(string=hai) anos
pai
achega
71ed1d640e
Modificáronse 1 ficheiros con 1 adicións e 1 borrados
  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