30 lines
689 B
Python
30 lines
689 B
Python
import cv2
|
|
import robot
|
|
|
|
def find_aruco(arlo):
|
|
image = arlo.take_photo()
|
|
|
|
aruco_dict = cv2.aruco.Dictionary_get(cv2.aruco.DICT_6X6_250)
|
|
aruco_params = cv2.aruco.DetectorParameters_create()
|
|
corners, ids, _ = cv2.aruco.detectMarkers(
|
|
image,
|
|
aruco_dict,
|
|
parameters=aruco_params
|
|
)
|
|
print("Corners: ", detected[0])
|
|
print("Ids: ", detected[1])
|
|
print("Rejected: ", detected[2])
|
|
|
|
for bounding, n in zip(corners, ids):
|
|
cv2.line(image, bounding[0], bounding[1], (0,255,0))
|
|
|
|
return image
|
|
|
|
def main():
|
|
arlo = robot.Robot()
|
|
image = find_aruco(arlo)
|
|
cv2.imwrite("test.png", image)
|
|
|
|
if __name__ == "__main__":
|
|
main()
|