YOLOv5 ROS 项目使用教程

项目目录结构及介绍

YOLOv5 ROS 项目的目录结构如下：

yolov5_ROS/
├── launch/
│   ├── yolov5s_simple.launch.py
│   └── ...
├── src/
│   ├── yolov5/
│   │   ├── data/
│   │   ├── models/
│   │   └── ...
│   └── ...
├── CMakeLists.txt
├── package.xml
└── README.md

目录结构介绍

• launch/: 包含项目的启动文件，如 yolov5s_simple.launch.py。
• src/: 包含项目的源代码，其中 yolov5/ 目录下是 YOLOv5 的核心代码。
• CMakeLists.txt: 项目的 CMake 配置文件。
• package.xml: 项目的 ROS 包描述文件。
• README.md: 项目的基本介绍和使用说明。

项目启动文件介绍

yolov5s_simple.launch.py

该启动文件用于启动 YOLOv5 的推理服务。以下是文件的主要内容和功能：

import os
from ament_index_python.packages import get_package_share_directory
from launch import LaunchDescription
from launch_ros.actions import Node

def generate_launch_description():
    ld = LaunchDescription()

    config = os.path.join(
        get_package_share_directory('yolov5_ros'),
        'config',
        'yolov5s.yaml'
    )

    node = Node(
        package='yolov5_ros',
        executable='yolov5_node',
        name='yolov5_node',
        parameters=[config]
    )

    ld.add_action(node)
    return ld

功能介绍

• 导入必要的模块: 导入 os, ament_index_python.packages, launch, 和 launch_ros.actions。
• 获取配置文件路径: 使用 get_package_share_directory 获取配置文件的路径。
• 定义节点: 创建一个 ROS2 节点，指定包名、可执行文件名、节点名和参数文件。
• 添加节点到启动描述: 将节点添加到 LaunchDescription 中并返回。

项目配置文件介绍

yolov5s.yaml

该配置文件包含 YOLOv5 推理所需的各种参数。以下是文件的主要内容和功能：

weights: 'yolov5s.pt'
data: 'coco128.yaml'
img_size: 640
conf_thres: 0.25
iou_thres: 0.45
device: 'cpu'

参数介绍

• weights: 指定使用的 YOLOv5 模型权重文件。
• data: 指定数据集配置文件。
• img_size: 输入图像的尺寸。
• conf_thres: 置信度阈值。
• iou_thres: IOU 阈值。
• device: 指定使用的设备（如 'cpu' 或 'cuda'）。

通过以上配置文件，可以灵活地调整 YOLOv5 的推理参数，以适应不同的应用场景。