UBCAgroBot / ROS

1 stars 1 forks source link

Feature/25 mvp integration #32

Open vangeliq opened 4 days ago

vangeliq commented 4 days 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