🐛 Fix scale_factor.

Author: shaneahmed

tiatoolbox/models/dataset/dataset_abc.py

@@ -399,10 +399,8 @@ def __init__(  # skipcq: PY-R1000  # noqa: PLR0915
                 `units`. Expected to be positive and of (height, width).
                 Note, this is not at level 0.
             resolution (Resolution):
-                Check (:class:`.WSIReader`) for details. When
-                `mode='tile'`, value is fixed to be `resolution=1.0` and
-                `units='baseline'` units: check (:class:`.WSIReader`) for
-                details.
+                Requested resolution corresponding to units. Check
+                (:class:`WSIReader`) for details.
             units (Units):
                 Units in which `resolution` is defined.
             auto_get_mask (bool):

tiatoolbox/models/engine/engine_abc.py

@@ -1052,6 +1052,21 @@ def _run_wsi_mode(
                 patch_mode=False,
                 ioconfig=self._ioconfig,
             )
+
+            # get units and resolution from dataloader.
+            dataloader_units = dataloader.dataset.units
+            dataloader_resolution = dataloader.dataset.resolution
+
+            slide_resolution = (1.0, 1.0)
+            if dataloader_units != "baseline":
+                wsimeta_dict = dataloader.dataset.reader.info.as_dict()
+                slide_resolution = wsimeta_dict[dataloader_units]
+
+            scale_factor = tuple(np.divide(slide_resolution, dataloader_resolution))
+
+            if dataloader_units != "mpp":
+                scale_factor = tuple(np.divide(dataloader_resolution, slide_resolution))
+
             raw_predictions = self.infer_wsi(
                 dataloader=dataloader,
                 save_path=save_path[image],
@@ -1062,6 +1077,7 @@ def _run_wsi_mode(
                 **kwargs,
             )
             kwargs["output_file"] = out[image]
+            kwargs["scale_factor"] = scale_factor
             out[image] = self.save_predictions(
                 processed_predictions=processed_predictions,
                 output_type=output_type,