91 lines
2.9 KiB
Python
91 lines
2.9 KiB
Python
from picamera2 import Picamera2
|
|
import cv2
|
|
import numpy as np
|
|
import time
|
|
import requests
|
|
from datetime import datetime
|
|
|
|
# ntfy.sh topic (choose your unique topic name)
|
|
rpi_ip = "localhost:80"
|
|
ntfy_topic = "motion-sensing" # Replace with your ntfy.sh topic
|
|
ntfy_url = f"http://{rpi_ip}/{ntfy_topic}"
|
|
# If using ntfy.sh server and not the local docker container, set rpi_ip = "ntfy.sh"
|
|
|
|
# Initialize the camera
|
|
picam2 = Picamera2()
|
|
picam2.configure(picam2.create_preview_configuration(main={"format": "XRGB8888", "size": (640, 480)}))
|
|
picam2.start()
|
|
|
|
time.sleep(2) # Give the camera some time to initialize
|
|
|
|
# Read the initial frames
|
|
frame1 = picam2.capture_array()
|
|
frame2 = picam2.capture_array()
|
|
|
|
motion_detected = False
|
|
|
|
while True:
|
|
# Calculate the absolute difference between the two frames
|
|
diff = cv2.absdiff(frame1, frame2)
|
|
|
|
# Convert the difference to grayscale
|
|
gray = cv2.cvtColor(diff, cv2.COLOR_BGR2GRAY)
|
|
|
|
# Apply a Gaussian blur to the grayscale image
|
|
blur = cv2.GaussianBlur(gray, (5, 5), 0)
|
|
|
|
# Threshold the blurred image to highlight the motion
|
|
_, thresh = cv2.threshold(blur, 20, 255, cv2.THRESH_BINARY)
|
|
|
|
# Dilate the thresholded image to fill in holes
|
|
dilated = cv2.dilate(thresh, None, iterations=3)
|
|
|
|
# Find contours (motion areas) in the dilated image
|
|
contours, _ = cv2.findContours(dilated, cv2.RETR_TREE, cv2.CHAIN_APPROX_SIMPLE)
|
|
|
|
for contour in contours:
|
|
if cv2.contourArea(contour) < 500:
|
|
continue
|
|
x, y, w, h = cv2.boundingRect(contour)
|
|
cv2.rectangle(frame1, (x, y), (x + w, y + h), (0, 255, 0), 2)
|
|
|
|
# If motion is detected and not already notified, send a notification
|
|
if not motion_detected:
|
|
current_time = datetime.now().strftime("%H:%M:%S") # Get current time
|
|
message = f"Motion detected in room at {current_time}" # Format message
|
|
|
|
print("Motion detected! Sending notification...")
|
|
try:
|
|
response = requests.post(
|
|
ntfy_url,
|
|
json={"message": message} # Send only the message
|
|
)
|
|
if response.status_code == 200:
|
|
print("Notification sent successfully!")
|
|
else:
|
|
print(f"Failed to send notification. Status code: {response.status_code}")
|
|
except Exception as e:
|
|
print(f"Error sending notification: {e}")
|
|
|
|
motion_detected = True
|
|
|
|
# If no contours are found, reset the motion_detected flag
|
|
if len(contours) == 0:
|
|
motion_detected = False
|
|
|
|
# Display the result
|
|
cv2.imshow("Motion Detection", frame1)
|
|
|
|
# Update frames: frame1 becomes frame2, and a new frame is read as frame2
|
|
frame1 = frame2
|
|
frame2 = picam2.capture_array()
|
|
|
|
# Press 'q' to exit the loop
|
|
if cv2.waitKey(10) & 0xFF == ord('q'):
|
|
break
|
|
|
|
# Release resources and close all OpenCV windows
|
|
picam2.stop()
|
|
cv2.destroyAllWindows()
|
|
|