commit

Author: sasank174

__pycache__/facedetector.cpython-38.pyc

@@ -1,10 +1,13 @@
 import cv2
 from flask import Flask, Response, render_template, request, redirect, session, url_for, flash, jsonify
 from flask_mail import Mail, Message
+from numpy import diff
 from werkzeug.utils import secure_filename
 from facedetector import detectmask, detectface, faceencodingvalues, predata
 import db
 import os
+import cv2
+import datetime
 
 
 
@@ -23,6 +26,9 @@
 @app.route("/home")
 def home():
 	predata()
+	global maskdetected
+	global facedetected
+	maskdetected,facedetected = False, False
 	return render_template("index.html")
 
 
@@ -32,21 +38,25 @@ def home():
 def gen_frames():
 	global cap,maskdetected,facedetected
 	while True:
+		if maskdetected == "wait" and facedetected == "wait":
+			continue
 		sucess,img = cap.read()
-		if maskdetected == False and facedetected == False:
-			ans = detectmask(img)
-			if ans == "mask":maskdetected = True
-			else:
-				ans = detectface(img)
-				if ans:facedetected = ans
+		if facedetected == False:
+			ans = detectface(img)
+			if ans: facedetected = ans
+			ret,buffer=cv2.imencode('.jpg',img)
 		elif maskdetected == False:
-			ans = detectmask(img)
+			ans,box,pred = detectmask(img)
 			if ans == "mask":maskdetected = True
-		else:
-			ans = detectface(img)
-			if ans:facedetected = ans
-		yield(b'--frame\r\n'
-					b'Content-Type: image/jpeg\r\n\r\n' + img + b'\r\n')
+			(startX, startY, endX, endY) = box
+			color = (0, 255, 0) if ans == "mask" else (0, 0, 255)
+			(mask, withoutMask) = pred
+			cv2.putText(img, ans, (startX, startY - 10),cv2.FONT_HERSHEY_SIMPLEX, 0.45, color, 2)
+			cv2.rectangle(img, (startX, startY), (endX, endY), color, 2)
+			ret,buffer=cv2.imencode('.jpg',img)
+
+		frame=buffer.tobytes()
+		yield(b'--frame\r\n'b'Content-Type: image/jpeg\r\n\r\n' + frame + b'\r\n')
 
 @app.route("/video_feed")
 def video_feed():
@@ -59,10 +69,29 @@ def recorded():
 
 @app.route('/recorddone', methods = ['POST', 'GET'])
 def recorddone():
-	global maskdetected,facedetected
-	cap.release()
-	maskdetected,facedetected = False, False
-	return redirect(url_for('home'))
+	print("recorddone")
+	global maskdetected
+	global facedetected
+	ans = db.select("SELECT inside,fine FROM tarpusers WHERE email = '"+facedetected+"'")
+	inside,fine = ans[0]
+	if inside == "YES":
+		now = datetime.datetime.now()
+		now = now.strftime("%Y-%m-%d %H:%M:%S")
+		db.insert("UPDATE tarpusers SET inside = '"+ now +"' WHERE email = '"+facedetected+"'")
+	else:
+		diff = datetime.datetime.now() - datetime.datetime.strptime(inside, "%Y-%m-%d %H:%M:%S")
+		days, seconds = diff.days, diff.seconds
+		hours = days * 24 + seconds // 3600
+		if hours > 3:
+			fine = fine + (hours - 3) * 10
+			db.insert("UPDATE tarpusers SET fine = '"+ str(fine) +"', inside = 'YES'  WHERE email = '"+facedetected+"'")
+		else:db.insert("UPDATE tarpusers SET inside = 'YES' WHERE email = '"+facedetected+"'")
+	# cap.release()
+	cv2.destroyAllWindows()
+	x = facedetected
+	maskdetected,facedetected = "wait", "wait"
+	return render_template("redirect.html", email = x,fine = fine)
+	# return redirect(url_for('redirect'))
 
 # ============================================================================================================
 
@@ -121,7 +150,7 @@ def admin():
 @app.route("/logout")
 def logout():
 	session.clear()
-	return redirect(url_for("login"))
+	return redirect(url_for("home"))
 
 
 if __name__ == '__main__':

app.py

@@ -10,18 +10,13 @@
 allmails = []
 
 def faceencodingvalues(img):
