Add request info to return config label indexes (#1524)

* Add request info to return config label indexes

Signed-off-by: Andres <[email protected]>

* [pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci

* Update segmentation.py

Change argument name

Signed-off-by: Andres Diaz-Pinto <[email protected]>

* [pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci

* Update deepedit.py

Change argument name

Signed-off-by: Andres Diaz-Pinto <[email protected]>

* [pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci

---------

Signed-off-by: Andres <[email protected]>
Signed-off-by: Andres Diaz-Pinto <[email protected]>
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>

Author: diazandr3s

sample-apps/radiology/lib/infers/deepedit.py

@@ -119,5 +119,9 @@ def post_transforms(self, data=None) -> Sequence[Callable]:
             AsDiscreted(keys="pred", argmax=True),
             SqueezeDimd(keys="pred", dim=0),
             ToNumpyd(keys="pred"),
-            Restored(keys="pred", ref_image="image"),
+            Restored(
+                keys="pred",
+                ref_image="image",
+                config_labels=self.labels if data.get("restore_label_idx", False) else None,
+            ),
         ]

sample-apps/radiology/lib/infers/segmentation.py

@@ -92,5 +92,11 @@ def post_transforms(self, data=None) -> Sequence[Callable]:
 
         if data and data.get("largest_cc", False):
             t.append(KeepLargestConnectedComponentd(keys="pred"))
-        t.append(Restored(keys="pred", ref_image="image"))
+        t.append(
+            Restored(
+                keys="pred",
+                ref_image="image",
+                config_labels=self.labels if data.get("restore_label_idx", False) else None,
+            )
+        )
         return t