Add YOLO-World Model Support in tracker_node.py - Githubissues

Alpaca-zip / ultralytics_ros

ROS/ROS 2 package for Ultralytics YOLOv8 real-time object detection and segmentation. https://github.com/ultralytics/ultralytics

GNU Affero General Public License v3.0

205 stars 40 forks source link

Add YOLO-World Model Support in tracker_node.py #74

Open yukiharada1228 opened 5 months ago

yukiharada1228 commented 5 months ago

PR Type

[x] Feature

Overview

Added support for YOLO-World model in tracker_node.py.

Detail

Introduced use_world variable to check if the YOLO model name contains "world".
When using YOLO-World model, if self.classes is defined, it sets these classes in the model and then resets self.classes to None.
Otherwise, retains existing self.model.fuse() invocation.

Test

[x] I have tested this change with rosbag