CR-GAN系列1:CR-GAN:Learning Complete Representations for Multi-view Generation的訓練

paper:https://sites.google.com/site/xipengcshomepage/research/ijcai18

code:https://github.com/bluer555/CR-GAN

在開始前我先提醒各位進行的小伙伴砰蠢,本篇文章的訓練所需要的數(shù)據(jù)集相當大(10G有多吧),所以你們需要開足vpn才能去下載這個龐大的數(shù)據(jù)集拴还,不然就不要訓練了。

CR-GAN是用GAN(對抗網(wǎng)絡(luò))來進行人臉多角度的圖片生成的文章喘鸟,相對于先前的BiGAN街夭、DR-GAN触菜、TP-GAN,CR-GAN在網(wǎng)絡(luò)上做了一個改進通過采用雙支網(wǎng)絡(luò)已經(jīng)共享網(wǎng)絡(luò)參數(shù)的形式進行訓練兽埃,本篇博客僅對該文章的訓練做一個詳細的說明以及步驟的說明侥钳。

訓練環(huán)境的搭建:

1. Python 2.7

2. Pytorch 0.3.1

請根據(jù)自己電腦cuda的版本進行下載,不然就會出問題柄错,本人系統(tǒng)為linux舷夺,cuda為8.0,python 為2.7,所以一下給出的安裝指令以及是該環(huán)境下的0.3.1版本售貌,千萬別弄錯了给猾,不然會報錯誤或警告。

:~$ pip install http://download.pytorch.org/whl/cu80/torch-0.3.1-cp27-cp27mu-linux_x86_64.whl

相關(guān)配包教程:https://ptorch.com/news/145.html

官網(wǎng)安裝:https://pytorch.org/previous-versions/

訓練的步驟:

1. 下載預(yù)訓練模型颂跨;

Google驅(qū)動下載:https://drive.google.com/open?id=1J3VffWKe8akdiNM2hy7NI3lY4xM_xL-c

百度網(wǎng)盤下載:

1.https://pan.baidu.com/s/1Bc_Ipkz22Q28McfjH7thOQ? 密碼:ac63

2.https://pan.baidu.com/s/1DvCWRbgOJQpjaPV8J4lZIA密碼:avwe

3.https://pan.baidu.com/s/1391QFBo4wL7xZhiu4fWYyQ密碼:13zb

2. 下載訓練的數(shù)據(jù)庫敢伸;

(由于數(shù)據(jù)庫相對較大,我就不上傳了恒削,實在沒辦法池颈,請自行下載吧)

將下載好的數(shù)據(jù)存放入你的工程項目CR-GAN文件夾的data路徑下尾序,沒有就自行創(chuàng)建一個。

數(shù)據(jù)庫1:https://drive.google.com/open?id=1QxNCh6vfNSZkod1Rg_zHLI1FM8WyXix4

300w-LP 數(shù)據(jù)庫:http://www.cbsr.ia.ac.cn/users/xiangyuzhu/projects/3DDFA/main.htm

數(shù)據(jù)庫2:https://drive.google.com/open?id=1DD6AO9Y5rAgiiW7IJY2kBxI_bCcfhYo4

300w-LP(作用于dataload.py中的txt文件):https://drive.google.com/open?id=1TIfcpn4N3rgGlzWl0lXNZKhy7XWVWOoA

3. 從git中下載源碼躯砰;

:~$ git clone https://github.com/bluer555/CR-GAN

4. 在源碼中修改相關(guān)讀取文件的路徑每币;

(1)在train.py文件

# 訓練所需要圖片的路徑的列表parser.add_argument("-d","--data_list", type=str, default="./list_test.txt")# 訓練模型保存的路徑parser.add_argument('--outf', default='./evaluate', help='folder to output images and model checkpoints')# 預(yù)訓練模型的路徑parser.add_argument('--modelf', default='./evaluate_model', help='folder to input images and model checkpoints')

(2)在data_loader.py文件

defget_multiPIE_img(img_path):.....? ? ? ? img2_path ='/porject-path/data/multi_PIE_crop_128/'+ ID +'/'+ ID +'_01_'+ status +'_'+ view +'_'+ bright +'_crop_128.png'.....

這個地方需要修改你工程項目的路徑。

5. 在項目的當前目錄下輸入:

:~$ cd CRGAN

:~$ python train.py

6. 當你開始訓練時會遇到一些問題琢歇,請不要緊張兰怠,這個問題不是大問題:

問題1:...data_parallel.py:24: UserWarning: .......... warnings.warn(imbalance_warn.format(...))

解析:這個問題是你安裝的pytorch版本不是0.3.1所造成的

問題2:...model.py:115: UserWarning: Implicit dimensiion choice ..........Change the call to include dim=x.... v = self.softmax(v)....

解析:這個問題是由于作者先前寫項目是給予pytorch較低的版本寫的,所以會出現(xiàn)這個警告李茫,不用管揭保。

訓練的全過程到此結(jié)束,如果要測試請見下一篇博客魄宏。

以上是本作者為了解決問題所提出的建議掖举,不見得都能適用,但是基本能夠解決問題娜庇,如果有什么不正確的地方,請在下方評論區(qū)留言方篮,本文章僅代表作者本人意見名秀!

