克隆下载yolov5项目

• 准备yolov5项目, 从仓库下载5 5.0版本

  git clone -b v5.0 https://github.com/ultralytics/yolov5.git
  

• 下载权重地址

  https://github.com/ultralytics/yolov5/releases/tag/v5.0
  

• 自行安装必要的库

  pip3 install -r requirements.txt
  

准备tensorrtx项目

下载后可见该项目文件下有个readme.md，参考可快速上手。

• 克隆仓库

  git clone -b yolov5-v5.0 https://github.com/wang-xinyu/tensorrtx.git
  

• 将yolov5s转换成yolov5s.wts文件

  cd tensorrtx
  cp yolov5/gen_wts.py ../yolov5
  cd ../yolov5
  pyhton3 gen_wts.py -w yolov5s.pt -o yolov5s.wts
  

  然而很伤心的是报错了，python3 gen_wts.py -w yolov5s.pt -o yolov5s.wts段错误 (核心已转储)，弄了也没解决（希望有大佬帮忙解答)。
  

  然后就在服务器上将将yolov5s转换成yolov5s.wts文件，最后将生成的.wts放到Jetson nano上的tensorrtx/yolov5/下
  

开始编译代码

• 在tensorrtx/yolov5/中新建文件夹build

  mkdir build
  cd build
  cmake ..
  make
  

生成yolov5s.engine文件

• 在buiid下执行如下代码：

  sudo ./yolov5 -s yolov5s.wts yolov5s.engine s
  

• 验证

  sudo ./yolov5 -d yolov5s.engine ../samples