refactor: Rename metrics in SemanticSegmentationTask

microsoft · Jun 19, 2024 · b1526fa · b1526fa
1 parent b7d8305
commit b1526fa
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/torchgeo/trainers/segmentation.py b/torchgeo/trainers/segmentation.py
@@ -221,7 +221,7 @@ def configure_metrics(self) -> None:
                     average='micro',
                     multidim_average='global',
                 ),
-                'OverallIoU': JaccardIndex(
+                'OverallJaccardIndex': JaccardIndex(
                     task='multiclass',
                     num_classes=num_classes,
                     ignore_index=ignore_index,
@@ -240,7 +240,7 @@ def configure_metrics(self) -> None:
                     average='macro',
                     multidim_average='global',
                 ),
-                'AverageIoU': JaccardIndex(
+                'AverageJaccardIndex': JaccardIndex(
                     task='multiclass',
                     num_classes=num_classes,
                     ignore_index=ignore_index,
@@ -283,7 +283,7 @@ def configure_metrics(self) -> None:
                     ),
                     labels=labels,
                 ),
-                'IoU': ClasswiseWrapper(
+                'JaccardIndex': ClasswiseWrapper(
                     JaccardIndex(
                         task='multiclass', num_classes=num_classes, average='none'
                     ),