Caffe
http://blog.csdn.net/zb1165048017/article/details/51355143
選的無CPU版,python2.7(Anaconda2),下載的源碼來自微軟的windows版(命名為caffe-master)
因為cmake不成功,沒有用BLVC的源碼。
如果重新編譯cmake windows版的話,需要改build_win腳本。
如果找不到generator,
1.很可能是因為vc\bin那個路徑沒有放在path變量里。
2.把scripts下的build文件夾清空,重啟cmd
之前用的是win7編譯,一直不成功,后來換了win10就ok了。
GAN
https://github.com/longriyao/caffe_GAN
這個教程說要修改源碼,改完src的caffe文件里的net.cpp和caffe.proto之后報錯了,棄了,換一個教程。
http://blog.csdn.net/seven_first/article/details/52999758
不知為何,換成python2.7后cmake又好使了。不過沒什么關系,我用的還是微軟的集成包(caffe-master)。
把教程里說的改修改的源碼替換掉之前的幾個文件,然后在include/caffe/layer下添加gan的頭文件(一共三個),src/caffe/layer下添加gan的cpp文件(也是三個)。如果是直接復制到文件夾的不要忘記在項目里也添加文件。
右鍵重新生成解決方案,編譯成功!
訓練一下網絡試試。
教程提供了train.prototxt和solver.prototxt,在train的最開頭引入了數據集,修改一下txt以及root的路徑(我用的絕對路徑)。txt里的格式是每行兩個文件名(原圖和label,空格隔開)。
solver.prototxt里最后添加一行
solver.mode=cpu
然后打開caffe.exe所在的目錄,應該是BUILD/x64。
cmd輸入:
caffe train --solver=(solver.prototxt的路徑)
就開始訓練啦!
注意:最好把路徑都換成絕對路徑,不然一不留神報很多路徑的錯。