diff --git a/robot/arlo.py b/robot/arlo.py index 15bfc8f..039e00d 100644 --- a/robot/arlo.py +++ b/robot/arlo.py @@ -14,7 +14,7 @@ FOCAL_LENGTH = 1691 DEFAULT_CALIBRATION_CODE = "40.40-40.40-40.40_ff-ff-ff-ff" CAMERA_MATRIX = np.array([[FOCAL_LENGTH,0,512],[0,FOCAL_LENGTH,360],[0,0,1]]) -DIST_COEF = np.array([1,1,1,1]) +DIST_COEF = np.array([0,0,0,0]) def test_forward(arlo, l_power, r_power): arlo.reset_encoder_counts() @@ -104,7 +104,7 @@ class Arlo(): # avg_size = (avg_width + avg_height)/2 # return (FOCAL_LENGTH * 14.5)/avg_size - print(bounding_box) + return cv2.aruco.estimatePoseSingleMarkers( bounding_box, 14.5, CAMERA_MATRIX, DIST_COEF )