diff --git a/src/active_bo_msgs/msg/ImageFeedback.msg b/src/active_bo_msgs/msg/ImageFeedback.msg index 1c6a82a..cc8ad92 100644 --- a/src/active_bo_msgs/msg/ImageFeedback.msg +++ b/src/active_bo_msgs/msg/ImageFeedback.msg @@ -1,5 +1,5 @@ int16 height int16 width -uint8[] red -uint8[] green -uint8[] blue \ No newline at end of file +uint16[] red +uint16[] green +uint16[] blue \ No newline at end of file diff --git a/src/active_bo_ros/active_bo_ros/rl_service.py b/src/active_bo_ros/active_bo_ros/rl_service.py index 6f95527..bf6d7b8 100644 --- a/src/active_bo_ros/active_bo_ros/rl_service.py +++ b/src/active_bo_ros/active_bo_ros/rl_service.py @@ -40,8 +40,15 @@ class RLService(Node): rgb_array = output[5] rgb_shape = rgb_array.shape + red = rgb_array[:, :, 0].flatten().tolist() + green = rgb_array[:, :, 1].flatten().tolist() + blue = rgb_array[:, :, 2].flatten().tolist() + feedback_msg.height = rgb_shape[0] feedback_msg.width = rgb_shape[1] + feedback_msg.red = red + feedback_msg.green = green + feedback_msg.blue = blue self.publisher.publish(feedback_msg)