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

9
trainer.py
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,12 +32,14 @@ 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'
@ -76,6 +78,7 @@ def addPerson(label, newFileName, oldFileName):
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:
print('Initializing camera') print('Initializing camera')
@ -97,6 +100,7 @@ def capture(count):
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':
@ -107,8 +111,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)
@ -125,4 +129,5 @@ def addName(newName):
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()