#main.py
import cv2
import numpy as np
cap = cv2.VideoCapture("assets/live cam.mp4")
_, frame1 = cap.read()
_, frame2 = cap.read()
i = 0
while cap.isOpened():
diff = cv2.absdiff(frame1, frame2)
gray = cv2.cvtColor(diff, cv2.COLOR_BGR2GRAY)
blur = cv2.GaussianBlur(gray, (5, 5), 0)
_, thresh = cv2.threshold(blur, 20, 255, cv2.THRESH_BINARY)
dilated = cv2.dilate(thresh, None, iterations=3)
contours, _ = cv2.findContours(dilated, cv2.RETR_TREE, cv2.CHAIN_APPROX_SIMPLE)
#cv2.drawContours(frame1, contours, -1, (0, 0, 255), 2)
for contour in contours:
(x, y, w, h) = cv2.boundingRect(contour)
if cv2.contourArea(contour) < 2000:
continue
cv2.rectangle(frame1, (x, y), (x+w, y+h), (0, 0, 255), 5)
cv2.imshow('feed', frame1)
i += 1
name = 'motion detection/img' + str(i) + '.jpg'
cv2.imwrite(name, frame1)
frame1 = frame2
_, frame2 = cap.read()
if cv2.waitKey(1) == ord('q'):
break
if cv2.waitKey(1) == ord('p'):
# wait until any key is pressed
cv2.waitKey(-1)
cap.release()
cv2.destroyAllWindows()
------------------------------------
#screenshot.py
import pyscreenshot as ImageGrab
for i in range(200):
im = ImageGrab.grab(bbox=(100, 500, 1900, 1500)) # X1,Y1,X2,Y2
name = 'screenshot/img' + str(i) + '.jpg'
im.save(name)
reference:
screenshot
pip install pyscreenshot
pip install image
live cam
No comments:
Post a Comment