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)文章
OpenPyXL是一個(gè)強(qiáng)大的Python庫,用于處理Excel文件,允許讀取、編輯和創(chuàng)建Excel工作簿和工作表,本文將詳細(xì)介紹OpenPyXL的各種功能,希望對(duì)大家有所幫助2023-11-11
python,Django實(shí)現(xiàn)的淘寶客登錄功能示例
這篇文章主要介紹了python,Django實(shí)現(xiàn)的淘寶客登錄功能,結(jié)合實(shí)例形式分析了Django框架基于淘寶接口的登錄功能相關(guān)操作技巧,需要的朋友可以參考下2019-06-06
Python垃圾回收機(jī)制三種實(shí)現(xiàn)方法
這篇文章主要介紹了Python垃圾回收機(jī)制三種實(shí)現(xiàn)方法,文中通過示例代碼介紹的非常詳細(xì)，對(duì)大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值,需要的朋友可以參考下2020-04-04
在?Python?中,多進(jìn)程編程是一種提高程序運(yùn)行效率的有效手段,本文將詳細(xì)講解?Python?多進(jìn)程編程的基本概念,使用方法及注意事項(xiàng),希望對(duì)大家有所幫助2024-12-12
Python實(shí)現(xiàn)遍歷子文件夾并將文件復(fù)制到不同的目標(biāo)文件夾
這篇文章主要介紹了如何基于Python語言實(shí)現(xiàn)遍歷多個(gè)子文件夾，將每一個(gè)子文件夾中大量的文件，按照每一個(gè)文件的文件名稱的特點(diǎn)復(fù)制到不同的目標(biāo)文件夾中，感興趣的可以了解下2023-08-08
這篇文章主要介紹了Python如何測(cè)試stdout輸出，幫助大家更好的理解和學(xué)習(xí)Python，感興趣的朋友可以了解下2020-08-08
這篇文章主要介紹了Python中asyncore的用法,asyncore提供了方便的網(wǎng)絡(luò)操作方法,本文以連接并解析www.python.org主頁為例加以說明,需要的朋友可以參考下2014-09-09
python動(dòng)態(tài)加載技術(shù)解析
這篇文章主要介紹了python動(dòng)態(tài)加載技術(shù)解析,說簡單點(diǎn)就是，如果開發(fā)者發(fā)現(xiàn)自己的代碼有bug，那么他可以在不關(guān)閉原來代碼的基礎(chǔ)之上，動(dòng)態(tài)替換模塊替換方法一般用reload來完成,需要的朋友可以參考下2023-07-07
今天小編就為大家分享一篇Python代碼打開本地.mp4格式文件的方法，具有很好的參考價(jià)值，希望對(duì)大家有所幫助。一起跟隨小編過來看看吧2019-01-01
本文給大家匯總介紹了Python安裝第三方庫及常見問題處理方法，非常的簡單使用，有需要的小伙伴可以參考下2016-09-09