Gun detection using Open CV is a very useful project for your resume and also useful for real-life projects. Many industries use this kind of technology for weapon detection. So it will be great if you can make such a project using python in very simple steps.
Difficulty:
It will be very simple code. By using some simple method of OpenCV(open source computer vision) you can create this project and have some fun
Modules:
- OpenCV(for installation use pip install opencv-python on your terminal)
- NumPy(for installation use pip install numpy)
- Imutils(for installation use pip install imutils)
- Datetime(you don't need to install it)
Notes:
In this project, I use a pre-ready dataset for detecting guns you can get it from here
Code:
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()
frame = imutils.resize(frame, width = 500)
gray = cv2.cvtColor(frame, cv2.COLOR_BGR2GRAY)
gun = gun_cascade.detectMultiScale(gray,1.3, 5,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
# print(datetime.date(2019))
# draw the text and timestamp on the frame
cv2.putText(frame, datetime.datetime.now().strftime("%m-%d-%Y %H:%M%p"),
(10, frame.shape[0] - 10),
cv2.FONT_HERSHEY_SIMPLEX,
0.35, (0, 0, 255), 1)
cv2.imshow("Security Feed", frame)
key = cv2.waitKey(1) & 0xFF
if key == ord('q'):
break
if gun_exist:
print("guns detected")
else:
print("guns NOT detected")
camera.release()
cv2.destroyAllWindows()
Github:
You can also get the code from here
Comments
Post a Comment