0
点赞
收藏
分享

微信扫一扫

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

八怪不姓丑 2022-03-11 阅读 109
ubuntu

基于 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
举报

相关推荐

0 条评论