feat: blacking out image

secure_pixelation/__main__.py

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

secure_pixelation/data_classes.py

@@ -1,6 +1,6 @@
 from __future__ import annotations
 
-from typing import Union, List
+from typing import Union, List, Tuple
 from pathlib import Path
 import json
 
@@ -17,13 +17,18 @@ class RawImage:
         self.meta_data = self.read_meta()
 
         self.image = self.get_image()
-    
+
     def _get_path(self, ending: str, original_suffix: bool = False) -> Path:
         if original_suffix:
             return self.file.with_name(self.file.stem + "_" + ending + self.file.suffix)
         else:
             return self.file.with_name(self.file.stem + "_" + ending)
 
+    def get_dir(self, name: str) -> Path:
+        p = self._get_path(ending=name, original_suffix=False)
+        p.mkdir(exist_ok=True, parents=True)
+        return p
+
     def read_meta(self) -> dict:
         if not self.meta_file.exists():
             return {}

secure_pixelation/get_bounding_boxes.py

@@ -18,5 +18,5 @@ def select_bounding_boxes(to_detect: str):
         fromCenter=False
     )
 
-    raw_image.bounding_boxes.extend(bounding_boxes)
+    raw_image.bounding_boxes.extend(bounding_boxes.tolist())
     raw_image.write_meta()