Compare commits

..

1 Commits

Author SHA1 Message Date
e17b2bf461 Exception of imports 2019-01-19 20:11:37 +03:00
2 changed files with 120 additions and 126 deletions

View File

@ -1,9 +1,7 @@
from time import sleep from time import sleep
from PIL import Image
import os
import serial
import sys import sys
import os
import json import json
import datetime import datetime
@ -11,6 +9,7 @@ try:
import numpy as np import numpy as np
from picamera import PiCamera from picamera import PiCamera
from picamera.array import PiRGBArray from picamera.array import PiRGBArray
import serial
import cv2 import cv2
except Exception as ex: except Exception as ex:

View File

@ -7,10 +7,10 @@ import os
import sys import sys
import json import json
picCount = 0 picCount = 0
new = False new = False
def start(): def start():
print('Starting training..') print('Starting training..')
label = int(input('Label (Integer): ')) label = int(input('Label (Integer): '))
@ -32,24 +32,22 @@ def start():
addPerson(label, newFileName, oldFileName) addPerson(label, newFileName, oldFileName)
addName(name) addName(name)
def clearTrainFolder(): def clearTrainFolder():
print('clearing train folder') print('clearing train folder')
filelist = [f for f in os.listdir('train/') if f.endswith(".jpg")] filelist = [ f for f in os.listdir('train/') if f.endswith(".jpg") ]
for f in filelist: for f in filelist:
os.remove(os.path.join('train/', f)) os.remove(os.path.join('train/', f))
def addPerson(label, newFileName, oldFileName): def addPerson(label, newFileName, oldFileName):
try: try:
# faceFilePath = '/faces/hesham-saeed2' #faceFilePath = '/faces/hesham-saeed2'
faceFile = 'faces/' + oldFileName faceFile = 'faces/'+ oldFileName
face_recognizer = cv2.createLBPHFaceRecognizer() face_recognizer = cv2.createLBPHFaceRecognizer()
global new global new
if new: if new:
print('new file. skipping load function') print ('new file. skipping load function')
else: else:
print('loading old file') print ('loading old file')
face_recognizer.load(faceFile) face_recognizer.load(faceFile)
except: except:
print('[ Error ] Problem in init/loading LBPHfacerecognizer') print('[ Error ] Problem in init/loading LBPHfacerecognizer')
@ -65,19 +63,18 @@ def addPerson(label, newFileName, oldFileName):
gray = cv2.cvtColor(image, cv2.COLOR_BGR2GRAY) gray = cv2.cvtColor(image, cv2.COLOR_BGR2GRAY)
face_cascade = cv2.CascadeClassifier('haarcascade_frontalface_default.xml') face_cascade = cv2.CascadeClassifier('haarcascade_frontalface_default.xml')
faces = face_cascade.detectMultiScale(gray, 1.3, 5) faces = face_cascade.detectMultiScale(gray, 1.3, 5)
for (x, y, w, h) in faces: for (x,y,w,h) in faces:
images.append(gray[y:y + h, x:x + w]) images.append(gray[y:y+h, x:x+w])
labels.append(label) labels.append(label)
cv2.imshow('face', gray[y:y + h, x:x + w]) cv2.imshow('face',gray[y:y+h, x:x+w])
cv2.waitKey(100) cv2.waitKey(100)
print('Learning ' + image_name) print('Learning '+ image_name)
if new: if new:
face_recognizer.train(images, np.array(labels)) face_recognizer.train(images, np.array(labels))
else: else:
face_recognizer.update(images, np.array(labels)) face_recognizer.update(images, np.array(labels))
face_recognizer.save('faces/' + newFileName) face_recognizer.save('faces/'+newFileName)
print('Updated and saved file in faces/' + newFileName) print('Updated and saved file in faces/'+newFileName)
def capture(count): def capture(count):
try: try:
@ -89,18 +86,17 @@ def capture(count):
except: except:
print('[ Error ] Can not initialize PiCamera') print('[ Error ] Can not initialize PiCamera')
sys.exit() sys.exit()
for i in range(1, count + 1): for i in range(1, count+1):
pic = 'train/' + str(i) + '.jpg' pic = 'train/'+str(i)+'.jpg'
camera.capture(pic) camera.capture(pic)
photo = cv2.imread(pic, 1) photo = cv2.imread(pic,1)
cv2.imshow('Photo', photo) cv2.imshow('Photo',photo)
print('Captured ' + str(i) + '\nPath ' + pic) print('Captured '+str(i)+'\nPath '+ pic)
cv2.waitKey(100) cv2.waitKey(100)
sleep(1) sleep(1)
cv2.destroyAllWindows() cv2.destroyAllWindows()
print('Done!') print('Done!')
def promptNew(): def promptNew():
filemode = raw_input('Make new File? (y/n): ') filemode = raw_input('Make new File? (y/n): ')
if filemode == 'y': if filemode == 'y':
@ -111,8 +107,8 @@ def promptNew():
print('incorrect input') print('incorrect input')
promptNew() promptNew()
def addName(newName): def addName(newName):
with open("faces/names.json", "r") as read_file: with open("faces/names.json", "r") as read_file:
exist = False exist = False
namesJson = json.load(read_file) namesJson = json.load(read_file)
@ -121,13 +117,12 @@ def addName(newName):
for name in names: for name in names:
if newName == name: if newName == name:
exist = True exist = True
print('name already in names list with index ' + str(i)) print('name already in names list with index '+str(i))
i += 1 i += 1
if not exist: if not exist:
names.append(newName) names.append(newName)
print('Added ' + newName + 'to names.json file') print('Added '+newName+'to names.json file')
with open("faces/names.json", "w") as write_file: with open("faces/names.json","w") as write_file:
json.dump(names, write_file) json.dump(names,write_file)
start() start()