From fc663f6efe85e0e922a6a31a1a340ced5e54bae8 Mon Sep 17 00:00:00 2001 From: AlexeyAB Date: Thu, 7 Feb 2019 15:02:36 +0300 Subject: [PATCH] Another minor fix --- build/darknet/x64/darknet_video.py | 8 +++++--- darknet_video.py | 8 +++++--- 2 files changed, 10 insertions(+), 6 deletions(-) diff --git a/build/darknet/x64/darknet_video.py b/build/darknet/x64/darknet_video.py index 9367b313..a5e462fc 100644 --- a/build/darknet/x64/darknet_video.py +++ b/build/darknet/x64/darknet_video.py @@ -25,12 +25,12 @@ def cvDrawBoxes(detections, img): float(x), float(y), float(w), float(h)) pt1 = (xmin, ymin) pt2 = (xmax, ymax) - cv2.rectangle(img, pt1, pt2, (0, 255, 0), 2) + cv2.rectangle(img, pt1, pt2, (0, 255, 0), 1) cv2.putText(img, detection[0].decode() + " [" + str(round(detection[1] * 100, 2)) + "]", - (pt1[0], pt1[1] + 20), cv2.FONT_HERSHEY_SIMPLEX, 1, - [0, 255, 0], 4) + (pt1[0], pt1[1] - 5), cv2.FONT_HERSHEY_SIMPLEX, 0.5, + [0, 255, 0], 2) return img @@ -106,6 +106,8 @@ def YOLO(): image = cvDrawBoxes(detections, frame_resized) image = cv2.cvtColor(image, cv2.COLOR_BGR2RGB) print(1/(time.time()-prev_time)) + cv2.imshow('Demo', image) + cv2.waitKey(3) cap.release() out.release() diff --git a/darknet_video.py b/darknet_video.py index 9367b313..a5e462fc 100644 --- a/darknet_video.py +++ b/darknet_video.py @@ -25,12 +25,12 @@ def cvDrawBoxes(detections, img): float(x), float(y), float(w), float(h)) pt1 = (xmin, ymin) pt2 = (xmax, ymax) - cv2.rectangle(img, pt1, pt2, (0, 255, 0), 2) + cv2.rectangle(img, pt1, pt2, (0, 255, 0), 1) cv2.putText(img, detection[0].decode() + " [" + str(round(detection[1] * 100, 2)) + "]", - (pt1[0], pt1[1] + 20), cv2.FONT_HERSHEY_SIMPLEX, 1, - [0, 255, 0], 4) + (pt1[0], pt1[1] - 5), cv2.FONT_HERSHEY_SIMPLEX, 0.5, + [0, 255, 0], 2) return img @@ -106,6 +106,8 @@ def YOLO(): image = cvDrawBoxes(detections, frame_resized) image = cv2.cvtColor(image, cv2.COLOR_BGR2RGB) print(1/(time.time()-prev_time)) + cv2.imshow('Demo', image) + cv2.waitKey(3) cap.release() out.release()