当前位置:   article > 正文

nvidia jetson agx-----yolov5_Docker-Quickstart_yolov5s.pt下载

yolov5s.pt下载

https://developer.nvidia.com/compute/cudnn/secure/8.6.0/local_installers/11.8/cudnn-local-repo-ubuntu2004-8.6.0.163_1.0-1_arm64.deb
Docker-Quickstart
https://github.com/ultralytics/yolov5/wiki/Docker-Quickst

1. Install Docker and Nvidia-Docker

第一步在刷机时就已经安装了Nvidia-Docker,所以直接跳过。

2. Pull Image

jetson 是arm架构,所以后面加上-arm64执行

//sudo docker pull ultralytics/yolov5:latest
sudo docker pull ultralytics/yolov5:latest-arm64  
  • 1
  • 2

3. Run Container

1、下载yolov5s.pt
下载地址https://github.com/ultralytics/yolov5/releases
在这里插入图片描述2、下载测试图片,新建一个文件夹,将yolov5s.pt和测试图片放到文件夹中。
在这里插入图片描述

Run a container with local file access (like COCO training data in /datasets) using -v:
换成刚才建的文件的地址来执行:

//sudo docker run --ipc=host -it -v "$(pwd)"/datasets:/usr/src/datasets ultralytics/yolov5:latest
sudo nvidia-docker run --ipc=host -it --gpus all -v /home/xmm17/Downloads/yolo5s:/usr/src/yolo5s ultralytics/yolov5:latest-arm64
  • 1
  • 2

在这里插入图片描述

4. Run YOLOv5 rocket

切换文件夹,执行测试

cd /usr/src/app
python3 detect.py
python3 detect.py --weights ../yolo5s/yolov5s.pt --source ../yolo5s/12.png 
  • 1
  • 2
  • 3

测试结果
在这里插入图片描述
在这里插入图片描述

总结

在docker上测试yolov5,要方便很多,相比于前面的部署yolov5的环境过程来说。。。
多学会利用现有环境资源。

声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/IT小白/article/detail/291050
推荐阅读
相关标签
  

闽ICP备14008679号