如何对图片进行卷积计算

电子说

1.2w人已加入

描述

**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=322828,\\

out_features=10)

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

x = torch.flatten(x,1) # [128,322828]

out = fc(x)

print(out.shape)

函数

**3 **结语

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

打开APP阅读更多精彩内容
声明:本文内容及配图由入驻作者撰写或者入驻合作网站授权转载。文章观点仅代表作者本人,不代表电子发烧友网立场。文章及其配图仅供工程师学习之用,如有内容侵权或者其他违规问题,请联系本站处理。 举报投诉

全部0条评论

快来发表一下你的评论吧 !

×
20
完善资料,
赚取积分