发布时间:2023-05-17 15:00
最近为了找创新点,重拾pytorch,在看小土堆的网课时,报了些错误,于是整理了一下。
(1)需要引用的库不同
Conv2d需要引用torch.nn,(即import torch.nn)
conv2d是torch.nn.functional中的方法,需要引用torch.nn.functional(即 import torch.nn.functional),引用错会报NameError: name ‘conv2d’ is not defined
(2)传参方式不同
Conv2d()的传参方式如下:
in_channels代表输入图片的通道数
out_channels代表输出图片的通道数
kernal_size代表卷积核的大小,如果是nn的卷积核,则kernal_size = n,如果是ab的卷积核,则kernal_size = [a,b],在这里Conv2d()的作用是在前向网络中搭建一个卷积层,所以用户没法自定义卷积核中的每个参数,只能由pytorch自己初始化。
stride代表卷积核的步长
padding代表在图片周围补充的像素数量,如果在图片上下分别补充了x1行0,在图片的左右分别补充了x2列0,那么我们可以令padding = [x1, x2]
dilation代表卷积核之间参数的距离,如果dilation = 1,那么卷积核中各参数连续排列,如果dilation = 2,那么卷积核中各参数(在横向和纵向上)相距一个像素。(如下图所示)
其余参数一般保持默认
conv2d()的传参方式如下:
input是输入图像,必须是tensor形式,且需要通过reshape()转换为特定的格式,具体转换方式见代码。
(tensor形式可使用torchvision.transforms.ToTensor(imag)进行类型转换)
这里的weight其实是卷积核kernal,也必须是tensor形式
conv2d()执行的操作是:用我们定义好的卷积核kernal对输入图像进行卷积操作,而不是作为前向通道的一部分。
如果误将conv2d的参数传给Conv2d,例如误将conv2d(input, kernal, stride=1)写成Conv2d(input,kernal,kernel_size=3,stride=1),那么可能报错:RuntimeError: Boolean value of Tensor with more than one value is ambiguous
conv2d代码演示
程序功能:输入一个矩阵或图像(input),接着自定义一个卷积核kernal,用kernal对输入矩阵进行卷积操作
import torch.nn.functional as F
#import torch.nn as F
input = torch.tensor([[1, 2, 0, 1, 1],
[2, 2, 0, 3, 6],
[1, 2, 5, 3, 1],
[5, 3, 0, 4, 1],
[1, 2, 3, 4, 5]])
kernal = torch.tensor([[1, 2, 1],
[0, 0, 0],
[-1, -2, -1]])
input = torch.reshape(input, (1, 1, 5, 5)) #(channel数为1, ??, input的size为4*5)
kernal = torch.reshape(kernal, (1, 1, 3, 3)) #(channel数为1, ??, input的size为3*3)
#RuntimeError: shape '[1, 1, 3, 3]' is invalid for input of size 20
# 这种错误是因为reshape的第一个参数和第二个没对应起来
print(input.shape)
print(kernal.shape)
output = F.conv2d(input, kernal, stride=1) #之前使用Conv2d,一直不好使。后来发现要用conv2d,又报错找不到conv2d,但是conv2d下还不报错,以为是版本问题,
#最后发现我在一开始引用了torch.nn ,改为import torch.nn.functional 问题解决。
#output2 = F.Conv2d(input, kernal, kernel_size=3,stride=1) #要import torch.nn 才能使用Conv2d,但是使用方法是不同的
print(output)
Conv2d代码演示
程序功能:搭建一个只有单层卷积层的前向传播网络,卷积核由pytorch随机初始化,卷积核大小为3
#搭建一个卷积层
import torch
import torchvision
from tensorboardX import SummaryWriter
from torch import nn
from torch.nn import Conv2d
from torch.utils.data import DataLoader
from torchvision.transforms import ToTensor
dataset = torchvision.datasets.CIFAR10("./dataset", train=False, transform=torchvision.transforms.ToTensor(),download=True) #获取测试集
dataloader = DataLoader(dataset, batch_size=64) #以64张为一组,抽取数据
class single_conv(nn.Module):
def __init__(self):
super().__init__()
self.conv1 = Conv2d(in_channels=3, out_channels=6, kernel_size=3, stride=1, bias=True)
def forward(self, input):
output = self.conv1(input)
return output
writer = SummaryWriter("./Ni_Conv2d")
Ni_Conv2d = single_conv()
step = 1
for data in dataloader:
image, target = data
print(image.type)
# x = ToTensor(image) image已经是tensor数据类型了,所以不需要转换了
output = Ni_Conv2d(image) #这里写Ni_Conv2d()或者Ni_Conv2d.forward()都是可以的,前者是通过内置的__call__来调用的
print(image.shape) # torch.Size([64, 3, 32, 32])
print(output.shape) # torch.Size([64, 6, 30, 30]) 从数据结果来看,因为未加padding,所以从32*32变为了30*30
output2 = torch.reshape(output, (-1, 3, 30, 30)) #最好一步一步实例化,而不要只写一行 torch.reshape(output, (-1, 3, 30, 30)),
# 这样是没法改变output的数据形式的
print(output2.shape)
#output是六通道的,无法显示所以我们需要从【64, 6, 30, 30】 -->【xx, 3, 30, 30】 所以xx=128
writer.add_images("output", output2, step)
step = step + 1
writer.close()
高阶JavaScript笔记2(函数,回调函数,IIFE,函数中的this,语句加分号的问题,原型, 显式原型和隐式原型,原型链,原型链的补充,原型链的属性问题,探索instanceof,原型面试题)
华为昇腾MindSpore训练过程经验分享——D-Linknet
合宙esp32c3不带串口芯片的型号如何arduino和micropython
Pycocotools 报error: Microsoft Visual C++ 14.0 or greater is required. Get it with “Microsoft C++ Bui
php开发为什么要使用docker,Docker? 为什么要使用Docker?
.net webapi 实现 接口版本控制并打通swagger支持
基于.NetCore开发博客项目 StarBlog - (14) 实现主题切换功能