feat(preview): add ability to preview the effect on transforms on single images

src/nrtk_explorer/app/core.py

@@ -23,6 +23,12 @@
     config_features_to_enabled_features,
     config_preset_to_enabled_features,
 )
+from nrtk_explorer.app.images.image_ids import (
+    GROUND_TRUTH_MODEL,
+)
+from nrtk_explorer.app.images.stateful_annotations import (
+    make_stateful_annotations,
+)
 from nrtk_explorer.app.images.images import Images
 from nrtk_explorer.app.images.image_server import ImageServer
 from nrtk_explorer.app.applet import Applet
@@ -132,6 +138,8 @@ def __init__(self, server=None, **kwargs):
 
         images = Images(server=self.server)
         self._image_server = ImageServer(server=self.server, images=images)
+        ground_truth_annotations = make_stateful_annotations(self.server, GROUND_TRUTH_MODEL)
+        self.state.inference_models_obj = {0: {"name": "groundtruth"}}
 
         self._datasets_app = None
         if self.datasets_enabled:
@@ -150,15 +158,23 @@ def __init__(self, server=None, **kwargs):
             from nrtk_explorer.app.features.transforms import TransformsApp
 
             self._transforms_app = TransformsApp(
-                server=self.server.create_child_server(), images=images, **kwargs
+                server=self.server.create_child_server(),
+                images=images,
+                ground_truth_annotations=ground_truth_annotations,
+                **kwargs,
             )
 
+            self.context.tramsforms_app = self._transforms_app
+
         self._images_app = None
         if self.images_enabled:
             from nrtk_explorer.app.features.images import ImagesApp
 
             self._images_app = ImagesApp(
-                server=self.server.create_child_server(), images=images, **kwargs
+                server=self.server.create_child_server(),
+                images=images,
+                ground_truth_annotations=ground_truth_annotations,
+                **kwargs,
             )
 
         self._inference_app = None
@@ -207,6 +223,7 @@ def __init__(self, server=None, **kwargs):
         self.state.dataset_ids = []
         self.state.hovered_id = None
         self.state.maximised_id = None
+        self.state.maximised_annotations = []
 
         def clear_hovered(**kwargs):
             self.state.hovered_id = None

src/nrtk_explorer/app/features/images.py

@@ -83,7 +83,7 @@ def __init__(
 
         self.images = images or Images(server)
 
-        ground_truth_annotations = ground_truth_annotations or make_stateful_annotations(
+        self.ground_truth_annotations = ground_truth_annotations or make_stateful_annotations(
             server, GROUND_TRUTH_MODEL
         )
         self.context.ground_truth_annotations = ground_truth_annotations.annotations_factory

src/nrtk_explorer/app/features/inference.py

@@ -46,7 +46,7 @@ def __init__(
         self.state.inference_models = [self.state.inference_models_options[0]]
         self.state.inference_multi_model = False
 
-        inference_models_obj = {0: {"name": "ground-truth"}}
+        inference_models_obj = {0: {"name": "groundtruth"}}
         for i, model in enumerate(self.state.inference_models):
             inference_models_obj[i + 1] = {"name": model}
 
@@ -101,7 +101,7 @@ def update_inference_models(self, models):
                     "transformed_annotations": transformed_annotations.annotations_factory,
                 }
 
-        models_obj = {0: {"name": "ground-truth"}}
+        models_obj = {0: {"name": "groundtruth"}}
         for i, model in enumerate(models):
             models_obj[i + 1] = {"name": model}
 

src/nrtk_explorer/app/features/parameters.py

@@ -115,6 +115,15 @@ def transform_apply_ui(self):
                 flat=True,
             )
 
+    def transform_preview_ui(self):
+        with html.Div(trame_server=self.server):
+            quasar.QBtn(
+                "Preview Transforms",
+                click=(self.server.controller.preview_transform),
+                classes="full-width",
+                flat=True,
+            )
+
     def transforms_ui(self):
         with html.Div(trame_server=self.server):
             TransformsWidget(