ubuntu16.04 cuda8.0 opencv3.1 gtx1080 配置ssd

来源:互联网 发布:面试测评软件系统 编辑:程序博客网 时间:2024/05/20 06:29

1.git clone https://github.com/weiliu89/caffe.git

可能会报错,原因是没有安装git

解决方法:sudo apt install git

继续运行git clone https://github.com/weiliu89/caffe.git

cd caffe

git checkout ssd(出现“分支”则说明copy-check成功)

2.配置ssd(caffe)

将配置caffe成功的Makefile.config复制过来放在caffe目录下即可

配置caffe过程可以参考我的博客

cd caffe

make all -j16

make test -j16

make pycaffe(编译pycaffe)

成功即可

3、 预训练模型下载

下载地址:链接:http://pan.baidu.com/s/1slpaEO9 密码:loxo

来源于博客 http://blog.csdn.net/samylee/article/details/51822832

在caffe/models文件夹下新建文件夹,命名为VGGNet,将刚刚下载下来的文件放入这个VGGNet文件夹当中

4、下载VOC2007和VOC2012数据集

在主文件夹下(即/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
解压这三个文件,终端输入:
cd /home/**(您服务器的名字)/data
tar -xvf VOCtrainval_11-May-2012.tar
tar -xvf VOCtrainval_06-Nov-2007.tar
tar -xvf VOCtest_06-Nov-2007.tar
(请严格按照这个顺序解压)


接着终端输入:
cd /home/**(您服务器的名字)/caffe
./data/VOC0712/create_list.sh
./data/VOC0712/create_data.sh

在运行第三步时如果出现AttributeError: 'module' object has no attribute 'LabelMap',则在终端输入:

gedit ~/.bashrc

在后面添加

export PYTHONPATH=/home/服务器名字/caffe/python:$PYTHONPATH

接着运行source ~/.bashrc即可

可能会出现一些相似错误例如ImportError: No module named...

解决方法:缺少什么查询之后用apt-get  安装即可

然后再次运行第三步

训练:

原作者使用的是多个gpu

如果你是单个gpu的话需要修改example/ssd/ssd_pascal.py文件中的"gpus=0,1,2,3"(原代码)

改为gpu=0即可 

如果gpu性能不是很好的话可能会出错Check failed: error == cudaSuccess (2 vs. 0)  out of memory

只需要减小其中的batch_size即可开始训练。

测试:

网上有代码可以自己查找


阅读全文
0 0
原创粉丝点击