-	print("===============start=====================================================")
 	imgload = face_recognition.load_image_file(img)
 	imgload = cv2.cvtColor(imgload,cv2.COLOR_BGR2RGB)
 	try:
 		faceloc = face_recognition.face_locations(imgload)[0]  # (260, 825, 528, 557)
 	except:
 		return [],[]
 	encodeimg = face_recognition.face_encodings(imgload)[0]
-	# print("===============faceloc=====================================================")
-	# print(faceloc)
-	# print("==================encodeimg==================================================")
-	# print(encodeimg)
 
 	return (encodeimg,faceloc)
 
@@ -37,79 +32,22 @@ def predata():
 	knownencodings = encodinglist
 	allmails = emaillist
 
-
 def detectmask(img):
 	(locs,preds) = detect_and_predict_mask(img)
 	for (box,pred) in zip(locs,preds):
 		(mask,withoutmask) = pred
-		if mask > withoutmask:
-			return "mask"
-		else:
-			return "withoutmask"
+		if mask > withoutmask:return ("mask",box,pred)
+		else:return ("withoutmask",box,pred)
 
 def detectface(img):
 	imgS = cv2.resize(img,(0,0),None,0.25,0.25)
 	imgS = cv2.cvtColor(imgS,cv2.COLOR_BGR2RGB)
 	facesS = face_recognition.face_locations(imgS)
 	encodeS = face_recognition.face_encodings(imgS,facesS)
 
-
 	for encodeFace,faceLoc in zip(encodeS,facesS):
 		matches = face_recognition.compare_faces(knownencodings,encodeFace)
 		faceDis = face_recognition.face_distance(knownencodings,encodeFace)
 		matchindex = np.argmin(faceDis)
 		if faceDis[matchindex]<0.6:
-			return allmails[matchindex]
-# def detect(img,maskdetected,facedetected):
-# 	if maskdetected == False and facedetected == False:
-# 		(locs, preds) = detect_and_predict_mask(img)
-# 		for (box, pred) in zip(locs, preds):
-# 			(startX, startY, endX, endY) = box
-# 			(mask, withoutMask) = pred
-# 			if mask > withoutMask:
-# 				return "mask"
-# 	elif maskdetected == False and facedetected == True:
-# 		(locs, preds) = detect_and_predict_mask(img)
-# 		for (box, pred) in zip(locs, preds):
-# 			(startX, startY, endX, endY) = box
-# 			(mask, withoutMask) = pred
-# 			if mask > withoutMask:
-# 				return "mask"
-		# else:
-
-		# label = "Mask" if mask > withoutMask else "No Mask"
-		# color = (0, 255, 0) if label == "Mask" else (0, 0, 255)
-		# label = "{}: {:.2f}%".format(label, max(mask, withoutMask) * 100)
-		# cv2.putText(frame, label, (startX, startY - 10),cv2.FONT_HERSHEY_SIMPLEX, 0.45, color, 2)
-		# cv2.rectangle(frame, (startX, startY), (endX, endY), color, 2)
-
-	# cv2.imshow('frame', frame)
-	# if cv2.waitKey(1) & 0xFF == ord('q'):
-	# 	break
-	
-
-
-# def detect(img):
-# 	imgS = cv2.resize(img,(0,0),None,0.25,0.25)
-# 	imgS = cv2.cvtColor(imgS,cv2.COLOR_BGR2RGB)
-# 	facesS = face_recognition.face_locations(imgS)
-# 	encodeS = face_recognition.face_encodings(imgS,facesS)
-
-# 	for encodeFace,faceLoc in zip(encodeS,facesS):
-# 		matches = face_recognition.compare_faces(knownencodings,encodeFace)
-# 		faceDis = face_recognition.face_distance(knownencodings,encodeFace)
-# 		matchindex = np.argmin(faceDis)
-# 		# print("=======matchindex========================================")
-# 		print(matchindex)
-# 		# print(count)
-# 		print(matches[matchindex])
-# 		print(faceDis[matchindex])
-# 		# if matches[matchindex] and count == matchindex and faceDis[matchindex]<0.6:
-# 		if matches[matchindex] and faceDis[matchindex]<0.6:
-# 			print("================detect=================================================================")
-# 			ret,buffer = cv2.imencode('.jpg',img)
-# 			frame = buffer.tobytes()
-# 			return (frame,"YES")
-# 	ret,buffer = cv2.imencode('.jpg',img)
-# 	frame = buffer.tobytes()
-# 	return (frame,"NO")
+			return allmails[matchindex]

