depau/board_camera_bot
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

Change aruco dictionary

Browse source
This commit is contained in:
Davide Depau 2024-08-11 20:23:47 +02:00
parent 92240741cc
commit 34fcb1fc0a
2 changed files with 10 additions and 6 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

14
aruco_realtime.py
View file

@ -4,14 +4,14 @@ import cv2
from cv2 import aruco from cv2 import aruco
# Define the ArUco dictionary # Define the ArUco dictionary
aruco_dict = cv2.aruco.getPredefinedDictionary(cv2.aruco.DICT_6X6_250) aruco_dict = cv2.aruco.getPredefinedDictionary(cv2.aruco.DICT_4X4_50)
aruco_params = cv2.aruco.DetectorParameters() aruco_params = cv2.aruco.DetectorParameters()
# Specify the RTSP URL # Specify the RTSP URL
rtsp_url = sys.argv[1] # rtsp_url = sys.argv[1]
# Capture the video stream from the RTSP URL # webcam:
cap = cv2.VideoCapture(rtsp_url) cap = cv2.VideoCapture(0)
while True: while True:
ret, frame = cap.read() ret, frame = cap.read()
@ -32,7 +32,7 @@ while True:
frame = cv2.aruco.drawDetectedMarkers(frame, corners, ids) frame = cv2.aruco.drawDetectedMarkers(frame, corners, ids)
# Scale frame in half # Scale frame in half
frame = cv2.resize(frame, (0, 0), fx=0.5, fy=0.5) # frame = cv2.resize(frame, (0, 0), fx=0.5, fy=0.5)
# Display the frame # Display the frame
cv2.imshow("Frame", frame) cv2.imshow("Frame", frame)
@ -44,3 +44,7 @@ while True:
# Release the capture object and close all OpenCV windows # Release the capture object and close all OpenCV windows
cap.release() cap.release()
cv2.destroyAllWindows() cv2.destroyAllWindows()
if __name__ == '__main__':
pass

2
bot.py
View file

@ -95,7 +95,7 @@ class Bot:
# Iterate until we find all 4 aruco markers or timeout # Iterate until we find all 4 aruco markers or timeout
aruco_corners = {} aruco_corners = {}
annotated_image = None annotated_image = None
aruco_dict = aruco.getPredefinedDictionary(aruco.DICT_6X6_250) aruco_dict = aruco.getPredefinedDictionary(aruco.DICT_4X4_50)
aruco_params = aruco.DetectorParameters() aruco_params = aruco.DetectorParameters()
t0 = time.time() t0 = time.time()