NikolajDanger/Roboteksperimentarium
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

✨

Browse Source
This commit is contained in:
NikolajDanger
2022-09-21 13:07:00 +02:00
parent 280d9037c0
commit 0c13f8cb2e
2 changed files with 5 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
estimate_distance.py
View File

@ -2,9 +2,10 @@ from robot import Arlo
def main():
arlo = Arlo(False)
image = arlo.robot.take_photo()
bounding_boxes = arlo.find_aruco(image)
print(arlo.estimate_distance(bounding_boxes[0][0]))
for _ in range(5):
image = arlo.robot.take_photo()
bounding_boxes = arlo.find_aruco(image)
print(arlo.estimate_distance(bounding_boxes[0][0]))
if __name__ == "__main__":
main()

2
robot/arlo.py
View File

@ -99,7 +99,7 @@ class Arlo():
avg_size = (avg_width + avg_height)/2
return (FOCAL_LENGTH * 15)/avg_size
return (FOCAL_LENGTH * 14.5)/avg_size
def find_aruco(self, image):
aruco_dict = cv2.aruco.Dictionary_get(cv2.aruco.DICT_6X6_250)