Compare commits
	
		
			2 Commits
		
	
	
		
			master
			...
			webcam-sup
		
	
	| Author | SHA1 | Date | |
|---|---|---|---|
| 617bb779c0 | |||
| 83de5c2894 | 
							
								
								
									
										57
									
								
								main.py
									
									
									
									
									
								
							
							
						
						
									
										57
									
								
								main.py
									
									
									
									
									
								
							@ -7,28 +7,51 @@ import sys
 | 
			
		||||
import json
 | 
			
		||||
import datetime
 | 
			
		||||
 | 
			
		||||
piCamAvailable = False
 | 
			
		||||
 | 
			
		||||
try:
 | 
			
		||||
    import numpy as np
 | 
			
		||||
    from picamera import PiCamera
 | 
			
		||||
    from picamera.array import PiRGBArray
 | 
			
		||||
    import cv2
 | 
			
		||||
    piCamAvailable = True
 | 
			
		||||
 | 
			
		||||
except Exception as ex:
 | 
			
		||||
    print('[ Error ] some depandincies are missing\n'+ str(ex.args))
 | 
			
		||||
    piCamAvailable = False;
 | 
			
		||||
 | 
			
		||||
try:
 | 
			
		||||
    import cv2
 | 
			
		||||
 | 
			
		||||
except Exception as ex:
 | 
			
		||||
    print('Could not load OpenCV\Closing')
 | 
			
		||||
    sys.exit()
 | 
			
		||||
 | 
			
		||||
def init():
 | 
			
		||||
	go = True
 | 
			
		||||
	print('Initilizing..')
 | 
			
		||||
	piCam = False
 | 
			
		||||
	WebCam = False
 | 
			
		||||
	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')
 | 
			
		||||
			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('[ Error ] Could not start Webcam.. \nExiting')
 | 
			
		||||
			go = False
 | 
			
		||||
	sleep(0.2)
 | 
			
		||||
	try:
 | 
			
		||||
		face_cascade = cv2.CascadeClassifier('haarcascade_frontalface_default.xml')
 | 
			
		||||
		sleep(1)
 | 
			
		||||
@ -41,27 +64,39 @@ 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)
 | 
			
		||||
		start(camera, face_cascade, WebCam)
 | 
			
		||||
 | 
			
		||||
	else:
 | 
			
		||||
		print('closing')
 | 
			
		||||
 | 
			
		||||
def start( camera, face_cascade):
 | 
			
		||||
def start( camera, face_cascade, WebCam):
 | 
			
		||||
	j = 0
 | 
			
		||||
	if not WebCam:
 | 
			
		||||
		rawCapture = PiRGBArray(camera)
 | 
			
		||||
	face_recognizer = cv2.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:
 | 
			
		||||
			if not WebCam:
 | 
			
		||||
				camera.capture(rawCapture, format="bgr")
 | 
			
		||||
				img = rawCapture.array
 | 
			
		||||
			else:
 | 
			
		||||
				ret, img = camera.read()
 | 
			
		||||
		except:
 | 
			
		||||
			print('[ Error ] Can not capture image. Restarting..')
 | 
			
		||||
			init()
 | 
			
		||||
		img = rawCapture.array
 | 
			
		||||
		#img = rawCapture.array
 | 
			
		||||
		gray = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)
 | 
			
		||||
		faces = face_cascade.detectMultiScale(gray, 1.3,5)
 | 
			
		||||
		i = 0;
 | 
			
		||||
 | 
			
		||||
		Loading…
	
		Reference in New Issue
	
	Block a user