WIP: add multi-version support
Beter Init function
This commit is contained in:
parent
83de5c2894
commit
617bb779c0
27
main.py
27
main.py
@ -28,24 +28,30 @@ except Exception as ex:
|
||||
|
||||
def init():
|
||||
go = True
|
||||
piCam = False
|
||||
WebCam = False
|
||||
print('Initilizing..')
|
||||
print('Initilizing Camera ..')
|
||||
if piCamAvailable:
|
||||
try:
|
||||
camera = PiCamera()
|
||||
camera.resolution = (640, 480)
|
||||
sleep(0.2)
|
||||
print('[ OK ] Camera')
|
||||
go = True
|
||||
piCam = True
|
||||
print('[ OK ] Started PiCamera')
|
||||
except:
|
||||
print('[ Error ] Can not initialize PiCamera\Trying webcam..')
|
||||
print('[ Warning ] Could not start PiCamera. Trying Webcam.. ')
|
||||
piCam = False
|
||||
go = False
|
||||
else:
|
||||
try:
|
||||
camera = cv2.VideoCapture()
|
||||
print('[ OK ] WebCamera')
|
||||
go = True
|
||||
WebCam = True
|
||||
except Exception as ex:
|
||||
print(ex.args)
|
||||
print('[ Error ] Could not start Webcam.. \nExiting')
|
||||
go = False
|
||||
sleep(0.2)
|
||||
try:
|
||||
face_cascade = cv2.CascadeClassifier('haarcascade_frontalface_default.xml')
|
||||
sleep(1)
|
||||
@ -58,7 +64,7 @@ def init():
|
||||
except OSError as ex:
|
||||
print('Found (unknown) folder')
|
||||
|
||||
if (go):
|
||||
if go:
|
||||
print('Starting Photo loop..')
|
||||
print('Known people are '+str(loadNames()))
|
||||
start(camera, face_cascade, WebCam)
|
||||
@ -70,8 +76,15 @@ def start( camera, face_cascade, WebCam):
|
||||
j = 0
|
||||
if not WebCam:
|
||||
rawCapture = PiRGBArray(camera)
|
||||
face_recognizer = cv2.face.LBPHFaceRecognizer_create() #cv2.face.createLBPHFaceRecognizer()
|
||||
# face_recognizer = cv2.face.LBPHFaceRecognizer_create() #cv2.face.createLBPHFaceRecognizer()
|
||||
face_recognizer = cv2.face_LBPHFaceRecognizer('faces/h')
|
||||
# face_recognizer.read('faces/h') # Try on Pi
|
||||
##WIP
|
||||
try:
|
||||
face_recognizer.load('faces/h')
|
||||
except:
|
||||
pass
|
||||
|
||||
names = loadNames()
|
||||
while 1:
|
||||
try:
|
||||
|
Loading…
Reference in New Issue
Block a user