From e104a8f45c2fa6180b4817697c70b77136f4883c Mon Sep 17 00:00:00 2001 From: Lars Noack Date: Thu, 24 Apr 2025 10:52:12 +0200 Subject: [PATCH] feat: added bounding boxes to meta data --- secure_pixelation/data_classes.py | 11 ++++++++++- 1 file changed, 10 insertions(+), 1 deletion(-) 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]