Skip to content

Commit 0728235

Browse files
committed
change yolo dependencies
1 parent 21bb921 commit 0728235

15 files changed

+28
-28
lines changed

bt_nodes/bt_test/bt_xml/set_perception_model_test.xml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -2,7 +2,7 @@
22
<root main_tree_to_execute="BehaviorTree">
33
<!-- ////////// -->
44
<BehaviorTree ID="BehaviorTree">
5-
<Action ID="SetPerceptionModel" model_name="yolov8m-pose.pt"/>
5+
<Action ID="SetPerceptionModel" model_name="yolom-pose.pt"/>
66
</BehaviorTree>
77
<!-- ////////// -->
88
<TreeNodesModel>

bt_nodes/configuration/include/configuration/set_perception_model.hpp

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -23,14 +23,14 @@
2323
#include "configuration/bt_service_node.hpp"
2424
#include "rclcpp/rclcpp.hpp"
2525
#include "rclcpp_cascade_lifecycle/rclcpp_cascade_lifecycle.hpp"
26-
#include "yolov8_msgs/srv/change_model.hpp"
26+
#include "yolo_msgs/srv/change_model.hpp"
2727

2828

2929
namespace configuration
3030
{
3131

3232
class SetPerceptionModel
33-
: public configuration::BtServiceNode<yolov8_msgs::srv::ChangeModel,
33+
: public configuration::BtServiceNode<yolo_msgs::srv::ChangeModel,
3434
rclcpp_cascade_lifecycle::CascadeLifecycleNode>
3535
{
3636
public:

bt_nodes/configuration/src/configuration/set_perception_model.cpp

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -26,7 +26,7 @@ using namespace std::placeholders;
2626
SetPerceptionModel::SetPerceptionModel(
2727
const std::string & xml_tag_name, const std::string & action_name,
2828
const BT::NodeConfiguration & conf)
29-
: configuration::BtServiceNode<yolov8_msgs::srv::ChangeModel,
29+
: configuration::BtServiceNode<yolo_msgs::srv::ChangeModel,
3030
rclcpp_cascade_lifecycle::CascadeLifecycleNode>(
3131
xml_tag_name, action_name, conf)
3232
{

bt_nodes/perception/include/perception/ExtractObjectsFromScene.hpp

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -27,7 +27,7 @@
2727
#include "rclcpp/rclcpp.hpp"
2828
#include "rclcpp_cascade_lifecycle/rclcpp_cascade_lifecycle.hpp"
2929
#include "shape_msgs/msg/solid_primitive.hpp"
30-
#include "yolov8_msgs/msg/detection_array.hpp"
30+
#include "yolo_msgs/msg/detection_array.hpp"
3131
#include "rclcpp_cascade_lifecycle/rclcpp_cascade_lifecycle.hpp"
3232

3333
namespace perception
@@ -51,12 +51,12 @@ class ExtractObjectsFromScene : public BT::ActionNodeBase
5151
BT::OutputPort<std::string>("objects_count")});
5252
}
5353

54-
void detection_callback_(yolov8_msgs::msg::DetectionArray::UniquePtr msg);
54+
void detection_callback_(yolo_msgs::msg::DetectionArray::UniquePtr msg);
5555

5656
private:
5757
std::shared_ptr<rclcpp_cascade_lifecycle::CascadeLifecycleNode> node_;
58-
rclcpp::Subscription<yolov8_msgs::msg::DetectionArray>::SharedPtr detected_objs_sub_;
59-
yolov8_msgs::msg::DetectionArray::UniquePtr last_detected_objs_ = {nullptr};
58+
rclcpp::Subscription<yolo_msgs::msg::DetectionArray>::SharedPtr detected_objs_sub_;
59+
yolo_msgs::msg::DetectionArray::UniquePtr last_detected_objs_ = {nullptr};
6060
std::string interest_class_{""};
6161

6262
std::unique_ptr<tf2_ros::Buffer> tf_buffer_;

bt_nodes/perception/include/perception/switch_yolo_model.hpp

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -22,15 +22,15 @@
2222
#include "behaviortree_cpp_v3/bt_factory.h"
2323
#include "moveit_msgs/msg/collision_object.hpp"
2424
#include "perception/bt_service_node.hpp"
25-
#include "yolov8_msgs/srv/change_model.hpp"
25+
#include "yolo_msgs/srv/change_model.hpp"
2626
#include "rclcpp/rclcpp.hpp"
2727
#include "rclcpp_cascade_lifecycle/rclcpp_cascade_lifecycle.hpp"
2828

