Allow making screenshots

* face_thuglife.py: Likewise, click `k' to capture the webcam
TravorLZH · Sep 3, 2019 · fe20040 · fe20040
1 parent 9eb0596
commit fe20040
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/face_thuglife.py b/face_thuglife.py
@@ -94,5 +94,9 @@ def decor_img(name):
     frame=cv2.flip(frame,1)
     frame=mark_eye(frame)
     cv2.imshow("Thug Life",frame)
-    if cv2.waitKey(1) & 0xFF == ord('q'):
+    key=cv2.waitKey(1) & 0xFF
+    if key==ord('q'):
         break
+    elif key==ord('k'):
+        cv2.imwrite("sunglass_screenshot.png",frame)
+        print("Saved")