您好,欢迎来到化拓教育网。
搜索
您的当前位置:首页关于pytorch中的[...,n]

关于pytorch中的[...,n]

来源:化拓教育网

今天在看yolov4代码时看到的疑问,自己便实验了一下

以prediction维度为batchsize*3*13*13*85为例,prediction[...,0]即取第一个轴上的所有数据,即维度为batchsize*3*13*13。

yolov4中85分别代表先验框的x,y坐标,宽,高,是否包括物体,以及80个类别概率

prediction[...,0]对应x的信息

prediction[...,1]对应y的信息

prediction[...,2]对应宽的信息

prediction[...,3]对应高的信息

因篇幅问题不能全部显示,请点此查看更多更全内容

Copyright © 2019- huatuo9.cn 版权所有 赣ICP备2023008801号-1

违法及侵权请联系:TEL:199 18 7713 E-MAIL:2724546146@qq.com

本站由北京市万商天勤律师事务所王兴未律师提供法律服务