超维空间S2无人机使用说明书——32、使用yolov7进行目标识别

引言:为了提高yolo识别的质量,提高了yolo的版本,改用yolov7进行物体识别,同时系统兼容了低版本的yolo,包括基于C++的yolov3和yolov4,也有更高版本的yolov8。

简介,为了提高识别速度,系统采用了GPU进行加速,在使用7W功率的情况,大概可以稳定在20FPS,满功率情况下可以适当提高。

硬件:D435摄像头,Jetson orin nano 8G

环境:ubuntu20.04,ros-noetic, yolov7

步骤一: 启动摄像头,获取摄像头发布的图像话题

javascript 复制代码
roslaunch realsense2_camera rs_camera.launch  

没有出现红色报错,出现如下界面,表明摄像头启动成功

步骤二:启动yolov7识别节点

javascript 复制代码
roslaunch yolov7 yolov7.launch

launch文件如下,参数device设置为cuda,因为实际使用GPU加速,不是CPU跑,另外参数img_topic是订阅的节点话题,一定要与摄像头发布的实际话题名称对应上。其他参数可以根据实际情况进行调整即可

javascript 复制代码
<?xml version="1.0"?>
<launch>
    <node pkg="yolov7" type="YoloV7.py" name="yolov7">
        <!-- Path to your weight -->
        <param name="weights_path" type="str" value="/home/cwkj/cwkj_ws/src/ros-yolov7/cfg/weights/yolov7-tiny.pt"/>
        <!-- Path to a class_labels.txt file, if you leave it empty then no class labels are visualized.-->
        <param name="classes_path" type="str" value="//home/cwkj/cwkj_ws/src/ros-yolov7/cfg/config/coco.txt" />
        <!-- Input image topic name to subscribe to -->
        <param name="img_topic" type="str" value="/camera/color/image_raw" />
        <!-- [optional]  Confidence threshold (default=0.25) -->
        <param name="conf_thresh" type="double" value="0.20" />
        <!-- [optional]  Intersection over union threshold (default=0.45) -->
        <param name="iou_thresh" type="double" value="0.45" />
        <!-- [optional]  Queue size for publishing (default=3) -->
        <param name="queue_size" type="int" value="1" />
        <!-- [optional] Image size to which to resize each input image before feeding into the network (the final output is rescaled to the original image size) (default=640) -->
        <param name="img_size" type="int" value="640" />
        <!-- [optional] Flag whether to also publish image with the visualized detections (default=false) -->
        <param name="visualize" type="bool" value="true" />
        <!-- [optional] Torch device 'cuda' or 'cpu' (default="cuda") -->
        <param name="device" type="str" value="cuda" />
        <!-- [optional] Node frequency (default=10) -->
        <param name="frequency" type="int" value="10" />
    </node>
</launch>

出现如下界面表示yolov7启动成功

步骤三:打开rqt工具,查看识别效果

javascript 复制代码
rqt_image_view 

等待出现如下界面后,选择yolov7/visualize/image查看yolov7识别效果

总结:可以根据实际需要选择和是的yolo版本进行物体识别。系统中配置的基于darknet的yolov3和yolov4也有着非常好的识别效果。

相关推荐
小O的算法实验室17 小时前
IEEE TASE,基于异构注意力的图卷积网络求解非对称取送货问题
无人机
XBWRJ17 小时前
下沉市场观察:《无人机装调维修师》与县域维保能力的构建
无人机
AI吃大瓜19 小时前
人脸检测和行人检测4:Android实现YOLOv8 YOLO11 YOLO26人脸检测和人体检测(含源码,可实时检测)
android·yolo·人脸检测·人体检测·行人检测·yolo26
haishikeji696_19 小时前
低空一网统管怎么做?无人机智慧巡查平台项目选型指南
无人机·低空经济·无人机管理系统·无人机飞控平台·一网统飞·ai自动化巡检
Zentceh19 小时前
AI-ISP vs 传统ISP全面对比
人工智能·深度学习·计算机视觉·车载系统·transformer·无人机·智能硬件
光影跳跳糖1 天前
某矿山无人机航测案例,从储量管理到边坡监测
无人机·航测
深圳安车昇辉检测机构1 天前
DO-160G 爆炸大气试验:机载设备可燃环境安全验证要点
无人机·安全性测试
workflower1 天前
AI 转型正从工具部署转向生产关系重构
人工智能·安全·机器学习·机器人·无人机
飞猫的边缘AI1 天前
边缘AI-13:从YOLOv1到YOLO26:目标检测是怎么进化的
人工智能·yolo·目标检测·ai算法·边缘ai·yolo26