欧美bbbwbbbw肥妇,免费乱码人妻系列日韩,一级黄片

pytorch 如何自定義卷積核權(quán)值參數(shù)

 更新時間:2021年05月27日 08:40:43   作者:Mr.Jcak  
這篇文章主要介紹了pytorch 自定義卷積核權(quán)值參數(shù)的操作,具有很好的參考價值,希望對大家有所幫助。如有錯誤或未考慮完全的地方,望不吝賜教

pytorch中構(gòu)建卷積層一般使用nn.Conv2d方法,有些情況下我們需要自定義卷積核的權(quán)值weight,而nn.Conv2d中的卷積參數(shù)是不允許自定義的,此時可以使用torch.nn.functional.conv2d簡稱F.conv2d

torch.nn.functional.conv2d(input, weight, bias=None, stride=1, padding=0, dilation=1, groups=1)

F.conv2d可以自己輸入且也必須要求自己輸入卷積權(quán)值weight和偏置bias。因此,構(gòu)建自己想要的卷積核參數(shù),再輸入F.conv2d即可。

下面是一個用F.conv2d構(gòu)建卷積層的例子

這里為了網(wǎng)絡模型需要寫成了一個類:

class CNN(nn.Module):
    def __init__(self):
        super(CNN, self).__init__()
        self.weight = nn.Parameter(torch.randn(16, 1, 5, 5))  # 自定義的權(quán)值
        self.bias = nn.Parameter(torch.randn(16))    # 自定義的偏置
 
    def forward(self, x):
        x = x.view(x.size(0), -1)
        out = F.conv2d(x, self.weight, self.bias, stride=1, padding=0)
        return out

值得注意的是,pytorch中各層需要訓練的權(quán)重的數(shù)據(jù)類型設為nn.Parameter,而不是Tensor或者Variable。parameter的require_grad默認設置為true,而Varaible默認設置為False。

補充:pytorch中卷積參數(shù)的理解

在這里插入圖片描述

在這里插入圖片描述

kernel_size代表著卷積核,例如kernel_size=3或kernel_size=(3,7);

stride:表明卷積核在像素級圖像上行走的步長,如圖2,步長為1;

padding:為上下左右填充的大小,例如padding=0/1/(1,1)/(1,3),

padding=0 不填充;

padding=1/(1,1) 上下左右分別填充1個格;

padding=(1,3) 高(上下)填充2個格,寬(左右)填充6個格;

卷積代碼

torch.nn.Conv2d(512,512,kernel_size=(3,7),stride=2,padding=1)

指定輸出形狀的上采樣

def upsample_add(self,x,y):
        _,_,H,W = y.size()
        return F.interpolate(x, size=(H,W), mode='bilinear', align_corners=False) + y

反卷積上采樣

output_shape_w=kernel_size_w+(output_w-1)(kernel_size_w-1)+2padding
self.upscore2 = nn.ConvTranspose2d(
            512, 1, kernel_size=3, stride=2,padding=0, bias=False)

以上為個人經(jīng)驗,希望能給大家一個參考,也希望大家多多支持腳本之家。

相關(guān)文章

  • 簡單的Python解密rsa案例

    簡單的Python解密rsa案例

    這篇文章主要為大家介紹了簡單的Python解密rsa案例詳解,有需要的朋友可以借鑒參考下,希望能夠有所幫助,祝大家多多進步,早日升職加薪
    2022-07-07
  • Python處理數(shù)據(jù)之匹配兩個Excel文件數(shù)據(jù)的實現(xiàn)方法

    Python處理數(shù)據(jù)之匹配兩個Excel文件數(shù)據(jù)的實現(xiàn)方法

    這篇文章主要給大家介紹了關(guān)于Python處理數(shù)據(jù)之匹配兩個Excel文件數(shù)據(jù)的實現(xiàn)方法,可以使用Python的Pandas庫來實現(xiàn)兩層循環(huán)匹配兩個Excel文件,需要的朋友可以參考下
    2023-09-09
  • python進程和線程用法知識點總結(jié)

    python進程和線程用法知識點總結(jié)

    在本篇文章里小編給大家整理了關(guān)于python進程和線程用法以及相關(guān)實例內(nèi)容,需要的朋友們跟著學習下。
    2019-05-05
  • Pandas中DataFrame的常用用法分享

    Pandas中DataFrame的常用用法分享

    Pandas是Python中最流行的數(shù)據(jù)分析和處理工具之一,它提供了一個名為DataFrame的數(shù)據(jù)結(jié)構(gòu),可以被認為是一個二維表格或電子表格。本文主要來和大家分享一下Pandas中DataFrame的常用用法,希望對大家有所幫助
    2023-04-04
  • python實現(xiàn)簡單飛行棋

    python實現(xiàn)簡單飛行棋

    這篇文章主要為大家詳細介紹了python實現(xiàn)簡單飛行棋,文中示例代碼介紹的非常詳細,具有一定的參考價值,感興趣的小伙伴們可以參考一下
    2020-02-02
  • 最新python正則表達式(re模塊)詳解

    最新python正則表達式(re模塊)詳解

    在Python中需要通過正則表達式對字符串進?匹配的時候,可以使??個python自帶的模塊,名字為re,這篇文章主要介紹了python正則表達式(re模塊)詳解,需要的朋友可以參考下
    2023-01-01
  • JetBrains PyCharm(Community版本)的下載、安裝和初步使用圖文教程詳解

    JetBrains PyCharm(Community版本)的下載、安裝和初步使用圖文教程詳解

    這篇文章主要介紹了JetBrains PyCharm(Community版本)的下載、安裝和初步使用教程,本文圖文并茂給大家介紹的非常詳細,對大家的學習和工作具有一定的參考借鑒價值 ,需要的朋友可以參考下
    2020-03-03
  • 如何在Python中使用pyecharts圖形畫可視化大屏

    如何在Python中使用pyecharts圖形畫可視化大屏

    最近碰巧需要用到pyecharts,pyecharts庫是一個用于生成echarts圖表的類庫,這篇文章主要給大家介紹了關(guān)于如何在Python中使用pyecharts圖形畫可視化大屏的相關(guān)資料,需要的朋友可以參考下
    2024-05-05
  • 極簡的Python入門指引

    極簡的Python入門指引

    這篇文章是一個極簡的Python入門指引、原文在網(wǎng)絡上人氣相當高,將一些基本知識用實例代碼說明,需要的朋友可以參考下
    2015-04-04
  • Python?opencv圖像膨脹與腐蝕處理實例探究

    Python?opencv圖像膨脹與腐蝕處理實例探究

    這篇文章主要介紹了Python?opencv圖像膨脹與腐蝕處理實例探究
    2024-01-01

最新評論