diff --git a/bitbots_vision/bitbots_vision/vision_modules/yoeo/utils.py b/bitbots_vision/bitbots_vision/vision_modules/yoeo/utils.py index 15a11301d..22c8ae311 100644 --- a/bitbots_vision/bitbots_vision/vision_modules/yoeo/utils.py +++ b/bitbots_vision/bitbots_vision/vision_modules/yoeo/utils.py @@ -5,7 +5,7 @@ import cv2 import numpy as np import rclpy -from jaxtyping import UInt8 +from jaxtyping import Float64, UInt8 logger = rclpy.logging.get_logger("yoeo_handler_utils") @@ -158,10 +158,10 @@ def get_info(self) -> ImagePreProcessorData: padding_bottom=self._padding_bottom, padding_left=self._padding_left, padding_right=self._padding_right, - max_dim=np.max(self._image_dimensions_HW), + max_dim=int(np.max(self._image_dimensions_HW)), ) - def process(self, image: UInt8[np.ndarray, "h w 3"]) -> UInt8[np.ndarray, "3 network_input_h network_input_w"]: + def process(self, image: UInt8[np.ndarray, "h w 3"]) -> Float64[np.ndarray, "3 network_input_h network_input_w"]: self._image_dimensions_HW = image.shape[:2] # type: ignore[assignment] self._calculate_paddings() diff --git a/sync_includes_wolfgang_nuc.yaml b/sync_includes_wolfgang_nuc.yaml index 7c2e24c43..9f8005637 100644 --- a/sync_includes_wolfgang_nuc.yaml +++ b/sync_includes_wolfgang_nuc.yaml @@ -15,7 +15,6 @@ include: - bitbots_parameter_blackboard - bitbots_robot_description - bitbots_teleop - - bitbots_tf_buffer - bitbots_tts - bitbots_utils - system_monitor @@ -51,6 +50,7 @@ include: - bio_ik - bio_ik_msgs - biped_interfaces + - bitbots_tf_buffer - dynamic_stack_decider - dynamixel-workbench - DynamixelSDK