pytorch
Qian._.
这个作者很懒,什么都没留下…
展开
-
pytorch中x.view()和permute用法
pytorch中x.view()用法在pytorch中经常会看到x.view(),它表示将Tensor的维度转变为view指定的维度,有点类似于resize函数b=torch.Tensor([[[[1,2,3],[4,5,6],[7,8,9]],[[1,2,3],[4,5,6],[7,8,9]]]])print(b.size())(1, 2, 3, 3)print(b.view(b...原创 2019-03-18 17:39:28 · 6017 阅读 · 0 评论 -
pytorch中[..., 0]的用法
在看程序的时候看到了x[…, 0]的语句不是很理解,后来自己做实验略微了解,以此记录方便自己查看。b=torch.Tensor([[[[10,2],[4,5],[7,8]],[[1,2],[4,5],[7,8]]]])print(b.size())(1, 2, 3, 2)print(b[…,0])tensor([[[10., 4., 7.],[ 1., 4., 7.]]]...原创 2019-03-18 20:52:32 · 5495 阅读 · 1 评论 -
解读pytorch yolo3源码解读(1)
源码来源:https://github.com/eriklindernoren/PyTorch-YOLOv3config中存放配置文件utils存放需要调用的函数本文主要解读models.py和train.py文件以及其相关的函数。在解读源码之前先略微介绍一下yolo3的相关概念。yolo3包含卷积层(convolutional),跳过连接(short cut),yolo层,上采样层(u...原创 2019-03-22 16:55:59 · 1529 阅读 · 0 评论