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

Fix typo

Browse source
This commit is contained in:
Davide Depau 2024-02-22 01:10:33 +01:00
parent ba3c9369e1
commit 92240741cc
1 changed files with 8 additions and 6 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

14
bot.py
View file

@ -12,6 +12,7 @@ import cv2
import numpy as np import numpy as np
import telegram import telegram
from aiohttp import BasicAuth from aiohttp import BasicAuth
from cv2 import aruco
from pytapo import Tapo from pytapo import Tapo
from telegram import Update, Message from telegram import Update, Message
from telegram.ext import Updater from telegram.ext import Updater
@ -94,8 +95,8 @@ 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 = a.getPredefinedDictionary(a.DICT_6X6_250) aruco_dict = aruco.getPredefinedDictionary(aruco.DICT_6X6_250)
aruco_params = a.DetectorParameters() aruco_params = aruco.DetectorParameters()
t0 = time.time() t0 = time.time()
print("Taking image with ArUco markers...") print("Taking image with ArUco markers...")
@ -121,11 +122,12 @@ class Bot:
continue continue
# Detect the markers # Detect the markers
corners, ids, rejected = a.detectMarkers( corners, ids, rejected = aruco.detectMarkers(
annotated_image, aruco_dict, parameters=aruco_params annotated_image, aruco_dict, parameters=aruco_params
) )
for corner, i in zip(corners, ids): if corners is not None and ids is not None:
aruco_corners[i[0]] = corner for corner, i in zip(corners, ids):
aruco_corners[i[0]] = corner
assert annotated_image is not None assert annotated_image is not None
del vcap del vcap
@ -138,7 +140,7 @@ class Bot:
corners = [aruco_corners[i] for i in range(1, 5)] corners = [aruco_corners[i] for i in range(1, 5)]
ids = np.array([[i] for i in range(1, 5)]) ids = np.array([[i] for i in range(1, 5)])
a.drawDetectedMarkers(annotated_image, corners, ids) aruco.drawDetectedMarkers(annotated_image, corners, ids)
# Annotate the image with the detected markers and apply the perspective transform to the pretty image # Annotate the image with the detected markers and apply the perspective transform to the pretty image