博客

Jetson xavier NX Yolov5

  • Post author:
  • Post category:其他




1. 下载yolov5

git clone https://github.com/ultralytics/yolov5.git
cd yolov5-master



2. 安装依赖

pip3 install -r requirements.txt



3. 卸载torch & torchvision

pip3 uninstall torch torchvision



4. 安装jetson xavier 相应版本的torch torchvision


官方下载



5. 验证cuda

import torch
torch.cuda.is_available()
#True



6. 运行yolov5

python3 detect.py --source data/images/bus.jpg --weights yolov5s.pt

在这里插入图片描述

可以看到在使用yolov5s且使用cuda的情况下640×640分辨率的图片识别速度在50ms左右


版权声明：本文为weixin_44604730原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。