fixing center sound issue

AceCentre · Aug 13, 2021 · 58ebded · 58ebded
1 parent a2da0e6
commit 58ebded
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/eye_commander/image_capture/image_capture.py b/eye_commander/image_capture/image_capture.py
@@ -105,8 +105,8 @@ def gather_data(self, directions:list):
             if cv2.waitKey(1) & 0xFF == ord('n'):
                     # end demo when ESC key is entered 
 
-                for direction in directions:
-                    if direction != 'center':
+                for ind, direction in enumerate(directions):
+                    if ind != 0:
                         sounds.play_tone()
                     while self.camera.isOpened():