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 PIL import Image
import os
import serial
import sys
import os
import json
import datetime
@ -11,6 +9,7 @@ try:
import numpy as np
from picamera import PiCamera
from picamera.array import PiRGBArray
import serial
import cv2
except Exception as ex:

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,14 +32,12 @@ 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") ]
for f in filelist:
os.remove(os.path.join('train/', f))
def addPerson(label, newFileName, oldFileName):
try:
#faceFilePath = '/faces/hesham-saeed2'
@ -78,7 +76,6 @@ def addPerson(label, newFileName, oldFileName):
face_recognizer.save('faces/'+newFileName)
print('Updated and saved file in faces/'+newFileName)
def capture(count):
try:
print('Initializing camera')
@ -100,7 +97,6 @@ def capture(count):
cv2.destroyAllWindows()
print('Done!')
def promptNew():
filemode = raw_input('Make new File? (y/n): ')
if filemode == 'y':
@ -111,8 +107,8 @@ def promptNew():
print('incorrect input')
promptNew()
def addName(newName):
with open("faces/names.json", "r") as read_file:
exist = False
namesJson = json.load(read_file)
@ -129,5 +125,4 @@ def addName(newName):
with open("faces/names.json","w") as write_file:
json.dump(names,write_file)
start()