一、pth->wts（Python）

1. 将pth文件移动到yolov5-master文件夹中
2. 将yolov5_tensorrt中的gen_wts移动到yolov5-master文件夹中
3. 执行以下语句

python gen_wts.py -w yolov5s.pt -o yolov5s.wts

二、wts->engine（C++）

1. 将wts文件移动到yolov5_tensorrt中
2. 更改yololayer.h中的参数

    static constexpr int CLASS_NUM = 80;
    static constexpr int INPUT_H = 240;  // yolov5's input height and width must be divisible by 32.
    static constexpr int INPUT_W = 320;

1. 执行以下命令

cd build
cmake ..
make -j10
cd ..
./yolov5 -s model.wts model.engine s