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