ssd-caffe的執行環境搭建
阿新 • • 發佈:2018-12-09
系統:Ubuntu16.04 + cuda9.0 +cudnn7.0
一、下載原始碼
進入想要安裝ssd的目錄下,下載原始碼
mkdir ssd
git clone https://github.com/weiliu89/caffe.git
git checkout ssd
下載完成後,當前目錄為 : ~/ssd/caffe
二、編譯
1、配置
cd ~/ssd/caffe
cp Makefile.config.example Makefile.config
注:由於之前,安裝過caffe,Makefile.config已經配置好,將之前的修改好的檔案直接拷貝到ssd-caffe下。
(而Makefile不用動)
Makefile.config連結:https://pan.baidu.com/s/1VVtAtMA18pkN9hnML7c3_Q 密碼:ntui
2、編譯
在~/ssd/caffe下,執行命令:
make -j8
加入環境變數,否則在voc生成資料時會出現錯誤,新增環境變數到PYTHONPATH 。
vi ~/.bashrc
# 在最後一行新增
export PYTHONPATH=/home/myuser/ssd/caffe/python:$PYTHONPATH #以自己實際路徑為主
# 儲存退出後,終端執行,更新環境變數
source ~/.bashrc
make py make test -j8 make runtest -j8 # 可選
參考:官方教程
跳過此部分:
(廢棄)之前在build的目錄下cmake時方法,後面出現錯誤。後來沒有用該方法。
cmake .. 出現問題:
CMake Error at CMakeLists.txt:85 (add_dependencies):
The dependency target "pycaffe" of target "pytest" does not exist.
解決:sudo apt-get install python-numpy
make all -j8("‐j8"是使用 CPU 的多核進行編譯,可以極大地加速編譯的速度) make install make runtest(測試,這一步不是必須的) make pycaffe(編譯pycaffe)
三、下載voc資料集
1、下載VGGNET模型
By default, we assume the model is stored in $CAFFE_ROOT/models/VGGNet/
百度連結:https://pan.baidu.com/s/1giL3U5qbGzyTxRNSb7U99A 密碼:sam4
2、下載資料集
cd ~/data
# 下載
wget http://host.robots.ox.ac.uk/pascal/VOC/voc2012/VOCtrainval_11-May-2012.tar
wget http://host.robots.ox.ac.uk/pascal/VOC/voc2007/VOCtrainval_06-Nov-2007.tar
wget http://host.robots.ox.ac.uk/pascal/VOC/voc2007/VOCtest_06-Nov-2007.tar
# 解壓
cd ~/data
tar -xvf VOCtrainval_11-May-2012.tar
tar -xvf VOCtrainval_06-Nov-2007.tar
tar -xvf VOCtest_06-Nov-2007.tar # 請嚴格按照這個順序解壓
3、執行指令碼,生成 LMDB格式檔案
3.1 執行create_list.sh
cd ~/ssd/caffe
# Create the trainval.txt, test.txt, and test_name_size.txt in data/VOC0712/
./data/VOC0712/create_list.sh
在當前的 ./data/VOC0712/ 目錄下 ,生成txt的檔案,檔案每行內容大致如下:
VOC2012/JPEGImages/2008_005191.jpg VOC2012/Annotations/2008_005191.xml
3.2 執行create_data.sh
# You can modify the parameters in create_data.sh if needed.
# It will create lmdb files for trainval and test with encoded original image:
# - $HOME/data/VOCdevkit/VOC0712/lmdb/VOC0712_trainval_lmdb
# - $HOME/data/VOCdevkit/VOC0712/lmdb/VOC0712_test_lmdb
# and make soft links at examples/VOC0712/
./data/VOC0712/create_data.sh
生成lmdb格式檔案,同時建立軟連結 。
未完待續