Merge pull request #102 from Aayush-hub/master

Added notification pop up

CONTRIBUTORS.md

@@ -24,4 +24,6 @@ Thanks goes to these wonderful people ([emoji key](https://allcontributors.org/d
        <td align="center"><a href="https://github.com/abhinavmaharana"><img src="https://avatars0.githubusercontent.com/u/59000244?s=400&u=4d68a964de00ddf1d896a16264efef9498cb89cf&v=4" width="100px;" alt=""/><br /><sub><b>Abhinav Maharana
 </b></sub></a><br />
        <td align="center"><a href="https://github.com/Pisich"><img src="https://avatars1.githubusercontent.com/u/62033118?s=400&v=4" width="100px;" alt=""/><br /><sub><b>Carlos
+</b></sub></a><br />
+       <td align="center"><a href="https://github.com/Aayush-hub"><img src="https://avatars0.githubusercontent.com/u/65889104?s=60&v=4" width="100px;" alt=""/><br /><sub><b>Aayush Garg
 </b></sub></a><br />

detect_mask_image.py

@@ -9,7 +9,7 @@
 import argparse
 import cv2
 import os
-
+from plyer import notification
 def mask_image():
 	# construct the argument parser and parse the arguments
 	ap = argparse.ArgumentParser()
@@ -87,6 +87,13 @@ def mask_image():
 			# the bounding box and text
 			label = "Mask" if mask > withoutMask else "No Mask"
 			color = (0, 255, 0) if label == "Mask" else (0, 0, 255)
+			if label == "No Mask":
+			notification.notify(
+				title = "***No Mask Detected***",
+				message = "Wear Mask to stay safe! ",
+				app_icon = "images/1.ico",    #ico file should be downloaded
+				timeout = 1
+            			)
 
 			# include the probability in the label
 			label = "{}: {:.2f}%".format(label, max(mask, withoutMask) * 100)

detect_mask_video.py

@@ -13,6 +13,7 @@
 import time
 import cv2
 import os
+from plyer import notification
 
 def detect_and_predict_mask(frame, faceNet, maskNet):
 	# grab the dimensions of the frame and then construct a blob
@@ -125,7 +126,15 @@ def detect_and_predict_mask(frame, faceNet, maskNet):
 		# the bounding box and text
 		label = "Mask" if mask > withoutMask else "No Mask"
 		color = (0, 255, 0) if label == "Mask" else (0, 0, 255)
-
+		if label == "No Mask":
+			notification.notify(
+				title = "***No Mask Detected***",
+                		message = "Wear Mask to stay safe! ",
+                		app_icon = "images/1.ico",    #ico file should be downloaded
+                		timeout = 1
+            		)
+
+
 		# include the probability in the label
 		label = "{}: {:.2f}%".format(label, max(mask, withoutMask) * 100)
 
@@ -151,3 +160,4 @@ def detect_and_predict_mask(frame, faceNet, maskNet):
 # do a bit of cleanup
 cv2.destroyAllWindows()
 vs.stop()
+