facedetector.py

@@ -6,9 +6,9 @@
 
 
 def detect_and_predict_mask(frame):
-	maskNet = load_model(r"mask_detector.model")
-	prototxtPath = r"deploy.prototxt"
-	weightsPath = r"res10_300x300_ssd_iter_140000.caffemodel"
+	maskNet = load_model(r"mask/mask_detector.model")
+	prototxtPath = r"mask/deploy.prototxt"
+	weightsPath = r"mask/res10_300x300_ssd_iter_140000.caffemodel"
 	faceNet = cv2.dnn.readNet(prototxtPath, weightsPath)
 
 	(h, w) = frame.shape[:2]
@@ -40,23 +40,4 @@ def detect_and_predict_mask(frame):
 		faces = np.array(faces, dtype="float32")
 		preds = maskNet.predict(faces, batch_size=32)
 
-	return (locs, preds)
-
-# while True:
-# 	ret, frame = cap.read()
-# 	frame = imutils.resize(frame, width=800)
-# 	frame = cv2.flip(frame, 1)
-# 	(locs, preds) = detect_and_predict_mask(frame, faceNet, maskNet)
-
-# 	for (box, pred) in zip(locs, preds):
-# 		(startX, startY, endX, endY) = box
-# 		(mask, withoutMask) = pred
-# 		label = "Mask" if mask > withoutMask else "No Mask"
-# 		color = (0, 255, 0) if label == "Mask" else (0, 0, 255)
-# 		label = "{}: {:.2f}%".format(label, max(mask, withoutMask) * 100)
-# 		cv2.putText(frame, label, (startX, startY - 10),cv2.FONT_HERSHEY_SIMPLEX, 0.45, color, 2)
-# 		cv2.rectangle(frame, (startX, startY), (endX, endY), color, 2)
-
-# 	cv2.imshow('frame', frame)
-# 	if cv2.waitKey(1) & 0xFF == ord('q'):
-# 		break
+	return (locs, preds)

facedetectors.py

@@ -10,9 +10,17 @@
 
 <body>
     <h1>face and mask detection</h1>
+    <h3 id="facename"></h3>
 
     <img src="{{ url_for('video_feed') }}" height="60%">
 
+    
+    <label for="">face detected</label>
+    <input type="checkbox" id="face">
+    <label for="">mask detected</label>
+    <input type="checkbox" id="mask">
+
+
     <script src="//ajax.googleapis.com/ajax/libs/jquery/1.9.1/jquery.min.js"></script>
     <script>
         $(document).ready(function () {
@@ -22,11 +30,17 @@ <h1>face and mask detection</h1>
         function refresh() {
             setTimeout(function () {
                 $.getJSON('/recorded', function (data) {
-                    console.log(data);
-                    // if (data == "YES") {
-                    //   window.location.href = "http://127.0.0.1:7890/recorddone";
-                    // }
-                    //$('#test').text(data);
+                    console.log(data.face,data.mask);
+                    document.getElementById('facename').innerHTML = data.face;
+                    if (data.face) $('#face').prop('checked', true);
+                    else $('#face').prop('checked', false);
+                    
+                    if (data.mask) $('#mask').prop('checked', true);
+                    else $('#mask').prop('checked', false);
+                    if (data.face != false & data.mask == true) {
+                        //console.log('face and mask');
+                        window.location.href = "http://127.0.0.1:8000/recorddone";
+                    }
                 });
                 refresh();
             }, 1000);

maskdetect.py

@@ -0,0 +1,19 @@
+<!DOCTYPE html>
+<html lang="en">
+<head>
+    <meta charset="UTF-8">
+    <meta http-equiv="X-UA-Compatible" content="IE=edge">
+    <meta name="viewport" content="width=device-width, initial-scale=1.0">
+    <title>Document</title>
+</head>
+<body>
+    <h1>{{ email }} recorded</h1>
+    <h2>Fine {{ fine }} rs</h2>
+
+    <script>
+        setTimeout(function () {
+            window.location.href = "http://127.0.0.1:8000";
+        }, 5000);
+    </script>
+</body>
+</html>