3D-R2N2代碼分析

? ? ? 文章地址:https://chrischoy.github.io/publication/r2n2/

? ? ? 代碼地址:https://github.com/chrischoy/3d-r2n2/

? ? ? 整體代碼基于theano框架,github上有詳細(xì)的安裝過程,所有代碼均在virtualenv中實(shí)現(xiàn)肖爵,代碼對(duì)應(yīng)python版本應(yīng)在3.5及其以上

1咨演、該代碼的主程序使用demo為直接調(diào)用已有的模型參數(shù),進(jìn)行demo的重建喧笔,

2、訓(xùn)練程序主要通過sh工程文件中的參數(shù)來運(yùn)行

3、solver來運(yùn)行程序

? ? ? 代碼主程序有demo.py和main.py兩個(gè)部分沈矿,其中demo.py是為了直接運(yùn)行已有權(quán)重,main.py則是加入了訓(xùn)練的部分咬腋。網(wǎng)絡(luò)結(jié)構(gòu)結(jié)構(gòu)放在model中羹膳,res_gru_net為論文主要模型。其余主要庫文件在lib文件夾下根竿。本代碼中的網(wǎng)絡(luò)結(jié)構(gòu)多數(shù)為自寫陵像,少量使用了theano部分代碼,主要使用函數(shù)為theano.function和theano.scan寇壳。

? ? ? ?lib文件夾下醒颖,config.py存放全局所有調(diào)用的參變量。lib中的layer編寫了每個(gè)層面的各個(gè)操作以及最優(yōu)化所用的方法壳炎。

