working on perturbation detectors

giskard_vision/core/dataloaders/base.py

@@ -5,7 +5,13 @@
 import cv2
 import numpy as np
 
-from giskard_vision.core.dataloaders.meta import MetaData
+from giskard_vision.core.dataloaders.meta import (
+    MetaData,
+    get_brightness,
+    get_entropy,
+    get_image_channel_number,
+    get_image_size,
+)
 
 from ..types import TypesBase
 

giskard_vision/core/detectors/metadata_scan_detector.py

@@ -5,7 +5,8 @@
 import pandas as pd
 
 from giskard_vision.core.detectors.base import DetectorVisionBase, ScanResult
-from giskard_vision.core.issues import IssueGroup
+from giskard_vision.core.issues import IssueGroup, PerformanceIssueMeta
+
 from giskard_vision.core.tests.base import MetricBase
 from giskard_vision.utils.errors import GiskardImportError
 

giskard_vision/image_classification/dataloaders/loaders.py

@@ -145,7 +145,9 @@ def get_image(self, idx: int) -> np.ndarray:
         Returns:
             np.ndarray: The image data.
         """
-        return np.array(self.ds[idx]["image"])
+        raw_img = self.get_raw_hf_image(idx)
+
+        return np.array(raw_img)
 
     def get_labels(self, idx: int) -> Optional[np.ndarray]:
         """
@@ -246,7 +248,9 @@ def get_image(self, idx: int) -> np.ndarray:
         Returns:
             np.ndarray: The image data.
         """
-        return np.array(self.ds[idx]["img"])
+        raw_img = self.get_raw_hf_image(idx)
+
+        return np.array(raw_img)
 
     def get_labels(self, idx: int) -> Optional[np.ndarray]:
         """

giskard_vision/landmark_detection/dataloaders/loaders.py

@@ -5,11 +5,11 @@
 import cv2
 import numpy as np
 
-from giskard_vision.core.dataloaders.base import EthicalIssueMeta, PerformanceIssueMeta
 from giskard_vision.core.dataloaders.meta import MetaData
 from giskard_vision.core.dataloaders.tfds import DataLoaderTensorFlowDatasets
 from giskard_vision.core.dataloaders.utils import flatten_dict
 from giskard_vision.core.issues import EthicalIssueMeta, PerformanceIssueMeta
+
 from giskard_vision.landmark_detection.types import Types
 
 from .base import DataLoaderBase