reset key

1 files changed, 8 insertions, 1 deletions

diff --git a/test.py b/test.py
index 19afa83..3692b2b 100644
--- a/test.py
+++ b/test.py
@@ -90,5 +90,12 @@ while(cap.isOpened()):
     oldframe=frame
     oldgray=gray
 
-    if cv2.waitKey(1) & 0xFF == ord("q"):
+    key = cv2.waitKey(1) & 0xFF
+    if key == ord("q"):
         break
+
+    if key == ord("r"):
+        total_angle=0.
+        total_x=700
+        total_y=-200
+        screencontent = numpy.zeros((scr_height, scr_width,3), numpy.uint8)