Fix icevision task metric names

CHANGELOG.md

@@ -44,6 +44,8 @@ The format is based on [Keep a Changelog](http://keepachangelog.com/en/1.0.0/).
 
 - Fixed a bug where `InstanceSegmentation` would fail if samples had an inconsistent number of bboxes, labels, and masks (these will now be treated as negative samples) ([#1222](https://github.com/PyTorchLightning/lightning-flash/pull/1222))
 
+- Fixed a bug where `ObjectDetector`, `InstanceSegmentation`, and `KeypointDetector` would log train and validation metrics with the same name ([#1252](https://github.com/PyTorchLightning/lightning-flash/pull/1252))
+
 ## [0.7.0] - 2022-02-15
 
 ### Added

flash/core/data/utils.py

@@ -39,6 +39,7 @@
     RunningStage.VALIDATING: "val",
     RunningStage.PREDICTING: "predict",
     RunningStage.SERVING: "serve",
+    RunningStage.SANITY_CHECKING: "val",
 }
 _STAGES_PREFIX_VALUES = {"train", "test", "val", "predict", "serve"}
 

flash/core/integrations/icevision/adapter.py

@@ -42,8 +42,8 @@ class SimpleCOCOMetric(COCOMetric):
     def finalize(self) -> Dict[str, float]:
         logs = super().finalize()
         return {
-            "Precision (IoU=0.50:0.95,area=all)": logs["AP (IoU=0.50:0.95) area=all"],
-            "Recall (IoU=0.50:0.95,area=all,maxDets=100)": logs["AR (IoU=0.50:0.95) area=all maxDets=100"],
+            "precision (IoU=0.50:0.95,area=all)": logs["AP (IoU=0.50:0.95) area=all"],
+            "recall (IoU=0.50:0.95,area=all,maxDets=100)": logs["AR (IoU=0.50:0.95) area=all maxDets=100"],
         }
 
 

flash/core/integrations/icevision/wrappers.py

@@ -15,16 +15,19 @@
 
 import torch
 
+from flash.core.data.utils import _STAGES_PREFIX