From 8e037586bc1fa6d2afa7c346fc0d63df854a4563 Mon Sep 17 00:00:00 2001 From: Tobias Eidelpes Date: Thu, 19 Jan 2023 10:53:14 +0100 Subject: [PATCH] Remove comments --- yolo-second-run/model.py | 8 +------- 1 file changed, 1 insertion(+), 7 deletions(-) diff --git a/yolo-second-run/model.py b/yolo-second-run/model.py index 9b55d7b..0fbac52 100644 --- a/yolo-second-run/model.py +++ b/yolo-second-run/model.py @@ -47,10 +47,7 @@ def detect(img_path: str, yolo_path: str, resnet_path: str): # Classify ROI in RGB predictions[idx] = classify(second_stage, cropped_image[..., ::-1]) - # cv2.imshow('cropped ' + str(idx), cropped_image) - # cv2.waitKey(0) - - # Draw bounding box and class on original image + # Draw bounding box and number on original image original = cv2.rectangle(original, (xmin, ymin), (xmax, ymax), (0, 255, 0), 2) 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), 2, cv2.LINE_AA) - # cv2.waitKey(0) - # cv2.destroyAllWindows() - cv2.imshow('original with bounding box', original) cv2.waitKey(0) cv2.destroyAllWindows()