Jetson Nano Yolov5+tensorrt

该文章介绍了如何在JetsonNano上使用Alex-Beh的tensorrtx项目,结合YOLOv5模型和TensorRT进行对象检测加速。步骤包括构建、编译项目,修改配置以适应不同的模型(如yolov5s),调整锚框颜色,并通过USB摄像头获取视频流进行实时推理。此外,提到了特定的PyTorch版本要求以及在ROS环境中生成和运行engine的步骤。
摘要由CSDN通过智能技术生成

Alex-Beh/tensorrtx

Yolov5 TensorRT inteference with usb camera

mkdir build
cd build 
cmake ..
make
cd build 
./video_inteference_yolov5
# 里面engine 类别 video修改自己,yolov5s,修改完还要make一下
# 162行 修改锚框颜色 BGR cv::rectangle(pr_img, r, cv::Scalar(0x27, 0xC1, 0x36), 2);
cd /dev
ls | grep video

#see video0 detail info
udevadm info --query=all --name=/dev/video0

v5.0 - YOLOv5

参考alex配准tensorrtx

Jetson Nano Yolov5+tensorrt设置说明

Jetson nano + yolov5 + TensorRT加速+调用usb摄像头

Jetson Nano部署YOLOv5与Tensorrtx加速

自己电脑跑yolov5
环境 yolo
cuda10.2 torch 1.8.0 torchvision 0.9.0 一定要版本对应

conda activate yolo
pip install torch
pip install torchvision

pytorch版本地址
在这里插入图片描述

在这里插入图片描述
roscore
rosrun camera_yolo video_in执行
在tensorrtx中生成engine
如果修改,在camera_ws中执行catkin_make

然后rviz

python3 gen_wts.py new_best_test3.py

  • 1
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值