810f412821
- hailo infer node subscribes to an image topic - preprocess image via PIL to the desired image parameters (hard- coded currently for yolov5m) - processes inference on hailo chip - post process and prints bounding boxes into terminal
33 lines
1.0 KiB
Python
33 lines
1.0 KiB
Python
import rclpy
|
|
from rclpy.node import Node
|
|
from std_msgs.msg import String
|
|
from cv_bridge import CvBridge
|
|
from sensor_msgs.msg import Image
|
|
import cv2
|
|
import numpy as np
|
|
|
|
class MinimalPublisher(Node):
|
|
def __init__(self):
|
|
super().__init__('minimal_publisher')
|
|
self.publisher_ = self.create_publisher(Image, '/camera/color/image_raw', 10)
|
|
timer_period = 0.5 # seconds
|
|
self.timer = self.create_timer(timer_period, self.timer_callback)
|
|
self.i = 0
|
|
self.im_list = []
|
|
self.cv_image = cv2.imread('data/DJI_0001_0001.jpg') ### an RGB image
|
|
self.bridge = CvBridge()
|
|
|
|
def timer_callback(self):
|
|
self.publisher_.publish(self.bridge.cv2_to_imgmsg(np.array(self.cv_image), "bgr8"))
|
|
self.get_logger().info('Publishing an image')
|
|
|
|
def main(args=None):
|
|
rclpy.init(args=args)
|
|
minimal_publisher = MinimalPublisher()
|
|
rclpy.spin(minimal_publisher)
|
|
minimal_publisher.destroy_node()
|
|
rclpy.shutdown()
|
|
|
|
if __name__ == '__main__':
|
|
main()
|