diff --git a/giskard_vision/landmark_detection/dataloaders/wrappers.py b/giskard_vision/landmark_detection/dataloaders/wrappers.py index 6be46409..015c4a9c 100644 --- 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 index d30e04d9..229bdb7b 100644 --- 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(