**1 **問題
如何對圖片進行卷積計算?
**2 **方法
先導入torch和torch里的nn類,然后設置一個指定尺寸的隨機像素值的圖片,然后使用nn.conv2d函數(shù)進行卷積計算,然后建立全連接層,最后得到新的圖片的尺寸
步驟:
(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的有趣之處,通過對圖片進行卷積計算,設置卷積計算的通道,設置卷積核尺寸大小,設置步長,設置補充,最后進行拉伸,得到最后的圖片的尺寸,讓我對卷積有了進一步的了解,對卷積的使用以及深度學習的魅力有了進一步的了解。
-
函數(shù)
+關注
關注
3文章
4331瀏覽量
62630 -
卷積
+關注
關注
0文章
95瀏覽量
18512 -
pytorch
+關注
關注
2文章
808瀏覽量
13229
發(fā)布評論請先 登錄
相關推薦
評論