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ū)留言方篮,本文章僅代表作者本人意見名秀!