From d244b96a2608c974062b1975c60d36b4cf544128 Mon Sep 17 00:00:00 2001 From: imDigamber <118442466+imDigamber@users.noreply.github.com> Date: Fri, 27 Oct 2023 03:32:07 +0530 Subject: [PATCH] Create firstrepo.py --- firstrepo.py | 41 +++++++++++++++++++++++++++++++++++++++++ 1 file changed, 41 insertions(+) create mode 100644 firstrepo.py diff --git a/firstrepo.py b/firstrepo.py new file mode 100644 index 00000000..7c20620c --- /dev/null +++ b/firstrepo.py @@ -0,0 +1,41 @@ +import numpy as np +import cv2 +import imutils +import datetime + +gun_cascade = cv2.CascadeClassifier('cascade.xml') +camera = cv2.VideoCapture(0) +firstFrame = None +gun_exist = False +while True: + ret, frame = camera.read() + if frame is None: + break + frame = imutils.resize(frame, width=500) + gray = cv2.cvtColor(frame, cv2.COLOR_BGR2GRAY) + gun = gun_cascade.detectMultiScale(gray, 1.3, 20, minSize=(100, 100)) + if len(gun) > 0: + gun_exist = True + for (x, y, w, h) in gun: + frame = cv2.rectangle(frame, (x, y), (x + w, y + h), (255, 0, 0), 2) + roi_gray = gray[y:y + h, x:x + w] + roi_color = frame[y:y + h, x:x + w] + if firstFrame is None: + firstFrame = gray + continue + cv2.putText(frame, datetime.datetime.now().strftime("%A %d %B %Y %I:%M:%S %p"), + (10, frame.shape[0] - 10), + cv2.FONT_HERSHEY_SIMPLEX, + 0.35, (0, 0, 255), 1) + if gun_exist: + print("Guns detected") + plt.imshow(frame) + break + else: + cv2.imshow("Security Feed", frame) + key = cv2.waitKey(1) & 0xFF + if key == ord('q'): + break + +camera.release() +cv2.destroyAllWindows()