get data bugs fixed

Author: jerinkantony

README.md

@@ -4,4 +4,4 @@ Training deep learning model to play video games, self driving, etc, inspired by
 Steps to run:
 1) python3 getdata.py  #click on 'r' button and mouse left click top-left and bottom-right 
                           corners of window to be saved. Use keyboard and play the game.
-   
+2) 

getdata.py

@@ -7,6 +7,15 @@
 from getscreen import GetScreen
 import time
 import os
+import sys
+
+def compare_imgs(im1,im2):
+    if (im1 is None) or (im2 is None) :
+        return False
+    else:
+        difference = cv2.subtract(im1, im2)    
+        result = not np.any(difference)
+        return result
 
 
 if __name__ == '__main__': 
@@ -17,29 +26,46 @@
 
     scobj=GetScreen()
     kobj = GetKey()
-    
-    
+    cv2.namedWindow('crop', cv2.WINDOW_NORMAL)
+    im1=None
     while True:
         im,t2 = scobj.getimg()
-        cv2.imshow('screen', im)
-        
         keys = kobj.get_pressed_keys()
-        str1 = "" 
-  
-        # using join function join the list s by  
-        # separating words by str1 
-        str1 = str1.join(keys) 
-        print('Keys pressed: ',str1)
-        
-        if cv2.waitKey(1) == 27:
-            cv2.destroyAllWindows()
-            break
-        nam = os.path.join(dst, str(int(t2* 10**7)) +'_'+ str1+'.jpg')
-        #import pdb;pdb.set_trace()
-        cv2.imwrite(nam,im)
 
 
 
+        #import pdb;pdb.set_trace()
+        if (not compare_imgs(im1,im)) or keys :
+            
+            im1=im
+            cv2.imshow('crop', im)
+
+            str1 = ''
+            keys2=[]
+
+            if keys:
+                keys2.append('_')
+                for key in keys:
+                    keys2.append(key)
+                    keys2.append('_')
+                str1 = str1.join(keys2)
+                print('Keys pressed: ',str1)
+                #import pdb;pdb.set_trace()
+                
+            
+            if cv2.waitKey(50) == 27:
+                cv2.destroyAllWindows()
+                break
+            nam = os.path.join(dst, str(int(t2* 10**7)) +str1+'.jpg')
+            cv2.imwrite(nam,im)
+        if keys:
+            if keys[-1]=='esc':
+                break
+        
+    
+    sys.stdout.flush()
+    print('getdata end')
+            
 
 
 

getkey.py

@@ -6,6 +6,7 @@ class to manage keypress, returns keys that wr pressed on being continued to be
 '''
 class GetKey():
     def __init__(self):
+        print('GetKey constructor')
         self.keyp=[]#
         self.keyr=[]
 
@@ -14,29 +15,38 @@ def __init__(self):
             on_press=self.on_press,
             on_release=self.on_release)
         self.listener.start()
+    def __del__(self):
+        self.listener.stop()
+        print('GetKey destroctor')
+        
     def stop(self):
         print('stopping listener')
         self.listener.stop()
     def on_press(self,key):
+        #import pdb;pdb.set_trace()
         try:
             if key.char not in self.keyp:
                 self.keyp.append(key.char)
                 #print(' {0} pressed'.format(self.keyp))
 
         except AttributeError:
             #self.keyp.append(key.char)
-            print('special key {0} pressed'.format(key))
+            #print('special key {0} pressed'.format(key.name))
+            self.keyp.append(key.name)
+            '''
             if key == keyboard.Key.esc:
                 # Stop listener
                 print('esc pressed')
