diff --git a/secure_pixelation/data_classes.py b/secure_pixelation/data_classes.py index e6ad680..9bbdfe7 100644 --- a/secure_pixelation/data_classes.py +++ b/secure_pixelation/data_classes.py @@ -1,6 +1,6 @@ from __future__ import annotations -from typing import Union +from typing import Union, List from pathlib import Path import json @@ -15,6 +15,7 @@ class RawImage: self.meta_file = self._get_path("boxes.json") self.meta_data = self.read_meta() + self.image = self.get_image() def _get_path(self, ending: str, original_suffix: bool = False) -> Path: @@ -37,3 +38,11 @@ class RawImage: def get_image(self) -> np.ndarray: return cv2.imread(str(self.file)) + + @property + def bounding_boxes(self) -> List[List[int]]: + _key = "bounding_boxes" + if _key not in self.meta_data: + self.meta_data[_key] = [] + + return self.meta_data[_key]