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

Add clarifying comments

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

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

@ -33,10 +33,12 @@ def detect(img_path: str, yolo_path: str, resnet_path: str):
img = cv2.imread(img_path) img = cv2.imread(img_path)
original = img.copy() original = img.copy()
(first_stage, second_stage) = load_models(yolo_path, resnet_path) (first_stage, second_stage) = load_models(yolo_path, resnet_path)
# Get bounding boxes from object detection model
box_coords = get_boxes(first_stage, img) box_coords = get_boxes(first_stage, img)
box_coords.sort_values(by=['xmin'], ignore_index=True, inplace=True) box_coords.sort_values(by=['xmin'], ignore_index=True, inplace=True)
predictions = {}
predictions = {}
for idx, row in box_coords.iterrows(): for idx, row in box_coords.iterrows():
xmin, xmax = int(row['xmin']), int(row['xmax']) xmin, xmax = int(row['xmin']), int(row['xmax'])
ymin, ymax = int(row['ymin']), int(row['ymax']) ymin, ymax = int(row['ymin']), int(row['ymax'])