minor bug in preprocess function

bug reported by [@sydney0zq](#10)
nrupatunga · Dec 15, 2017 · 79b6d6e · 79b6d6e
1 parent 1a8b35d
commit 79b6d6e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/goturn/network/regressor.py b/goturn/network/regressor.py
@@ -65,7 +65,7 @@ def preprocess(self, image):
             image_out = cv2.cvtColor(image, cv2.COLOR_BGRA2GRAY)
         elif num_channels == 3 and image.shape[2] == 4:
             image_out = cv2.cvtColor(image, cv2.COLOR_BGRA2BGR)
-        elif num_channels == 3 and image.shape[2] == 2:
+        elif num_channels == 3 and image.shape[2] == 1:
             image_out = cv2.cvtColor(image, cv2.COLOR_GRAY2BGR)
         else:
             image_out = image