secure_pixelation/__main__.py

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

secure_pixelation/data_classes.py

@@ -1,6 +1,6 @@
 from __future__ import annotations
 
-from typing import Union, List, Tuple
+from typing import Union, List
 from pathlib import Path
 import json
 
@@ -17,18 +17,13 @@ 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.tolist())
+    raw_image.bounding_boxes.extend(bounding_boxes)
     raw_image.write_meta()