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

feat: improved bounding box format

Browse Source
This commit is contained in:
Hazel 2025-04-24 11:21:31 +02:00
parent cb9e594837
commit b88f9c22a3
3 changed files with 9 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
secure_pixelation/__main__.py
View File

@ -1,7 +1,8 @@
from .get_bounding_boxes import select_bounding_boxes from .get_bounding_boxes import select_bounding_boxes
from .pixelation_process import pixelate
def cli(): def cli():
print(f"Running secure_pixelation") print(f"Running secure_pixelation")
select_bounding_boxes("assets/human_detection/test.png") pixelate("assets/human_detection/test.png")

8
secure_pixelation/data_classes.py
View File

@ -1,6 +1,6 @@
from __future__ import annotations from __future__ import annotations
from typing import Union, List from typing import Union, List, Tuple
from pathlib import Path from pathlib import Path
import json import json
@ -18,15 +18,13 @@ class RawImage:
self.image = self.get_image() self.image = self.get_image()
self.steps_dir = self._get_dir("steps")
def _get_path(self, ending: str, original_suffix: bool = False) -> Path: def _get_path(self, ending: str, original_suffix: bool = False) -> Path:
if original_suffix: if original_suffix:
return self.file.with_name(self.file.stem + "_" + ending + self.file.suffix) return self.file.with_name(self.file.stem + "_" + ending + self.file.suffix)
else: else:
return self.file.with_name(self.file.stem + "_" + ending) return self.file.with_name(self.file.stem + "_" + ending)
def _get_dir(self, name: str) -> Path: def get_dir(self, name: str) -> Path:
p = self._get_path(ending=name, original_suffix=False) p = self._get_path(ending=name, original_suffix=False)
p.mkdir(exist_ok=True, parents=True) p.mkdir(exist_ok=True, parents=True)
return p return p
@ -47,7 +45,7 @@ class RawImage:
return cv2.imread(str(self.file)) return cv2.imread(str(self.file))
@property @property
def bounding_boxes(self) -> List[List[int]]: def bounding_boxes(self) -> List[Tuple[Tuple[int, int], Tuple[int, int]]]:
_key = "bounding_boxes" _key = "bounding_boxes"
if _key not in self.meta_data: if _key not in self.meta_data:
self.meta_data[_key] = [] self.meta_data[_key] = []

5
secure_pixelation/get_bounding_boxes.py
View File

@ -18,5 +18,8 @@ def select_bounding_boxes(to_detect: str):
fromCenter=False fromCenter=False
) )
raw_image.bounding_boxes.extend(bounding_boxes.tolist()) raw_image.bounding_boxes.extend([
((box[0], box[1]), (box[0] + box[2], box[1] + box[3]))
for box in bounding_boxes
])
raw_image.write_meta() raw_image.write_meta()