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,

注意： yolov5版本要与tensorrt版本一致