**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文章
4346瀏覽量
63012 -
卷積
+關注
關注
0文章
95瀏覽量
18550 -
pytorch
+關注
關注
2文章
808瀏覽量
13376
發布評論請先 登錄
相關推薦
什么是卷積碼? 什么是卷積碼的約束長度?
FFT與DFT計算時間的比較及圓周卷積代替線性卷積的有效性實
求函數計算序列線性卷積結果
怎么將一副卷積后的圖片中的負像素值加絕對值
什么是卷積_卷積的意義
![什么是<b class='flag-5'>卷積</b>_<b class='flag-5'>卷積</b>的意義](https://file1.elecfans.com//web2/M00/A6/F9/wKgZomUMQZGAEKBmAAAQZvSN4Xk045.png)
五分鐘了解卷積神經網絡
![五分鐘了解<b class='flag-5'>卷積</b>神經網絡](https://file.elecfans.com/web1/M00/50/22/pIYBAFrtHViAHKR0AAAS0kcom9U916.jpg)
評論