Hazel/secure_pixelation
1
0
Fork 0
generated from Hazel/python-project
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat: added bounding boxes to meta data

Browse Source
This commit is contained in:
Hazel 2025-04-24 10:52:12 +02:00
parent bb6eafbc74
commit e104a8f45c
1 changed files with 10 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
secure_pixelation/data_classes.py
View File

@ -1,6 +1,6 @@
from __future__ import annotations from __future__ import annotations
from typing import Union from typing import Union, List
from pathlib import Path from pathlib import Path
import json import json
@ -15,6 +15,7 @@ class RawImage:
self.meta_file = self._get_path("boxes.json") self.meta_file = self._get_path("boxes.json")
self.meta_data = self.read_meta() self.meta_data = self.read_meta()
self.image = self.get_image() self.image = self.get_image()
def _get_path(self, ending: str, original_suffix: bool = False) -> Path: def _get_path(self, ending: str, original_suffix: bool = False) -> Path:
@ -37,3 +38,11 @@ class RawImage:
def get_image(self) -> np.ndarray: def get_image(self) -> np.ndarray:
return cv2.imread(str(self.file)) 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]