1. 程式人生 > >ssd-caffe的執行環境搭建

ssd-caffe的執行環境搭建

系統: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格式檔案,同時建立軟連結 。

未完待續

參考 

官方參考

參考一

參考二