1. 从github上git下序列化程序
2. 修改gen_wts.py 文件,best.pt是要训练好的yolov5模型,wts是要生成的权重文件。
3.修改yololayer.h文件,只需要修改类数即可.
cd build
cmake ..
make -j6
sudo ./yolov5 -s yolov5s.wts yolov5s.engine s
最后一个s 表示训练的模型是smlx,
cd build
cmake ..
make -j6
sudo ./yolov5 -s yolov5s.wts yolov5s.engine s
最后一个s 表示训练的模型是smlx,