pytorch 拆分已定義的網(wǎng)絡結(jié)構(gòu)(slicing network)

在pytorch中餐禁,常會load已有模型甚至pretrained的模型氯析,用其中幾層作為特征提绕瘛(feature extraction)懦铺。比如用pytorch內(nèi)置的pretrained ResNet作為特征提取器,需要把fully connected layer去掉贯要∨龋可以用children()方法提出需要的層

import torch.nn as nn
from torchvision import models

model = models.resnet50(pretrained=True)
truncated_model = nn.Sequential(*list(model.children())[:8])
print(truncated_model)

truncted_model可作為feature extractor,需要注意輸入輸出大小即可崇渗。
PS: *list可以達到以下效果

l = ["./foo", "bar", "quux"]

funcXXX(*l)
# 等價于
funcXXX("./foo", "bar", "quux")

也即是字逗,iterate 提取list中的內(nèi)容,并以逗號分隔宅广。滿足nn.Sequential()的輸入條件

?著作權(quán)歸作者所有,轉(zhuǎn)載或內(nèi)容合作請聯(lián)系作者
  • 序言:七十年代末扳肛,一起剝皮案震驚了整個濱河市,隨后出現(xiàn)的幾起案子乘碑,更是在濱河造成了極大的恐慌挖息,老刑警劉巖,帶你破解...
    沈念sama閱讀 221,635評論 6 515
  • 序言:濱河連續(xù)發(fā)生了三起死亡事件兽肤,死亡現(xiàn)場離奇詭異套腹,居然都是意外死亡,警方通過查閱死者的電腦和手機资铡,發(fā)現(xiàn)死者居然都...
    沈念sama閱讀 94,543評論 3 399
  • 文/潘曉璐 我一進店門电禀,熙熙樓的掌柜王于貴愁眉苦臉地迎上來,“玉大人笤休,你說我怎么就攤上這事尖飞。” “怎么了?”我有些...
    開封第一講書人閱讀 168,083評論 0 360
  • 文/不壞的土叔 我叫張陵政基,是天一觀的道長贞铣。 經(jīng)常有香客問我,道長沮明,這世上最難降的妖魔是什么辕坝? 我笑而不...
    開封第一講書人閱讀 59,640評論 1 296
  • 正文 為了忘掉前任,我火速辦了婚禮荐健,結(jié)果婚禮上酱畅,老公的妹妹穿的比我還像新娘。我一直安慰自己江场,他們只是感情好纺酸,可當我...
    茶點故事閱讀 68,640評論 6 397
  • 文/花漫 我一把揭開白布。 她就那樣靜靜地躺著址否,像睡著了一般餐蔬。 火紅的嫁衣襯著肌膚如雪。 梳的紋絲不亂的頭發(fā)上在张,一...
    開封第一講書人閱讀 52,262評論 1 308
  • 那天,我揣著相機與錄音矮慕,去河邊找鬼帮匾。 笑死,一個胖子當著我的面吹牛痴鳄,可吹牛的內(nèi)容都是我干的瘟斜。 我是一名探鬼主播,決...
    沈念sama閱讀 40,833評論 3 421
  • 文/蒼蘭香墨 我猛地睜開眼痪寻,長吁一口氣:“原來是場噩夢啊……” “哼螺句!你這毒婦竟也來了?” 一聲冷哼從身側(cè)響起橡类,我...
    開封第一講書人閱讀 39,736評論 0 276
  • 序言:老撾萬榮一對情侶失蹤蛇尚,失蹤者是張志新(化名)和其女友劉穎,沒想到半個月后顾画,有當?shù)厝嗽跇淞掷锇l(fā)現(xiàn)了一具尸體取劫,經(jīng)...
    沈念sama閱讀 46,280評論 1 319
  • 正文 獨居荒郊野嶺守林人離奇死亡,尸身上長有42處帶血的膿包…… 初始之章·張勛 以下內(nèi)容為張勛視角 年9月15日...
    茶點故事閱讀 38,369評論 3 340
  • 正文 我和宋清朗相戀三年研侣,在試婚紗的時候發(fā)現(xiàn)自己被綠了谱邪。 大學時的朋友給我發(fā)了我未婚夫和他白月光在一起吃飯的照片。...
    茶點故事閱讀 40,503評論 1 352
  • 序言:一個原本活蹦亂跳的男人離奇死亡庶诡,死狀恐怖惦银,靈堂內(nèi)的尸體忽然破棺而出,到底是詐尸還是另有隱情,我是刑警寧澤扯俱,帶...
    沈念sama閱讀 36,185評論 5 350
  • 正文 年R本政府宣布书蚪,位于F島的核電站,受9級特大地震影響蘸吓,放射性物質(zhì)發(fā)生泄漏善炫。R本人自食惡果不足惜,卻給世界環(huán)境...
    茶點故事閱讀 41,870評論 3 333
  • 文/蒙蒙 一库继、第九天 我趴在偏房一處隱蔽的房頂上張望箩艺。 院中可真熱鬧,春花似錦宪萄、人聲如沸艺谆。這莊子的主人今日做“春日...
    開封第一講書人閱讀 32,340評論 0 24
  • 文/蒼蘭香墨 我抬頭看了看天上的太陽静汤。三九已至,卻和暖如春居凶,著一層夾襖步出監(jiān)牢的瞬間虫给,已是汗流浹背。 一陣腳步聲響...
    開封第一講書人閱讀 33,460評論 1 272
  • 我被黑心中介騙來泰國打工侠碧, 沒想到剛下飛機就差點兒被人妖公主榨干…… 1. 我叫王不留抹估,地道東北人。 一個月前我還...
    沈念sama閱讀 48,909評論 3 376
  • 正文 我出身青樓弄兜,卻偏偏與公主長得像药蜻,于是被迫代替她去往敵國和親。 傳聞我的和親對象是個殘疾皇子替饿,可洞房花燭夜當晚...
    茶點故事閱讀 45,512評論 2 359

推薦閱讀更多精彩內(nèi)容

  • 對不起 對不起 對不起 我不狡辯了 因為 無所遁形 有時候 你比我還要了解我 如果不是你 我早嚇跑了 越來越覺得自...
    卐溺者卐閱讀 365評論 0 0
  • 昨晚學了《學習的心身基礎(chǔ)》第五課歸屬感的微信語音輔導課和今天聽了第六課動機基礎(chǔ)视卢。才知道歸屬感是指個人自己感...
    幸福遇見_1a92閱讀 1,012評論 0 0
  • 或許是晚飯吃晚了据过,也或許是吃多吃咸了颊埃,再或者就是白天太忙了沒有時間好好喝一盞茶,總之蝶俱,我在22:30以后突然升起要...
    你好apple閱讀 261評論 0 2
  • 一片星空班利,通過手腕 ——或者其它入口,進入體內(nèi) 因了它榨呆,人們會成為一片宇宙的海 人馬一路領(lǐng)跑 他已目標明確罗标,是光明...
    貓貓葵閱讀 276評論 0 3
  • 編程:自己體驗一個產(chǎn)品從無到有的過程 設(shè)計:設(shè)計一些有意思的東西庸队,比如產(chǎn)品,比如海報闯割、封面彻消、字體,anyway 吉...
    kuileishi閱讀 161評論 0 1