UBCAgroBot / ROS

Central codebase for UBC Agrobot's System Architecture team, including packages for ROS2, GPU-accelerated processing and ML optimization
0 stars 1 forks source link

Feature/25 mvp integration #32

Closed vangeliq closed 2 weeks ago

vangeliq commented 1 month ago

Main changes:

Picture Node: Rewrote so this node attempts to open a folder and read the image files in the folder. This node now publishes the images as a ros2 image message called /input_image

Inference Node Created an inference node that takes in the /input_image topic, and publishes the output image and bounding box of a yolo model. This node calls on a util class to do the preprocessing, inference and post processing.

Util Class Added a utility class that uses the ultralytics API to do preprocessing, inference and post processing

Other changes