소스 검색

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