Fix shifting of pixels to return whole image array, not one element.

2018-12-18 13:26:36 +10:30
parent df8bec3b3e
commit 363122fe29
1 changed files with 2 additions and 1 deletions
--- a/GestureRecognition/SimpleHandRecogniser.py
+++ b/GestureRecognition/SimpleHandRecogniser.py
@@ -91,7 +91,8 @@ class SimpleHandRecogniser(HandRecogniser):

    def __shift_pixels(self, image, shift_radius):
        image[:,:,0] = image[:,:,0] + shift_radius
-        return np.where(image[:,:,0] > 179, image[:,:,0] - 179, image[:,:,0])
+        image[:,:,0] = np.where(image[:,:,0] > 179, image[:,:,0] - 179, image[:,:,0])
+        return image

    def get_gesture(self):
        """