如何对图片进行卷积计算

2023-08-22 14:16:54 浏览数 (1)

1 问题

如何对图片进行卷积计算?

2 方法

先导入torch和torch里的nn类,然后设置一个指定尺寸的随机像素值的图片,然后使用nn.conv2d函数进行卷积计算,然后建立全连接层,最后得到新的图片的尺寸

步骤:

(1) 导入实验所需要用到的包 from numpy import pad import torch from torch import nn

(2) 设置一个指定尺寸的图片 x = torch.rand(128,3,28,28)

(3) 设置卷积

conv1 = nn.Conv2d(in_channels=3,

out_channels=16,kernel_size=3,

stride=1,padding=1)

(4) 建立全连接层然后对图片进行卷积计算,然后对图片进行拉伸,再将拉伸后的图片交给全连接层,最后打印救过卷积计算的图片的尺寸

fc = nn.Linear(in_features=32*28*28,

out_features=10)

x = conv1(x) # [128,16,28,28]

x = torch.flatten(x,1) # [128,32*28*28]

out = fc(x)

print(out.shape)

3 结语

这次实验我们更加深入的了解了torch的有趣之处,通过对图片进行卷积计算,设置卷积计算的通道,设置卷积核尺寸大小,设置步长,设置补充,最后进行拉伸,得到最后的图片的尺寸,让我对卷积有了进一步的了解,对卷积的使用以及深度学习的魅力有了进一步的了解。

0 人点赞