diff --git a/bot.py b/bot.py index c1cc45e..0ce4c65 100644 --- a/bot.py +++ b/bot.py @@ -128,6 +128,10 @@ class Bot: print( "Timeout waiting for ArUco markers, returning only original image" ) + + ids, corners = zip(*filter(lambda x: x[1] is not None, enumerate(aruco_corners))) + aruco.drawDetectedMarkers(pretty_image, corners, ids) + self.tapo.setPrivacyMode(privacy_mode) self.tapo.setDayNightMode("auto") return [pretty_image]