欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

caffe+SSD+Ubuntu16.04+CPU的配置及运行

程序员文章站 2024-03-17 11:28:40
...

大致过程按照GitHub教程https://github.com/weiliu89/caffe/tree/ssd翻译而来,亲测可行
1、在home目录下,获取SSD的代码,下载完成有一个caffe的文件夹

 git clone https://github.com/weiliu89/caffe.git
 cd caffe
 git checkout ssd

2、进入下载好的caffe目录,复制配置文件。如果你安装caffe目录下的配置文件已经配好,可以将其复制过来

cd   /home/xxx/caffe
cp  Makefile.config.example  Makefile.config这里写代码片

3、编译caffe三部曲
其中-j8中的8代表虚拟机的处理器配置(我这里是2)
查看命令:#cat /proc/cpuinfo |grep “cores”|uniq

make -j2   
make test -j2
make runtest -j2

4、下载预训练模型fully convolutional reduced (atrous) VGGNet. 将它放入caffe/models/VGGNet目录下
5、下载VOC2007和VOC2012数据集,放在/home/data下

cd ..
mkdir data
cd data/

6、下载数据集

 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

7 、数据集解压,建议使用命令进行解压,手动解压可能会出现各种问题

tar -xvf VOCtrainval_11-May-2012.tar
tar -xvf VOCtrainval_06-Nov-2007.tar
tar -xvf VOCtest_06-Nov-2007.tar

8、将图片转化为LMDB文件,用于训练

 cd ..
 cd caffe/
 ./data/VOC0712/create_list.sh  
 ./data/VOC0712/create_data.sh

如果出现错误的话,查看你的python路径是否配置正确

echo "export PYTHONPATH=/home/xxx/caffe_root/python" >> ~/.profile  
source ~/.profile  
echo $PYTHONPATH #检查环境变量的值

caffe_root指的是你的ssd_caffe项目的路径
9、训练模型

python examples/ssd/ssd_pascal.py
相关标签: ssd

上一篇: SSD网络中的L2标准化

下一篇: