pytorch 實(shí)現(xiàn)在預(yù)訓(xùn)練模型的 input上增減通道
更新時(shí)間：2020年01月06日 11:31:51 作者：Hi_AI
今天小編就為大家分享一篇pytorch 實(shí)現(xiàn)在預(yù)訓(xùn)練模型的 input上增減通道，具有很好的參考價(jià)值，希望對(duì)大家有所幫助。一起跟隨小編過來看看吧
如何把imagenet預(yù)訓(xùn)練的模型，輸入層的通道數(shù)隨心所欲的修改，從而來適應(yīng)自己的任務(wù)
#增加一個(gè)通道 w = layers[0].weight layers[0] = nn.Conv2d(4, 64, kernel_size=(7, 7), stride=(2, 2), padding=(3, 3), bias=False) layers[0].weight = torch.nn.Parameter(torch.cat((w, w[:, :1, :, :]), dim=1)) #方式2 w = layers[0].weight layers[0] = nn.Conv2d(4, 64, kernel_size=(7, 7), stride=(2, 2), padding=(3, 3), bias=False) layers[0].weight = torch.nn.Parameter(torch.cat((w, torch.zeros(64, 1, 7, 7)), dim=1)) #單通道輸入 layers[0] = nn.Conv2d(1, 64, kernel_size=(7, 7), stride=(2, 2), padding=(3, 3), bias=False) layers[0].weight = torch.nn.Parameter(w[:, :1, :, :])
以上這篇pytorch 實(shí)現(xiàn)在預(yù)訓(xùn)練模型的 input上增減通道就是小編分享給大家的全部內(nèi)容了，希望能給大家一個(gè)參考，也希望大家多多支持腳本之家。
相關(guān)文章
python使用裝飾器和線程限制函數(shù)執(zhí)行時(shí)間的方法
這篇文章主要介紹了python使用裝飾器和線程限制函數(shù)執(zhí)行時(shí)間的方法,主要涉及timelimited函數(shù)的使用技巧,非常具有實(shí)用價(jià)值,需要的朋友可以參考下2015-04-04
Python利用臨時(shí)文件實(shí)現(xiàn)數(shù)據(jù)的保存
tempfile模塊專門用于創(chuàng)建臨時(shí)文件和臨時(shí)目錄，它既可以在?UNIX?平臺(tái)上運(yùn)行良好，也可以在?Windows?平臺(tái)上運(yùn)行良好。本文將利用tempfile模塊創(chuàng)建臨時(shí)文件來保存數(shù)據(jù)，感興趣的可以了解一下2022-07-07
這篇文章主要介紹了Python subprocess庫的使用詳解，小編覺得挺不錯(cuò)的，現(xiàn)在分享給大家，也給大家做個(gè)參考。一起跟隨小編過來看看吧2018-10-10
Python設(shè)置Word全局樣式和文本樣式的示例代碼
這篇文章主要介紹了如何利用Python對(duì)Word內(nèi)容進(jìn)行各種樣式的設(shè)置，讓其能夠看起來更加的美觀。文中的示例代碼講解詳細(xì)，需要的可以參考一下2022-05-05
在本文總小編給大家整理了關(guān)于python之中的迭代與迭代對(duì)象的相關(guān)基礎(chǔ)知識(shí)點(diǎn)內(nèi)容，需要的朋友們學(xué)習(xí)下。2018-10-10
python遠(yuǎn)程連接MySQL數(shù)據(jù)庫
這篇文章主要為大家詳細(xì)介紹了python遠(yuǎn)程連接MySQL數(shù)據(jù)庫，拉取數(shù)據(jù)存至本地文件，具有一定的參考價(jià)值，感興趣的小伙伴們可以參考一下2019-04-04
這篇文章主要介紹了Python讀寫Excel文件方法介紹,本文講解了xlrd、xlwt、xlutils等類庫的使用,需要的朋友可以參考下2014-11-11
在本篇文章里小編給大家整理的是關(guān)于如何用python免費(fèi)看美劇的方法內(nèi)容，需要的朋友們可以學(xué)習(xí)下。2020-08-08
這篇文章主要為大家詳細(xì)介紹了基于python3的socket聊天編程，文中示例代碼介紹的非常詳細(xì)，具有一定的參考價(jià)值，感興趣的小伙伴們可以參考一下2020-02-02
Python基本類型的連接組合和互相轉(zhuǎn)換方式(13種)
這篇文章主要介紹了Python中基本類型的連接組合和互相轉(zhuǎn)換13種方式,本文給大家介紹的非常詳細(xì)，具有一定的參考借鑒價(jià)值，需要的朋友可以參考下2019-12-12