126 lines
2.7 KiB
Python
126 lines
2.7 KiB
Python
|
|
from time import sleep
|
|
from picamera import PiCamera
|
|
import cv2
|
|
import numpy as np
|
|
import os
|
|
import serial
|
|
|
|
#ser = serial.Serial('/dev/ttyUSB0',9600)
|
|
def init():
|
|
go = True
|
|
#OpenShifter()
|
|
#if Sensors():
|
|
print('Initilizing Camera and cascade components..')
|
|
try:
|
|
camera = PiCamera()
|
|
camera.resolution = (640, 480)
|
|
sleep(2)
|
|
print('[ OK ] Camera')
|
|
except:
|
|
print('[ Error ] Can not initialize PiCamera')
|
|
go = False
|
|
|
|
try:
|
|
face_cascade = cv2.CascadeClassifier('haarcascade_frontalface_default.xml')
|
|
sleep(1)
|
|
print('[ OK ] CascadeClassifier')
|
|
except:
|
|
print('[ Error ] Can not load cascade File')
|
|
go = False
|
|
if (go):
|
|
print('Starting Photo loop..')
|
|
start(camera, face_cascade)
|
|
|
|
else:
|
|
print('closing')
|
|
#else:
|
|
#print('Sensors not Satisfied')
|
|
|
|
def start( camera, face_cascade):
|
|
j = 0
|
|
while True:
|
|
#try:
|
|
camera.capture('img.jpg')
|
|
img = cv2.imread('img.jpg',1)
|
|
gray = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)
|
|
faces = face_cascade.detectMultiScale(gray, 1.3, 5)
|
|
i = 0;
|
|
for (x,y,w,h) in faces:
|
|
i += 1
|
|
j += 1
|
|
roi_gray = gray[y:y+h, x:x+w]
|
|
roi_color = img[y:y+h, x:x+w]
|
|
#try:
|
|
#cv2.imwrite('train/'+str(j)+'.jpg',img)
|
|
#print(j)
|
|
#except Exception as ex:
|
|
#print(ex)
|
|
#print('Drawing on face ', i)
|
|
cv2.rectangle(img,(x,y),(x+w,y+h),(255,0,0),2)
|
|
cv2.putText(img, recognize(roi_gray), (x, y), cv2.FONT_HERSHEY_PLAIN, 1.7, (255, 0, 0), 2)
|
|
cv2.imshow('image',img)
|
|
cv2.waitKey(100)
|
|
if cv2.waitKey(1) & 0xFF == ord('q'):
|
|
cv2.destroyAllWindows()
|
|
break
|
|
|
|
#except:
|
|
#print('[ Error ] Unexpected exception')
|
|
#print('Closing')
|
|
#break
|
|
|
|
def train(number):
|
|
|
|
train_path = 'train/'
|
|
face_recognizer = cv2.createLBPHFaceRecognizer()
|
|
images = []
|
|
labels = []
|
|
image_names = os.listdir(train_path)
|
|
for image_name in image_names:
|
|
if image_name.startswith("."):
|
|
continue;
|
|
image_path = train_path + image_name
|
|
image = cv2.imread(image_path)
|
|
gray = cv2.cvtColor(image, cv2.COLOR_BGR2GRAY)
|
|
images.append(gray)
|
|
labels.append(number)
|
|
print('training')
|
|
face_recognizer.train(images, np.array(labels))
|
|
face_recognizer.save('faces/hesham')
|
|
|
|
|
|
def recognize(image):
|
|
|
|
subjects = ['','Hesham','Saeed']
|
|
face_recognizer = cv2.createLBPHFaceRecognizer()
|
|
face_recognizer.load('faces/Hesham-Saeed')
|
|
img = image.copy()
|
|
label= face_recognizer.predict(img)
|
|
print(label)
|
|
if label[0] == 1 and int(label[1]) < 200:
|
|
return 'Hesham'
|
|
elif label[0] == 2 and int(label[1]) < 200:
|
|
return 'Saeed'
|
|
else:
|
|
return 'Unknown'
|
|
|
|
def Sensors():
|
|
|
|
while 1:
|
|
try:
|
|
line = ser.readline()
|
|
print(line)
|
|
if 'A' in line:
|
|
return True
|
|
else:
|
|
print('Sensors not OK..')
|
|
|
|
except:
|
|
print('[ Error ] Can not connect with arduino. Retrying..')
|
|
|
|
def OpenShifter():
|
|
ser.write('O')
|
|
|
|
init()
|