**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的有趣之處,通過對圖片進行卷積計算,設置卷積計算的通道,設置卷積核尺寸大小,設置步長,設置補充,最后進行拉伸,得到最后的圖片的尺寸,讓我對卷積有了進一步的了解,對卷積的使用以及深度學習的魅力有了進一步的了解。
-
函數
+關注
關注
3文章
4365瀏覽量
63899 -
卷積
+關注
關注
0文章
95瀏覽量
18672 -
pytorch
+關注
關注
2文章
808瀏覽量
13694
發布評論請先 登錄
什么是卷積碼? 什么是卷積碼的約束長度?
FFT與DFT計算時間的比較及圓周卷積代替線性卷積的有效性實
求函數計算序列線性卷積結果
怎么將一副卷積后的圖片中的負像素值加絕對值
什么是卷積_卷積的意義

五分鐘了解卷積神經網絡

評論