Error Code 3: Internal Error (Parameter check failed at: optimizer/api/network.cpp::addScale::616,

基于 https://blog.csdn.net/qq_42362891/article/details/116107108
YOLOv5 部署的 评论区 进一步degug

tensorrtx目前支持很多种网络(vgg,googlenet,yolov5等)的转换,这里我们主要关注yolov5

注意:由于不同版本的Yolov5网络架构不同,所以这里Yolov5和tensorrtx也有版本对应

首先看一下tensorrtx的Tags,会发现他目前最高只支持到 yolov5-v5.0

请添加图片描述

请添加图片描述
而yolov5最新版本是6.1,所以我们需要退而求其次,下载yolov5-v5.0

请添加图片描述

注意:在releases中下载的,可能没有注意到 yolov5的版本,直接下载了最新的。

请添加图片描述

在继续执行 即可

https://github.com/wang-xinyu/tensorrtx/tree/yolov5-v5.0/yolov5

python gen_wts.py =2 yolov5s.pt -o yolov5s.wts
  • 5
    点赞
  • 5
    收藏
    觉得还不错? 一键收藏
  • 2
    评论
评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值