def stream_response_generator():
#this makes a web cam object
i=1
video_capture=cv2.VideoCapture(0)
while True:
ret, frame = video_capture.read()
# cv2.imwrite("test.jpg", frame)
# frame_number += 1
face_locations = face_recognition.face_locations(frame)
face_encodings = face_recognition.face_encodings(frame, face_locations)
face_names = []
for face_encoding in face_encodings:
match = face_recognition.compare_faces(accepted_face_data, face_encoding, tolerance=0.50)
name = None
if True in match:
first_match_index = match.index(True)
name = accepted_name_list[first_match_index]
# print('sss')
if('aid_' in name):
try:
data=select.select_user(name)
print ("*"*60)
print ("\t\t::USER DISCRIPTIONS::")
print ("*"*60)
print ("USER ID :",data[1])
time=str(data[2])
print ("RECOGNIZED TIME :",time[11:],time[0:11])
print ("*"*60,'\n\n')
current_time=datetime.now()
update.update_time_stamp(current_time,name)
except:
print("EXCEPTIONS:",name)
# print(name)
else:
# name="unknow"
for face_encoding in face_encodings:
match = face_recognition.compare_faces(rejected_face_data, face_encoding, tolerance=0.50)
name = None
if True in match:
first_match_index = match.index(True)
name = rejected_name_list[first_match_index]
else:
for face_encoding in face_encodings:
match = face_recognition.compare_faces(waiting_face_data, face_encoding, tolerance=0.50)
name = None
if True in match:
first_match_index = match.index(True)
name = waiting_name_list[first_match_index]
else:
gray=cv2.cvtColor(frame,cv2.COLOR_BGR2GRAY)
faces=face_cascade.detectMultiScale(gray,1.3,5)
for (top, right, bottom, left) in faces:
fram=frame[right-50:right+left+50,top-50:top+bottom+100]
filename="iSee/static/temp.jpg"
out = cv2.imwrite(filename, fram)
face_locations = face_recognition.face_locations(fram)
face_encodings = face_recognition.face_encodings(fram, face_locations)
for face_encoding in face_encodings:
match = face_recognition.compare_faces(waiting_face_data, face_encoding, tolerance=0.50)
name = None
if True in match:
first_match_index = match.index(True)
name = waiting_name_list[first_match_index]
else:
current_time=datetime.now()
id=insert.create_new_user(current_time)
filename='iSee/static/img_data/wait_list/'+id+'.jpg'
out = cv2.imwrite(filename, fram)
# out = cv2.imwrite(filename, frame)
image = face_recognition.load_image_file(filename)
data = face_recognition.face_encodings(image)[0]
waiting_face_data.append(data)
waiting_name_list.append(id)
row = [id]
for i in data:
row.append(float(i))
with open('iSee/static/trained_data/waiting_list.csv', 'a') as csvFile:
writer = csv.writer(csvFile)
writer.writerow(row)
csvFile.close()
print ("new face recognized Time:",current_time)
data=select.select_user(id)
# os.system(' telegram-cli -k server.pub -W -e "msg Alertsystem WARNING: A NEW PERSON HAS ENTERED !!!! " "safe_quit" ')
# os.system(' telegram-cli -k server.pub -W -e "send_photo Alertsystem %s" "safe_quit"' %(filename) )
# os.system(' telegram-cli -k server.pub -W -e "msg Alertsystem NEW USER_ID: %s " "safe_quit" '%(id))
# os.system(' telegram-cli -k server.pub -W -e "msg Alertsystem Accept : http://%s:%s/iSee/accept/%s " "safe_quit" '%(local_ip,port,id))
# os.system(' telegram-cli -k server.pub -W -e "msg Alertsystem Reject : http://%s:%s/iSee/reject/%s " "safe_quit" '%(local_ip,port,id))
face_names.append(name)
for (top, right, bottom, left), name in zip(face_locations, face_names):
if not name:
continue
if('wid_'in name):
c=(0,165,255)
elif('aid_'in name):
c=(34,139,34)
else:
c= (0, 0, 255)
cv2.rectangle(frame, (left, top), (right, bottom),c, 2)
crop_img = frame[top:bottom, left:right]
cv2.rectangle(frame, (left, bottom - 35), (right, bottom), c, cv2.FILLED)
font = cv2.FONT_HERSHEY_DUPLEX
cv2.putText(frame, name, (left + 6, bottom - 6), font, 1.0, (255, 255, 255), 1)
imgencode=cv2.imencode('.jpg',frame)[1]
stringData=imgencode.tostring()
yield (b'--frame\r\n'
b'Content-Type: text/plain\r\n\r\n'+stringData+b'\r\n')
i+=1
del(camera)
# store the new image with randon name in the directory "img_data/"
id = randint(0, 10000000)
filename = 'img_data/' + str(id) + '.jpg'
out = cv2.imwrite(filename, frame)
# Load a new image and learn how to recognize it.
image = face_recognition.load_image_file(filename)
data = face_recognition.face_encodings(image)[0]
known_face_encodings.append(data)
name = filename.split('/')
user_name, ext = os.path.splitext(name[1])
known_face_names.append(user_name)
known_face_names.append(filename)
current_time = datetime.now()
print("new face recognized Time:", current_time)
#it will insert and update the time stamp of an user's into database
insert.create_new_user(id, current_time)
# os.system(' telegram-cli -k server.pub -W -e "msg Alertsystem WARNING !!!!" "safe_quit"'%())
os.system(
' telegram-cli -k server.pub -W -e "msg Alertsystem WARNING: A NEW PERSON IS ENTERED !!!! " "safe_quit" '
)
os.system(
' telegram-cli -k server.pub -W -e "send_photo Alertsystem %s" "safe_quit"'
% (filename))
os.system(
' telegram-cli -k server.pub -W -e "msg Alertsystem NEW USER_ID: %s " "safe_quit" '
% (id))
print(id)
data = select.select_user(id)
print("------------------------------------------------")
def demo():
i = 1
while True:
ret, frame = video_capture.read()
# cv2.imwrite("test.jpg", frame)
# frame_number += 1
face_locations = face_recognition.face_locations(frame)
face_encodings = face_recognition.face_encodings(frame, face_locations)
face_names = []
for face_encoding in face_encodings:
match = face_recognition.compare_faces(face_data,
face_encoding,
tolerance=0.50)
name = None
if True in match:
first_match_index = match.index(True)
name = face_name_list[first_match_index]
if ('aid_' in name):
try:
data = select.select_user(name)
print("*" * 60)
print("\t\t::USER DISCRIPTIONS::")
print("*" * 60)
print("USER ID :", data[1])
time = str(data[2])
print("RECOGNIZED TIME :", time[11:], time[0:11])
print("*" * 60, '\n\n')
current_time = datetime.now()
update.update_time_stamp(current_time, name)
except:
print("EXCEPTIONS:", name)
# print(name)
else:
# name="unknow"
gray = cv2.cvtColor(frame, cv2.COLOR_BGR2GRAY)
faces = face_cascade.detectMultiScale(gray, 1.3, 5)
for (top, right, bottom, left) in faces:
fram = frame[right - 50:right + left + 50,
top - 50:top + bottom + 100]
filename = "iSee/static/temp.jpg"
out = cv2.imwrite(filename, fram)
face_locations = face_recognition.face_locations(fram)
face_encodings = face_recognition.face_encodings(
fram, face_locations)
for face_encoding in face_encodings:
match = face_recognition.compare_faces(face_data,
face_encoding,
tolerance=0.50)
name = None
if True in match:
first_match_index = match.index(True)
name = face_name_list[first_match_index]
else:
current_time = datetime.now()
id = insert.create_new_user(current_time)
filename = 'iSee/static/img_data/wait_list/' + id + '.jpg'
out = cv2.imwrite(filename, fram)
# out = cv2.imwrite(filename, frame)
image = face_recognition.load_image_file(filename)
data = face_recognition.face_encodings(image)[0]
face_data.append(data)
face_name_list.append(id)
print("new face recognized Time:", current_time)
data = select.select_user(id)
os.system(
' telegram-cli -k server.pub -W -e "msg Alertsystem WARNING: A NEW PERSON HAS ENTERED !!!! " "safe_quit" '
)
os.system(
' telegram-cli -k server.pub -W -e "send_photo Alertsystem %s" "safe_quit"'
% (filename))
os.system(
' telegram-cli -k server.pub -W -e "msg Alertsystem NEW USER_ID: %s " "safe_quit" '
% (id))
os.system(
' telegram-cli -k server.pub -W -e "msg Alertsystem Accept : http://192.168.10.5:8000/iSee/accept/%s " "safe_quit" '
% (id))
os.system(
' telegram-cli -k server.pub -W -e "msg Alertsystem Reject : http://192.168.10.5:8000/iSee/reject/%s " "safe_quit" '
% (id))
face_names.append(name)
for (top, right, bottom,
left), name in zip(face_locations, face_names):
if not name:
continue
if ('wid_' in name):
c = (0, 165, 255)
elif ('aid_' in name):
c = (34, 139, 34)
else:
c = (0, 0, 255)
cv2.rectangle(frame, (left, top), (right, bottom), c, 2)
crop_img = frame[top:bottom, left:right]
cv2.rectangle(frame, (left, bottom - 35), (right, bottom), c,
cv2.FILLED)
font = cv2.FONT_HERSHEY_DUPLEX
cv2.putText(frame, name, (left + 6, bottom - 6), font, 1.0,
(255, 255, 255), 1)
cv2.imshow("iVigil", frame)
cv2.waitKey(10)