Computer >> 컴퓨터 >  >> 프로그램 작성 >> Python

Python의 웹캠 동작 감지기 프로그램?

<시간/>

여기에서 우리는 웹캠에서 가져온 이미지를 분석하고 움직임을 감지하고 웹캠 비디오의 시간 간격을 csv 파일에 저장하는 파이썬 프로그램을 작성할 것입니다.

필수 라이브러리

이를 위해 OpenCV 및 pandas 라이브러리를 사용할 것입니다. 아직 설치되지 않은 경우 다음과 같이 pip를 사용하여 설치할 수 있습니다.

$pip install opencv2, pandas

예시 코드

#Import required libraries
import cv2
import pandas as pd
import time
from datetime import datetime

#Initialise variables
stillImage = None
motionImage = [ None, None ]
time = []

# Initializing the DataFrame with start and end time
df = pd.DataFrame(columns = ["start", "end"])

# Capturing video
video = cv2.VideoCapture(0)

while True:
   # Start reading image from video
   check, frame = video.read()
   motion = 0

   # Convert color image to gray_scale image
   gray = cv2.cvtColor(frame, cv2.COLOR_BGR2GRAY)
   gray = cv2.GaussianBlur(gray, (21, 21), 0)
   if stillImage is None:
      stillImage = gray
      continue
   # Still Image and current image.
   diff_frame = cv2.absdiff(stillImage, gray)

   # change the image to white if static background and current frame is greater than 25.
   thresh_frame = cv2.threshold(diff_frame, 25, 255, cv2.THRESH_BINARY)[1]
   thresh_frame = cv2.dilate(thresh_frame, None, iterations = 2)
   # Finding contour and hierarchy from a moving object.
   contours,hierachy = cv2.findContours(thresh_frame.copy(),
      cv2.RETR_EXTERNAL, cv2.CHAIN_APPROX_SIMPLE)
   for contour in contours:
      if cv2.contourArea(contour) < 10000:
         continue
      motion = 1
      (x, y, w, h) = cv2.boundingRect(contour)
      cv2.rectangle(frame, (x, y), (x + w, y + h), (0, 255, 0), 3)
   # Append current status of motion
   motionImage.append(motion)
   motionImage = motionImage[-2:]
   # Append Start time of motion
   if motionImage[-1] == 1 and motionImage[-2] == 0:
      time.append(datetime.now())

   # Append End time of motion
   if motionImage[-1] == 0 and motionImage[-2] == 1:
      time.append(datetime.now())
   # Displaying image in gray_scale
   cv2.imshow("Gray_Frame", gray)

   # Display black and white frame & if the intensity difference is > 25, it turns white
   cv2.imshow("Threshold Frame", thresh_frame)
   # Display colored frame
   cv2.imshow("Colored_Frame", frame)

   key = cv2.waitKey(1)
   # Press q to stop the process
   if key == ord('q'):
      if motion == 1:
         time.append(datetime.now())
      break

# Append time of motion
for i in range(0, len(time), 2):
   df = df.append({"Start":time[i], "End":time[i + 1]}, ignore_index = True)

# Creating a csv file in which time of movements will be saved
df.to_csv("FrameInMotion_time.csv")

video.release()

# close window
cv2.destroyAllWindows()

출력

Python의 웹캠 동작 감지기 프로그램?

3가지 다른 모드(회색조, 컬러 및 흑백)로 웹캠의 현재 움직임을 표시할 3가지 다른 Windows가 있음을 알 수 있습니다.

또한 웹캠 모션의 날짜/시간을 csv에 저장하고 csv의 출력은 다음과 같습니다.

FrameMotion_time.csv(출력)

start     end      End                            Start
0                  2019-02-21 18:10:59.718005     2019-02-21 18:08:35.791487