? ? ? ?demo流程:1泞歉、記錄保存的文件名,2匿辩、load圖片 3腰耙、下載模型4、netclass 通過load——model調(diào)用铲球,load_model的返回值為兩個(gè)model的object挺庞,netclass為models中的ResidualGRUNet,5稼病、net調(diào)用ResidualGRUNet所繼承的net中的load函數(shù)选侨,通過函數(shù)enumerate(http://blog.csdn.net/churximi/article/details/51648388),讀取路徑文件的各個(gè)數(shù)據(jù)的索引 權(quán)重然走,按順序依次存放在net.params中援制,6、這時(shí)調(diào)用theano.function,以net.x=demo_imgs芍瑞,net.y=none為輸入晨仑,以net.output net.loss為輸出,其中函數(shù)調(diào)用為gru_net中的net.output和net.loss

? ? ? ?如何修改程序:

1、調(diào)用部分網(wǎng)絡(luò)參數(shù)部分依然可以沿用load函數(shù)寻歧,返回的是list掌栅,而在額外初始化的部分可以繼續(xù)append添加添加層的參數(shù),


main流程:

1码泛、附加參數(shù)初始化猾封,

2、train_net()

3噪珊、train_net()中的main()

4晌缘、train_net()中的train_net(),讀取網(wǎng)絡(luò)類型參數(shù)值為cfg.CONST.NETWORK_CLASS痢站,通過函數(shù)make_data_processes讀取測(cè)試數(shù)據(jù)磷箕,調(diào)用solver中的train函數(shù)

最后編輯于
?著作權(quán)歸作者所有,轉(zhuǎn)載或內(nèi)容合作請(qǐng)聯(lián)系作者
  • 序言:七十年代末,一起剝皮案震驚了整個(gè)濱河市阵难,隨后出現(xiàn)的幾起案子岳枷,更是在濱河造成了極大的恐慌,老刑警劉巖呜叫,帶你破解...
    沈念sama閱讀 206,378評(píng)論 6 481
  • 序言:濱河連續(xù)發(fā)生了三起死亡事件空繁,死亡現(xiàn)場(chǎng)離奇詭異,居然都是意外死亡朱庆,警方通過查閱死者的電腦和手機(jī)盛泡,發(fā)現(xiàn)死者居然都...
    沈念sama閱讀 88,356評(píng)論 2 382
  • 文/潘曉璐 我一進(jìn)店門,熙熙樓的掌柜王于貴愁眉苦臉地迎上來娱颊,“玉大人傲诵,你說我怎么就攤上這事∠渌叮” “怎么了拴竹?”我有些...
    開封第一講書人閱讀 152,702評(píng)論 0 342
  • 文/不壞的土叔 我叫張陵,是天一觀的道長(zhǎng)颅痊。 經(jīng)常有香客問我殖熟,道長(zhǎng),這世上最難降的妖魔是什么斑响? 我笑而不...
    開封第一講書人閱讀 55,259評(píng)論 1 279
  • 正文 為了忘掉前任,我火速辦了婚禮钳榨,結(jié)果婚禮上舰罚,老公的妹妹穿的比我還像新娘。我一直安慰自己薛耻,他們只是感情好营罢,可當(dāng)我...
    茶點(diǎn)故事閱讀 64,263評(píng)論 5 371
  • 文/花漫 我一把揭開白布。 她就那樣靜靜地躺著,像睡著了一般饲漾。 火紅的嫁衣襯著肌膚如雪蝙搔。 梳的紋絲不亂的頭發(fā)上,一...
    開封第一講書人閱讀 49,036評(píng)論 1 285
  • 那天考传,我揣著相機(jī)與錄音吃型,去河邊找鬼。 笑死僚楞,一個(gè)胖子當(dāng)著我的面吹牛勤晚,可吹牛的內(nèi)容都是我干的。 我是一名探鬼主播泉褐,決...
    沈念sama閱讀 38,349評(píng)論 3 400
  • 文/蒼蘭香墨 我猛地睜開眼赐写,長(zhǎng)吁一口氣:“原來是場(chǎng)噩夢(mèng)啊……” “哼!你這毒婦竟也來了膜赃?” 一聲冷哼從身側(cè)響起挺邀,我...
    開封第一講書人閱讀 36,979評(píng)論 0 259
  • 序言:老撾萬榮一對(duì)情侶失蹤,失蹤者是張志新(化名)和其女友劉穎跳座,沒想到半個(gè)月后悠夯,有當(dāng)?shù)厝嗽跇淞掷锇l(fā)現(xiàn)了一具尸體,經(jīng)...
    沈念sama閱讀 43,469評(píng)論 1 300
  • 正文 獨(dú)居荒郊野嶺守林人離奇死亡躺坟,尸身上長(zhǎng)有42處帶血的膿包…… 初始之章·張勛 以下內(nèi)容為張勛視角 年9月15日...
    茶點(diǎn)故事閱讀 35,938評(píng)論 2 323
  • 正文 我和宋清朗相戀三年沦补,在試婚紗的時(shí)候發(fā)現(xiàn)自己被綠了。 大學(xué)時(shí)的朋友給我發(fā)了我未婚夫和他白月光在一起吃飯的照片咪橙。...
    茶點(diǎn)故事閱讀 38,059評(píng)論 1 333
  • 序言:一個(gè)原本活蹦亂跳的男人離奇死亡夕膀,死狀恐怖,靈堂內(nèi)的尸體忽然破棺而出美侦,到底是詐尸還是另有隱情产舞,我是刑警寧澤,帶...
    沈念sama閱讀 33,703評(píng)論 4 323
  • 正文 年R本政府宣布菠剩,位于F島的核電站易猫,受9級(jí)特大地震影響,放射性物質(zhì)發(fā)生泄漏具壮。R本人自食惡果不足惜准颓,卻給世界環(huán)境...
    茶點(diǎn)故事閱讀 39,257評(píng)論 3 307
  • 文/蒙蒙 一、第九天 我趴在偏房一處隱蔽的房頂上張望棺妓。 院中可真熱鬧攘已,春花似錦、人聲如沸怜跑。這莊子的主人今日做“春日...
    開封第一講書人閱讀 30,262評(píng)論 0 19
  • 文/蒼蘭香墨 我抬頭看了看天上的太陽。三九已至峡眶,卻和暖如春剧防,著一層夾襖步出監(jiān)牢的瞬間,已是汗流浹背辫樱。 一陣腳步聲響...
    開封第一講書人閱讀 31,485評(píng)論 1 262
  • 我被黑心中介騙來泰國打工峭拘, 沒想到剛下飛機(jī)就差點(diǎn)兒被人妖公主榨干…… 1. 我叫王不留,地道東北人搏熄。 一個(gè)月前我還...
    沈念sama閱讀 45,501評(píng)論 2 354
  • 正文 我出身青樓棚唆,卻偏偏與公主長(zhǎng)得像,于是被迫代替她去往敵國和親心例。 傳聞我的和親對(duì)象是個(gè)殘疾皇子宵凌,可洞房花燭夜當(dāng)晚...
    茶點(diǎn)故事閱讀 42,792評(píng)論 2 345

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