第一步:在主文件夹的yolov5文件夹下用终端打开,输入“python3 gen_wts.py -w 文件名.pt -o 文件名.wts”
若第一步中遇到“非法指令(核心已转储)”情况时,
输入“export OPENBLAS_CORETYPE=ARMV8”
第二步:将转换成wts的文件复制到“桌面--esiwin--tensorrtx-yolov5-v6.0--yolov5--build”路径下,删除其他所有该路径内文件
第三步:该路径下用终端打开,输入“cmake ..”
第四步:接着输入“make”
第五步:最后输入“sudo ./yolov5 -s 文件名.wts 文件名.engine n”此处user的密码为:User@1949!
注意事项:
1、若训练集的标签数量发生改变,需将“主文件夹--下载--tensorrtx-yolov5-v6.0--yolov5”路径下的yololayer.h文件中第20行代码“static constexpr int CLASS NUM = X”中的X改成对应标签数。
2、若有某个标签物体在识别过程中持续不识别,原因有可能是模型训练不准确(重训练),也有可能是模型置信度过高(在yolov5.cpp文件中修改conf_thresh的值至0.1~0.2)。