Hesham/FaceRecognition-python
1
0
Fork 0
Code Issues Pull Requests Releases 1 Activity

Formatting

Browse Source
This commit is contained in:
hesham 2019-01-22 20:28:13 +03:00 committed by HeshamTB
parent a82b98154f
commit fb84cf1bb5
1 changed files with 108 additions and 103 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

49
trainer.py
View File

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