paper:https://sites.google.com/site/xipengcshomepage/research/ijcai18
code:https://github.com/bluer555/CR-GAN
在開始前我先提醒各位進(jìn)行的小伙伴，本篇文章的訓(xùn)練所需要的數(shù)據(jù)集相當(dāng)大（10G有多吧），所以你們需要開足vpn才能去下載這個龐大的數(shù)據(jù)集，不然就不要訓(xùn)練了。
CR-GAN是用GAN(對抗網(wǎng)絡(luò))來進(jìn)行人臉多角度的圖片生成的文章，相對于先前的BiGAN、DR-GAN、TP-GAN，CR-GAN在網(wǎng)絡(luò)上做了一個改進(jìn)通過采用雙支網(wǎng)絡(luò)已經(jīng)共享網(wǎng)絡(luò)參數(shù)的形式進(jìn)行訓(xùn)練，本篇博客僅對該文章的訓(xùn)練做一個詳細(xì)的說明以及步驟的說明。
訓(xùn)練環(huán)境的搭建：
1. Python 2.7
2. Pytorch 0.3.1
請根據(jù)自己電腦cuda的版本進(jìn)行下載，不然就會出問題，本人系統(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/
訓(xùn)練的步驟:
1. 下載預(yù)訓(xùn)練模型；
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. 下載訓(xùn)練的數(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文件
# 訓(xùn)練所需要圖片的路徑的列表parser.add_argument("-d","--data_list", type=str, default="./list_test.txt")# 訓(xùn)練模型保存的路徑parser.add_argument('--outf', default='./evaluate', help='folder to output images and model checkpoints')# 預(yù)訓(xùn)練模型的路徑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. 在項目的當(dāng)前目錄下輸入：
:~$ cd CRGAN
:~$ python train.py
6. 當(dāng)你開始訓(xùn)練時會遇到一些問題，請不要緊張，這個問題不是大問題：
問題１：...data_parallel.py:24: UserWarning: .......... warnings.warn(imbalance_warn.format(...))
解析：這個問題是你安裝的pytorch版本不是0.3.1所造成的
問題２：...model.py:115: UserWarning: Implicit dimensiion choice ..........Change the call to include dim=x.... v = self.softmax(v)....
解析：這個問題是由于作者先前寫項目是給予pytorch較低的版本寫的，所以會出現(xiàn)這個警告，不用管。
訓(xùn)練的全過程到此結(jié)束，如果要測試請見下一篇博客。
以上是本作者為了解決問題所提出的建議，不見得都能適用，但是基本能夠解決問題，如果有什么不正確的地方，請在下方評論區(qū)留言，本文章僅代表作者本人意見！