+            '''
 
     def on_release(self,key):
         try:
             if key.char not in self.keyr:
                     self.keyr.append(key.char)
             #print('{0} released'.format(self.keyr))
         except AttributeError:
-            print('special key {0} released'.format(key))
+            self.keyr.append(key.name)
+            #print('special key {0} released'.format(key))
 
     def get_pressed_keys(self):
         #returns keys pressed 
@@ -45,6 +55,7 @@ def get_pressed_keys(self):
         #print('{0} released2'.format(self.keyr))
         self.keyp=list(set(self.keyp)-set(self.keyr))
         self.keyr=[]
+        sys.stdout.flush()
         return k
 
     def checkforkey(self,key):
@@ -59,16 +70,22 @@ def checkforkey(self,key):
 
 if __name__ == '__main__':       
     kobj = GetKey()
+    keys=[]
     while 1:
         time.sleep(.5)
 
         keys = kobj.get_pressed_keys()
-        print('Keys pressed: ',keys)
+        if keys:
+            print('Keys pressed: ',keys)
 
         '''
         kflag = kobj.checkforkey('r')
         print('kflag: ',kflag)
         '''
+        
+        if keys:
+            if keys[-1]=='esc':
+                break
 
 
 

getmouse.py

@@ -18,8 +18,7 @@ def stop(self):
         self.listener.stop()
 
     def on_move(self,x, y):
-        print('Pointer moved to {0}'.format(
-            (x, y)))
+        #print('Pointer moved to {0}'.format((x, y)))
         self.pos=[x,y]
 
     def on_click(self,x, y, button, pressed):
@@ -33,9 +32,8 @@ def on_click(self,x, y, button, pressed):
             #return False
 
     def on_scroll(self,x, y, dx, dy):
-        print('Scrolled {0} at {1}'.format(
-            'down' if dy < 0 else 'up',
-            (x, y)))
+        pass
+        #print('Scrolled {0} at {1}'.format('down' if dy < 0 else 'up',(x, y)))
     def getpts(self):
         #get clicked points
         return self.pts

getscreen.py

@@ -6,13 +6,13 @@
 from getmouse import GetMouse 
 import time
 import os
+import sys
 
 class GetScreen():
     def __init__(self):
         self.sct = mss()
+        self.im1=None
 
-        
-            
         kobj = GetKey()
         print('Press r and select roi using mouse leftbutton diagonal clicks ')
         while True:
@@ -22,6 +22,7 @@ def __init__(self):
             if kflag:
                 print('\rkey pressed: ',kflag)
                 kobj.stop()
+                sys.stdout.flush()
                 break
 
         mobj = GetMouse()
@@ -30,15 +31,14 @@ def __init__(self):
         while True:
             time.sleep(.5)
             pts = mobj.getpts()
-            pos = mobj.pos
+            
             if len(pts)==1:
                 if not pt1:
                     pt1 = pts[0]
-                    print('pt1: ',pt1)
+                pt2 = mobj.pos
 
-                elif not pt2:
-                    pt2 = pos
-            elif len(pts)==2:
+            elif len(pts)>=2:
+                mobj.stop()
                 break
 
         print('box pts: ',pt1,pt2)
@@ -47,12 +47,13 @@ def __init__(self):
 
     def getimg(self):
         t2 = time.time()
-        sct_img = np.array(self.sct.grab(self.bounding_box))
-        im = cv2.cvtColor(sct_img, cv2.COLOR_BGR2RGB)
 
-        print('FPS: {} ',format(1/(t2-self.t1)))
+        im = np.array(self.sct.grab(self.bounding_box))
+            
+
+        #print('FPS: {} ',format(1/(t2-self.t1)))
         self.t1=t2
-        return im,t2
+        return im,self.t1
 
 if __name__ == '__main__': 
 
@@ -61,18 +62,18 @@ def getimg(self):
         os.makedirs(dst) 
 
     scobj=GetScreen()
-    
+    cv2.namedWindow('crop', cv2.WINDOW_NORMAL)
 
     while True:
         im,t2 = scobj.getimg()
-        cv2.imshow('screen', im)
+        cv2.imshow('crop', im)
 
         if cv2.waitKey(1) == 27:
             cv2.destroyAllWindows()
             break
         nam = os.path.join(dst, str(int(t2* 10**7))+'.jpg')
         #import pdb;pdb.set_trace()
-        cv2.imwrite(nam,im)
+        #cv2.imwrite(nam,im)