- 博客(3)
- 收藏
- 关注
RSS订阅原创 解读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
1580
原创 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
5640
1
原创 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
6202
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人