First ROS package working

tue-robotics · May 21, 2024 · a94b931 · a94b931
1 parent a5a8407
commit a94b931
Show file tree

Hide file tree

Showing 4 changed files with 3 additions and 52 deletions.
diff --git a/people_tracking_v2/CMakeLists.txt b/people_tracking_v2/CMakeLists.txt
@@ -11,7 +11,7 @@ catkin_package()
 install(PROGRAMS
   scripts/face_recognition_node.py
   scripts/pose_estimation_node.py
-  scripts/yolo2.py
+  scripts/yolo.py
   DESTINATION ${CATKIN_PACKAGE_BIN_DESTINATION}
 )
 

diff --git a/people_tracking_v2/scripts/yolo2.py → people_tracking_v2/scripts/yolo.py b/people_tracking_v2/scripts/yolo2.py → people_tracking_v2/scripts/yolo.py
@@ -12,7 +12,7 @@ def __init__(self):
         self.bridge = CvBridge()
         self.model = YOLO("yolov8l.pt")
 
-        self.image_sub = rospy.Subscriber("/cv_camera/image_raw", Image, self.image_callback)
+        self.image_sub = rospy.Subscriber("/Webcam/image_raw", Image, self.image_callback)
         self.detection_pub = rospy.Publisher("/yolo_detections", String, queue_size=10)
 
     def image_callback(self, data):

diff --git a/people_tracking_v2/src/people_tracking/launchers/yolo_launcher.launch b/people_tracking_v2/src/people_tracking/launchers/yolo_launcher.launch
@@ -20,7 +20,7 @@
 
   <node
     pkg="people_tracking_v2"
-    type="yolo2.py"
+    type="yolo.py"
     name="People_detection"
     output="screen"
   />

diff --git a/people_tracking_v2/src/people_tracking/yolo.py b/people_tracking_v2/src/people_tracking/yolo.py