diff --git a/drive_between_arucos.py b/drive_between_arucos.py index 5f435ae..47086c3 100644 --- a/drive_between_arucos.py +++ b/drive_between_arucos.py @@ -58,8 +58,8 @@ def main(): arucos = find_arucos(arlo) aruco_positions = np.array([cv2.aruco.estimatePoseSingleMarkers( - np.array([aruco[0]]), 14.5, CAMERA_MATRIX, DIST_COEF - )[0][0] for aruco in arucos]) + np.array([aruco]), 14.5, CAMERA_MATRIX, DIST_COEF + )[1][0][0] for aruco in arucos]) position = [ np.average(aruco_positions[:,0]),