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

Remove comments

Browse Source
This commit is contained in:
Tobias Eidelpes 2023-01-19 10:53:14 +01:00
parent ad1bcc1dab
commit 8e037586bc
1 changed files with 1 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -47,10 +47,7 @@ def detect(img_path: str, yolo_path: str, resnet_path: str):
# Classify ROI in RGB # Classify ROI in RGB
predictions[idx] = classify(second_stage, cropped_image[..., ::-1]) predictions[idx] = classify(second_stage, cropped_image[..., ::-1])
# cv2.imshow('cropped ' + str(idx), cropped_image) # Draw bounding box and number on original image
# cv2.waitKey(0)
# Draw bounding box and class on original image
original = cv2.rectangle(original, (xmin, ymin), (xmax, ymax), original = cv2.rectangle(original, (xmin, ymin), (xmax, ymax),
(0, 255, 0), 2) (0, 255, 0), 2)
original = cv2.putText(original, str(idx), (xmin + 5, ymin + 25), original = cv2.putText(original, str(idx), (xmin + 5, ymin + 25),
@ -60,9 +57,6 @@ 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)
# cv2.waitKey(0)
# cv2.destroyAllWindows()
cv2.imshow('original with bounding box', original) cv2.imshow('original with bounding box', original)
cv2.waitKey(0) cv2.waitKey(0)
cv2.destroyAllWindows() cv2.destroyAllWindows()