yolov5训练后续:pt文件转engine流程

第一步:在主文件夹的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)。

  • 4
    点赞
  • 16
    收藏
    觉得还不错? 一键收藏
  • 2
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值