C/C++教程

Ubuntu 20.04下通过Anaconda配置安装Caffe SSD记录

本文主要是介绍Ubuntu 20.04下通过Anaconda配置安装Caffe SSD记录，对大家解决编程问题具有一定的参考价值，需要的程序猿们随着小编来一起学习吧！
  1. 使用conda创建一个虚拟环境，名称以“caffe”为例
    因为他们的代码是用python2写的，所以选择2.7版本避免更多的问题。
conda create -n caffe python=2.7 -y
conda activate caffe

通过conda创建虚拟环境

  1. 下载GitHub上的caffe框架
git clone https://github.com/weiliu89/caffe
cd caffe
git checkout ssd

或者可以直接一步到位

git clone https://github.com/weiliu89/caffe -b ssd && cd caffe

下载Caffe SSD框架

  1. 安装编译工具链
    Ubuntu 20.04默认的gcc等编译工具版本较高，需要降级到7或以下，否则会报编译工具版本过高的错误。
sudo apt install -y build-essential gcc-7 g++-7
sudo update-alternatives --install /usr/bin/gcc gcc /usr/bin/gcc-7 10
sudo update-alternatives --install /usr/bin/g++ g++ /usr/bin/g++-7 10
  1. 安装依赖
conda install -y cmake boost libboost boost-cpp glog gflags protobuf hdf5 lmdb leveldb libopencv scikit-image
sudo apt install -y nvidia-cuda-toolkit-gcc libatlas-base-dev
  1. 编译pycaffe
mkdir build
cd build
cmake  ..  -DPYTHON_EXECUTABLE=`which python`
make -j

执行cmake后结果
执行make后最终结果

  1. 以Pascal VOC数据集为例
    注：[/path/to/caffe]代表第二步下载的caffe文件夹，注意替换。
# 下载模型
mkdir -p [/path/to/caffe]/models/VGGNet && cd [/path/to/caffe]/models/VGGNet
wget https://github.com/conner99/VGGNet/raw/master/VGG_ILSVRC_16_layers_fc_reduced.caffemodel

# 下载数据集
mkdir $HOME/data && cd $HOME/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

# 解压
tar -xvf VOCtrainval_11-May-2012.tar
tar -xvf VOCtrainval_06-Nov-2007.tar
tar -xvf VOCtest_06-Nov-2007.tar
  1. 创建数据集的LMDB文件
cd [/path/to/caffe]
bash ./data/VOC0712/create_list.sh
PYTHONPATH=$PWD/python bash ./data/VOC0712/create_data.sh

create_list.sh结果图
create_data.sh效果图

  1. 需要修改的地方

    1. 使用的GPU
      如果只有一个GPU就改成gpus = "0"
      修改使用的GPU编号

    2. 调整batch_size（可选）
      调整batch_size（如果出现Aborted (core dumped)可能是因为batch_size较大）

  2. 训练
    首先需要下载模型文件

mkdir -p [/path/to/caffe]/models/VGGNet
wget -O [/path/to/caffe]/models/VGGNet/VGG_ILSVRC_16_layers_fc_reduced.caffemodel https://github.com/conner99/VGGNet/raw/master/VGG_ILSVRC_16_layers_fc_reduced.caffemodel

PYTHONPATH=$PWD/python python examples/ssd/ssd_pascal.py
  1. Train
PYTHONPATH=$PWD/python python examples/ssd/ssd_pascal.py
  1. Evaluate
PYTHONPATH=$PWD/python python examples/ssd/score_ssd_pascal.py
  1. Example
PYTHONPATH=$PWD/python python examples/ssd/ssd_detect.py --image_file=/home/yinziming/Downloads/a.jpg

运行效果图

这篇关于Ubuntu 20.04下通过Anaconda配置安装Caffe SSD记录的文章就介绍到这儿，希望我们推荐的文章对大家有所帮助，也希望大家多多支持为之网！
您可能喜欢
栏目导航