[Fix] Fix the overflow error in calculating kappa (#1788)

PaddlePaddle · Mar 2, 2022 · f665b18 · f665b18
1 parent e88279d
commit f665b18
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/paddleseg/utils/metrics.py b/paddleseg/utils/metrics.py
@@ -200,9 +200,9 @@ def kappa(intersect_area, pred_area, label_area):
     Returns:
         float: kappa coefficient.
     """
-    intersect_area = intersect_area.numpy()
-    pred_area = pred_area.numpy()
-    label_area = label_area.numpy()
+    intersect_area = intersect_area.numpy().astype(np.float64)
+    pred_area = pred_area.numpy().astype(np.float64)
+    label_area = label_area.numpy().astype(np.float64)
     total_area = np.sum(label_area)
     po = np.sum(intersect_area) / total_area
     pe = np.sum(pred_area * label_area) / (total_area * total_area)