yolov5的网络结构输入输出讲解,利用netron工具进行分析,详细解释yolov5s的输入输出信息,进行tensorrt加速解析数据,网络模型最后一块中head/prediction的叠加过程讲解

yolov5的网络结构输入输出讲解,利用netron工具进行分析

1,使用的工具

netorn工具,查看yolov5s.pt导出的onnx模型,目前netorn支持pt智能看到大的架构,看不到具体的输入输出,所以最好转为onnx后,进行查看。
如果在导出的时候有报错,注意安装onnx:请查看nvidia jetson 错误 no module named onnx

2,模型输出介绍

name: output output的type: float32[1,25200,85]

85=80分类+4个坐标+置信度 25200=80* 80* 3+40* 40* 3+20* 20* 3(不同尺寸的卷积核计算,将计算结果进行拼接),其中80指的是80个分类的置信度概率,而最后一位也是置信度,指的是检测框里存在目标的概率大小,注意两个置信度不一样。

最后一个85可能是255,Prediction中用于预测的三个特征图①19* 19* 255、②38* 38* 255、③76* 76* 255。[注:255表示80类别(1+4+80)×3=255] 其中3表示的是anchor

在这里插入图片描述

3,深入解读yolov5的输出的内容</

  • 0
    点赞
  • 5
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 0
    评论

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

鼾声鼾语

感谢您的支持鼓励!

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值