![](https://img-blog.csdnimg.cn/20201014180756757.png?x-oss-process=image/resize,m_fixed,h_64,w_64)
pytorch
文章平均质量分 92
深度学习
小尼ndq
炼丹师 C
展开
-
关于Yet-Another-EfficientDet-Pytorch 的输出,loss分析
链接: https://github.com/zylo117/Yet-Another-EfficientDet-Pytorch.features, regressions, classifications, anchors = model(imgs)第一个输出features,代入第二,三,四个输出相关的函数中,相当于第二阶网络,得到第二,三,四个输出regressions= self.regressor(features)classifications= self.classifer(feat原创 2020-07-07 13:54:21 · 844 阅读 · 0 评论 -
torch.max / torch.full_like / torch.ones_like / torch.lt / torch.ge
torch.maxa = np.random.rand(5,2)a = torch.tensor(a)a_max,a_argmax = torch.max(a, dim=1)a : tensor([[0.2805, 0.4071],[0.4589, 0.8937],[0.5888, 0.5372],[0.5745, 0.8074],[0.6305, 0.4937]], dtype=torch.float64)a_max :tensor([0.4071, 0.8937, 0.5888, 0原创 2020-06-30 15:03:34 · 873 阅读 · 0 评论 -
关于pytorch 转 onnx ,TracerWarning解决
已遇到情况举例:1 不能有if…else如:if nx == na: x = x + a因为pytorch过于灵活,个人感觉 onnx计算图,需要固定的计算公式,逻辑判断语句会出问题,如上面语句,我debug了下,在我需要的yolov3情况下,发现nx==na一直成立,所以直接去掉“ if nx == na: ”,就OK了2 原地修改(计算)如:p[:,0:2] = torch.sigmoid(p[:,0:2])下面是问题描述:TracerWarning: There a原创 2020-06-19 14:21:00 · 3172 阅读 · 2 评论 -
Pytorch版 yolov3 (ultralytics) loss计算变换流程
Pytorch版 yolov3 (ultralytics),关于loss(box,obj,class)转换流程已全部理清。链接: https://github.com/ultralytics/yolov3.一. 输出各维度含义已yolov3-tiny(2尺度)举例(以下数组全为尺寸大小,不是数值)生成 pi [4,3,16,16,7] / pi [4,3,32,32,7] 输出(网络输出),‘4’: batch_size‘3’ : 3个anchor‘16’/’32’: [512,512]图像缩原创 2020-06-07 23:04:49 · 1891 阅读 · 6 评论