Niko/ActiveBOROS
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Image data is not correct

Browse Source
This commit is contained in:
Niko Feith 2023-03-02 15:20:56 +01:00
parent 45f789f9b7
commit ba0c730bd7
2 changed files with 10 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
src/active_bo_msgs/msg/ImageFeedback.msg
View File

@ -1,5 +1,5 @@
int16 height int16 height
int16 width int16 width
uint8[] red uint16[] red
uint8[] green uint16[] green
uint8[] blue uint16[] blue

7
src/active_bo_ros/active_bo_ros/rl_service.py
View File

@ -40,8 +40,15 @@ class RLService(Node):
rgb_array = output[5] rgb_array = output[5]
rgb_shape = rgb_array.shape 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.height = rgb_shape[0]
feedback_msg.width = rgb_shape[1] feedback_msg.width = rgb_shape[1]
feedback_msg.red = red
feedback_msg.green = green
feedback_msg.blue = blue
self.publisher.publish(feedback_msg) self.publisher.publish(feedback_msg)