UPD: display only one image instead of two for slicing

Giskard-AI · Jan 23, 2024 · a851683 · a851683
1 parent ac9b268
commit a851683
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 6 deletions.
diff --git a/giskard_vision/landmark_detection/dataloaders/wrappers.py b/giskard_vision/landmark_detection/dataloaders/wrappers.py
@@ -362,6 +362,8 @@ class FilteredDataLoader(DataLoaderWrapper):
         FilteredDataLoader: Filtered data loader instance.
     """
 
+    dataloader_type = "filter"
+
     @property
     def name(self):
         """

diff --git a/giskard_vision/landmark_detection/detectors/base.py b/giskard_vision/landmark_detection/detectors/base.py
@@ -45,12 +45,14 @@ def get_results(self, model: Any, dataset: Any) -> Sequence[ScanResult]:
             os.makedirs("examples_images", exist_ok=True)
             filename_examples = []
 
-            filename_example_dataloader_ref = f"examples_images/{dataset.name}.png"
-            cv2.imwrite(
-                filename_example_dataloader_ref,
-                cv2.resize(dataset[0][0][0], (0, 0), fx=0.3, fy=0.3)
-            )
-            filename_examples.append(filename_example_dataloader_ref)
+            if dl.dataloader_type != "filter":
+
+                filename_example_dataloader_ref = f"examples_images/{dataset.name}.png"
+                cv2.imwrite(
+                    filename_example_dataloader_ref,
+                    cv2.resize(dataset[0][0][0], (0, 0), fx=0.3, fy=0.3)
+                )
+                filename_examples.append(filename_example_dataloader_ref)
 
             filename_example_dataloader = f"examples_images/{dl.name}.png"
             cv2.imwrite(