51 lines
1.1 KiB
Python
51 lines
1.1 KiB
Python
|
|
from time import sleep
|
|
from picamera import PiCamera
|
|
from io import BytesIO
|
|
import cv2
|
|
import numpy as np
|
|
|
|
def init():
|
|
print('Initilizing Camera and cascade components..')
|
|
camera = PiCamera()
|
|
print('[ OK ] Camera')
|
|
camera.resolution = (640, 480)
|
|
face_cascade = cv2.CascadeClassifier('haarcascade_frontalface_default.xml')
|
|
print('[ OK ] CascadeClassifier')
|
|
print('Starting Photo loop..')
|
|
start()
|
|
|
|
def start():
|
|
while True:
|
|
print('Taking photo')
|
|
camera.capture('img.jpg')
|
|
print('Photo Captured')
|
|
img = cv2.imread('img.jpg',1)
|
|
gray = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)
|
|
print('Photo Converted')
|
|
faces = face_cascade.detectMultiScale(gray, 1.3, 5)
|
|
print('Photo Processed')
|
|
i = 0;
|
|
for (x,y,w,h) in faces:
|
|
i = i + 1
|
|
cv2.rectangle(img,(x,y),(x+w,y+h),(255,0,0),2)
|
|
cv2.imwrite('result.jpg', gray)
|
|
roi_gray = gray[y:y+h, x:x+w]
|
|
roi_color = img[y:y+h, x:x+w]
|
|
print('Drawing on face ', i)
|
|
|
|
cv2.imshow('image',img)
|
|
cv2.waitKey(100)
|
|
if cv2.waitKey(1) & 0xFF == ord('q'):
|
|
cv2.destroyAllWindows()
|
|
break
|
|
|
|
print('Finshed')
|
|
|
|
def recognize():
|
|
|
|
def sendOK():
|
|
|
|
init()
|
|
|