fix yolo world node rosparam

ojh6404 · Aug 3, 2024 · 70dd01c · 70dd01c
1 parent 655f77e
commit 70dd01c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/deep_vision_ros/node_scripts/yolo_node.py b/deep_vision_ros/node_scripts/yolo_node.py
@@ -21,7 +21,7 @@ def __init__(self):
         super(YOLONode, self).__init__()
         self.get_mask = rospy.get_param("~get_mask", False)
         self.yolo_config = YOLOConfig.from_args(
-            model_type=rospy.get_param("~model_type", "yolov8x_worldv2"),
+            model_type=rospy.get_param("~yolo_model_type", "yolov8x_worldv2"),
             device=rospy.get_param("~device", "cuda:0"),
         )
         self.model = YOLOModel(self.yolo_config)