2929
namespace perception
3030
{
3131

3232
class SwitchYoloModel : public perception::BtServiceNode<
33-
yolov8_msgs::srv::ChangeModel,
33+
yolo_msgs::srv::ChangeModel,
3434
rclcpp_cascade_lifecycle::CascadeLifecycleNode>
3535
{
3636
public:

bt_nodes/perception/script/detection_test.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -3,7 +3,7 @@
33
import rclpy
44
from rclpy.node import Node
55
from std_msgs.msg import Header
6-
from yolov8_msgs.msg import BoundingBox2D, Detection, DetectionArray
6+
from yolo_msgs.msg import BoundingBox2D, Detection, DetectionArray
77

88
DEFAULT_NUM_OBJECTS = 5
99

bt_nodes/perception/src/perception/ExtractObjectsFromScene.cpp

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -32,7 +32,7 @@ ExtractObjectsFromScene::ExtractObjectsFromScene(
3232
{
3333
config().blackboard->get("node", node_);
3434

35-
detected_objs_sub_ = node_->create_subscription<yolov8_msgs::msg::DetectionArray>(
35+
detected_objs_sub_ = node_->create_subscription<yolo_msgs::msg::DetectionArray>(
3636
"detections_3d", 100, std::bind(&ExtractObjectsFromScene::detection_callback_, this, _1));
3737
tf_buffer_ = std::make_unique<tf2_ros::Buffer>(node_->get_clock());
3838
tf_listener_ = std::make_shared<tf2_ros::TransformListener>(*tf_buffer_);
@@ -43,7 +43,7 @@ void ExtractObjectsFromScene::halt()
4343
RCLCPP_INFO(node_->get_logger(), "ExtractObjectsFromScene halted");
4444
}
4545

46-
void ExtractObjectsFromScene::detection_callback_(yolov8_msgs::msg::DetectionArray::UniquePtr msg)
46+
void ExtractObjectsFromScene::detection_callback_(yolo_msgs::msg::DetectionArray::UniquePtr msg)
4747
{
4848
last_detected_objs_ = std::move(msg);
4949
}

bt_nodes/perception/src/perception/switch_yolo_model.cpp

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -27,7 +27,7 @@ SwitchYoloModel::SwitchYoloModel(
2727
const std::string & xml_tag_name, const std::string & action_name,
2828
const BT::NodeConfiguration & conf)
2929
: perception::BtServiceNode<
30-
yolov8_msgs::srv::ChangeModel,
30+
yolo_msgs::srv::ChangeModel,
3131
rclcpp_cascade_lifecycle::CascadeLifecycleNode>(xml_tag_name, action_name, conf)
3232
{
3333
}

robocup_bringup/launch/carry_my_luggage_dependencies.launch.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -29,7 +29,7 @@ def generate_launch_description():
2929

3030
move_group_dir = get_package_share_directory('tiago_mtc_examples')
3131
manipulation_dir = get_package_share_directory('manipulation_action_server')
32-
yolo3d_dir = get_package_share_directory('yolov8_bringup')
32+
yolo3d_dir = get_package_share_directory('yolo_bringup')
3333
navigation_dir = get_package_share_directory('navigation_system')
3434
whisper_dir = get_package_share_directory('whisper_bringup')
3535
package_dir = get_package_share_directory('robocup_bringup')
@@ -82,7 +82,7 @@ def generate_launch_description():
8282

8383
yolo3d = IncludeLaunchDescription(
8484
PythonLaunchDescriptionSource(
85-
os.path.join(yolo3d_dir, 'launch', 'yolov8_3d.launch.py')
85+
os.path.join(yolo3d_dir, 'launch', 'yolo_3d.launch.py')
8686
),
8787
launch_arguments={
8888
# 'namespace': 'perception_system',

robocup_bringup/launch/gpsr_dependencies.launch.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -29,7 +29,7 @@ def generate_launch_description():
2929

3030
move_group_dir = get_package_share_directory('tiago_mtc_examples')
3131
manipulation_dir = get_package_share_directory('manipulation_action_server')
32-
yolo3d_dir = get_package_share_directory('yolov8_bringup')
32+
yolo3d_dir = get_package_share_directory('yolo_bringup')
3333
navigation_dir = get_package_share_directory('navigation_system')
3434
package_dir = get_package_share_directory('robocup_bringup')
3535

@@ -62,7 +62,7 @@ def generate_launch_description():
6262

6363
yolo3d = IncludeLaunchDescription(
6464
PythonLaunchDescriptionSource(
65-
os.path.join(yolo3d_dir, 'launch', 'yolov8_3d.launch.py')
65+
os.path.join(yolo3d_dir, 'launch', 'yolo_3d.launch.py')
6666
),
6767
launch_arguments={
6868
# 'namespace': 'perception_system',

0 commit comments

Comments
 (0)