- 博客(2)
- 收藏
- 关注
RSS订阅原创 numpy和pytorch里的通道转换问题
在学习深度学习的时候，在将得到的特征tenso想要显示的时候，在代码里总是会看到transpose（1，2，0），这是因为在numpy中，图像的shape一般是（H, W,C)，即图像的长宽是在前面，而通道数则是在最后一维，而在pytorch中呢，图像的shape一般是（C,H,W),通道数则是在第0维，因此，在tensor转换为numpy并进行显示时，需要将格式转换为numpy的格式，即（1，2，0），如果是numpy转换为tensor的话，那就是transpose（2，0，1）。...
2020-07-04 09:50:51
2528
1
原创 在python中创建类时，self的作用
在深度学习中，有时候创建模型总是要先创建一个class(类)，而在初始化类的时候__init__的第一个参数是self，那么这个self有什么作用呢？对于一个类来说，在创建实例的时候，如果需要添加比较多的参数，那么self的作用就展现出来了，例如，在构建一个生成模型时，在定义init方法的时候，可以将输入噪声的维度、卷积维度以及输出的size等属性添加进去，如下所示：class Generator(nn.Module): def __init__(self, noise_dim=1...
2020-06-22 22:45:10
849
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人