ssd-caffe的运行环境搭建

5 篇文章 0 订阅
1 篇文章 0 订阅

系统: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不用动)

配置文件修改参考:配置caffe的文件

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模型

Download fully convolutional reduced (atrous) VGGNet.

By default, we assume the model is stored in $CAFFE_ROOT/models/VGGNet/

官方链接:fully convolutional reduced (atrous) 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格式文件,同时建立软链接 。

 

 

未完待续

参考 

官方参考

参考一

参考二

  • 1
    点赞
  • 2
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
1. 安装依赖 首先需要安装 Caffe 和一些相关的依赖项: sudo apt-get update sudo apt-get install libprotobuf-dev libleveldb-dev libsnappy-dev libopencv-dev libhdf5-serial-dev protobuf-compiler sudo apt-get install --no-install-recommends libboost-all-dev sudo apt-get install libgflags-dev libgoogle-glog-dev liblmdb-dev 2. 下载源码 下载 Caffe 的源码,建议使用 GitHub 上的官方仓库: git clone https://github.com/weiliu89/caffe.git 3. 编译 Caffe 进入 Caffe 的根目录,然后执行以下命令进行编译: cd caffe cp Makefile.config.example Makefile.config make all -j8 make test -j8 make runtest -j8 4. 下载预训练模型 下载 SSD 的预训练模型,可以从以下链接中选择: https://github.com/weiliu89/caffe/tree/ssd#models 将下载好的模型放到 Caffe 的根目录下的 models 目录中。 5. 测试 SSD 在 Caffe 的根目录下执行以下命令,测试 SSD 的效果: ./build/tools/caffe test \ --model=models/VGGNet/VOC0712/SSD_300x300/deploy.prototxt \ --weights=models/VGGNet/VOC0712/SSD_300x300/VGG_VOC0712_SSD_300x300_iter_120000.caffemodel \ --iterations=5000 \ --gpu=0 6. 使用 SSD 如果已经成功测试了 SSD,就可以在自己的代码中使用它了。需要引入以下头文件: #include "caffe/caffe.hpp" #include "caffe/util/db.hpp" #include "caffe/blob.hpp" #include "caffe/common.hpp" #include "caffe/net.hpp" #include "caffe/proto/caffe.pb.h" #include "caffe/layers/input_layer.hpp" #include "caffe/layers/conv_layer.hpp" #include "caffe/layers/pooling_layer.hpp" #include "caffe/layers/softmax_layer.hpp" #include "caffe/layers/relu_layer.hpp" #include "caffe/layers/detection_output_layer.hpp" 然后可以通过以下方式加载模型: caffe::Net<float> net("models/VGGNet/VOC0712/SSD_300x300/deploy.prototxt", caffe::TEST); net.CopyTrainedLayersFrom("models/VGGNet/VOC0712/SSD_300x300/VGG_VOC0712_SSD_300x300_iter_120000.caffemodel"); 最后就可以使用 SSD 进行目标检测了。
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值