Fixed critical bug in goal center detection

2018-06-30 17:11:33 +02:00
parent 6d9d522692
commit b9d6898691
5 changed files with 12 additions and 6 deletions
--- a/pykick/finders.py
+++ b/pykick/finders.py
@@ -118,7 +118,7 @@ class GoalFinder(object):
        return goal

    def left_right_post(self, contour):
-        return contour[:,0].min(), contour[:,0].max()
+        return contour[...,0].min(), contour[...,0].max()

    def goal_center(self, contour):
        l, r = self.left_right_post(contour)