generated from Hazel/python-project
Compare commits
No commits in common. "ff2088c1d0bc24ad1930355df841e0dd0b1383d1" and "bb6eafbc74b207b78cce323e10a24f01cd87e88d" have entirely different histories.
ff2088c1d0
...
bb6eafbc74
@ -1,6 +1,6 @@
|
||||
from __future__ import annotations
|
||||
|
||||
from typing import Union, List
|
||||
from typing import Union
|
||||
from pathlib import Path
|
||||
import json
|
||||
|
||||
@ -15,7 +15,6 @@ 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:
|
||||
@ -38,11 +37,3 @@ 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]
|
||||
|
@ -18,5 +18,4 @@ def select_bounding_boxes(to_detect: str):
|
||||
fromCenter=False
|
||||
)
|
||||
|
||||
raw_image.bounding_boxes.extend(bounding_boxes)
|
||||
raw_image.write_meta()
|
||||
print(bounding_boxes)
|
||||
|
Loading…
x
Reference in New Issue
Block a user