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 __future__ import annotations
|
||||||
|
|
||||||
from typing import Union, List
|
from typing import Union
|
||||||
from pathlib import Path
|
from pathlib import Path
|
||||||
import json
|
import json
|
||||||
|
|
||||||
@ -15,7 +15,6 @@ 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:
|
||||||
@ -38,11 +37,3 @@ 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]
|
|
||||||
|
@ -18,5 +18,4 @@ def select_bounding_boxes(to_detect: str):
|
|||||||
fromCenter=False
|
fromCenter=False
|
||||||
)
|
)
|
||||||
|
|
||||||
raw_image.bounding_boxes.extend(bounding_boxes)
|
print(bounding_boxes)
|
||||||
raw_image.write_meta()
|
|
||||||
|
Loading…
x
Reference in New Issue
Block a user