?著作權(quán)歸作者所有,轉(zhuǎn)載或內(nèi)容合作請聯(lián)系作者
  • 序言:七十年代末,一起剝皮案震驚了整個濱河市藕溅,隨后出現(xiàn)的幾起案子匕得,更是在濱河造成了極大的恐慌,老刑警劉巖巾表,帶你破解...
    沈念sama閱讀 206,378評論 6 481
  • 序言:濱河連續(xù)發(fā)生了三起死亡事件汁掠,死亡現(xiàn)場離奇詭異,居然都是意外死亡集币,警方通過查閱死者的電腦和手機考阱,發(fā)現(xiàn)死者居然都...
    沈念sama閱讀 88,356評論 2 382
  • 文/潘曉璐 我一進店門,熙熙樓的掌柜王于貴愁眉苦臉地迎上來鞠苟,“玉大人乞榨,你說我怎么就攤上這事〉庇椋” “怎么了吃既?”我有些...
    開封第一講書人閱讀 152,702評論 0 342
  • 文/不壞的土叔 我叫張陵,是天一觀的道長跨细。 經(jīng)常有香客問我鹦倚,道長,這世上最難降的妖魔是什么冀惭? 我笑而不...
    開封第一講書人閱讀 55,259評論 1 279
  • 正文 為了忘掉前任震叙,我火速辦了婚禮掀鹅,結(jié)果婚禮上,老公的妹妹穿的比我還像新娘捐友。我一直安慰自己淫半,他們只是感情好,可當我...
    茶點故事閱讀 64,263評論 5 371
  • 文/花漫 我一把揭開白布匣砖。 她就那樣靜靜地躺著科吭,像睡著了一般。 火紅的嫁衣襯著肌膚如雪猴鲫。 梳的紋絲不亂的頭發(fā)上对人,一...
    開封第一講書人閱讀 49,036評論 1 285
  • 那天,我揣著相機與錄音拂共,去河邊找鬼牺弄。 笑死啄骇,一個胖子當著我的面吹牛栏渺,可吹牛的內(nèi)容都是我干的盒延。 我是一名探鬼主播丈甸,決...
    沈念sama閱讀 38,349評論 3 400
  • 文/蒼蘭香墨 我猛地睜開眼顾画,長吁一口氣:“原來是場噩夢啊……” “哼笛洛!你這毒婦竟也來了蛤奢?” 一聲冷哼從身側(cè)響起奋隶,我...
    開封第一講書人閱讀 36,979評論 0 259
  • 序言:老撾萬榮一對情侶失蹤俭驮,失蹤者是張志新(化名)和其女友劉穎回溺,沒想到半個月后,有當?shù)厝嗽跇淞掷锇l(fā)現(xiàn)了一具尸體混萝,經(jīng)...
    沈念sama閱讀 43,469評論 1 300
  • 正文 獨居荒郊野嶺守林人離奇死亡遗遵,尸身上長有42處帶血的膿包…… 初始之章·張勛 以下內(nèi)容為張勛視角 年9月15日...
    茶點故事閱讀 35,938評論 2 323
  • 正文 我和宋清朗相戀三年,在試婚紗的時候發(fā)現(xiàn)自己被綠了逸嘀。 大學時的朋友給我發(fā)了我未婚夫和他白月光在一起吃飯的照片车要。...
    茶點故事閱讀 38,059評論 1 333
  • 序言:一個原本活蹦亂跳的男人離奇死亡,死狀恐怖崭倘,靈堂內(nèi)的尸體忽然破棺而出屯蹦,到底是詐尸還是另有隱情,我是刑警寧澤绳姨,帶...
    沈念sama閱讀 33,703評論 4 323
  • 正文 年R本政府宣布登澜,位于F島的核電站,受9級特大地震影響飘庄,放射性物質(zhì)發(fā)生泄漏脑蠕。R本人自食惡果不足惜,卻給世界環(huán)境...
    茶點故事閱讀 39,257評論 3 307
  • 文/蒙蒙 一、第九天 我趴在偏房一處隱蔽的房頂上張望谴仙。 院中可真熱鬧迂求,春花似錦、人聲如沸晃跺。這莊子的主人今日做“春日...
    開封第一講書人閱讀 30,262評論 0 19
  • 文/蒼蘭香墨 我抬頭看了看天上的太陽掀虎。三九已至凌盯,卻和暖如春,著一層夾襖步出監(jiān)牢的瞬間烹玉,已是汗流浹背驰怎。 一陣腳步聲響...
    開封第一講書人閱讀 31,485評論 1 262
  • 我被黑心中介騙來泰國打工, 沒想到剛下飛機就差點兒被人妖公主榨干…… 1. 我叫王不留二打,地道東北人县忌。 一個月前我還...
    沈念sama閱讀 45,501評論 2 354
  • 正文 我出身青樓,卻偏偏與公主長得像继效,于是被迫代替她去往敵國和親症杏。 傳聞我的和親對象是個殘疾皇子,可洞房花燭夜當晚...
    茶點故事閱讀 42,792評論 2 345

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

  • 古詩描繪瑞信,學生還是不按四條要求來擴寫鸳慈,我覺得該加大檢查力度。讓課代表代為檢查喧伞。和大家分享了《唐詩描繪》其中《渭城曲...
    春光里的小米閱讀 127評論 0 0
  • 你天真的以為情誼會永恒 所以相信了他的承諾 因為 你天真的以為這些普通的承諾 有或者沒有 并不影響什么 然而你不...
    青禾吖閱讀 1,738評論 1 9
  • 《LinuxShell腳本攻略》筆記潘鲫,Chap-5: 一團亂麻?沒這回事 入門 本章會研究一些用于解析網(wǎng)站內(nèi)容肋杖、下...
    Zhang21閱讀 1,276評論 0 2
  • 1 運用演繹結(jié)構(gòu)溉仑,讓說服更有力! 娘娘在一開場就運用提問的方式與大家達成共識:“你們當兵是為了建功立業(yè)和榮耀”状植,這...
    鹿偉倫閱讀 305評論 0 1