zenon/master-thesis
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Return the currently processed image

Browse Source
This commit is contained in:
Tobias Eidelpes 2023-01-19 10:58:44 +01:00
parent 2f3d3da32c
commit 0ef8eb0cd4
1 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
yolo-second-run/model.py
View File

@ -27,7 +27,8 @@ def detect(img_path: str, yolo_path: str, resnet_path: str):
:param str img_path: path to image :param str img_path: path to image
:param yolo_path: path to yolo weights :param yolo_path: path to yolo weights
:param resnet_path: path to resnet weights :param resnet_path: path to resnet weights
:returns: dict of bounding boxes and their predictions :returns: tuple of recent image and dict of bounding boxes and
their predictions
""" """
img = cv2.imread(img_path) img = cv2.imread(img_path)
@ -59,7 +60,7 @@ def detect(img_path: str, yolo_path: str, resnet_path: str):
cv2.FONT_HERSHEY_SIMPLEX, 0.9, (255, 255, 255), cv2.FONT_HERSHEY_SIMPLEX, 0.9, (255, 255, 255),
2, cv2.LINE_AA) 2, cv2.LINE_AA)
return predictions return (original, predictions)
def get_boxes(model, img): def get_boxes(model, img):