docker训练caffe的SSD

博客介绍了SSD训练的相关步骤,包括使用git克隆caffe仓库、编译caffe时遇到的错误及解决方法,还提到放置VOC07格式数据集并按指定链接修改文件,给出预训练模型链接,最后说明了训练所需的参数,如学习率、批量大小等。

硬件配置:GTX 1080,

docker 的镜像配置是CUDA8,,在哪里看到说gtx1080不能用CUDA7.5及以下,todo:找出那个链接

镜像地址为docker pull xiaoxinyi/caffe-gpu:8.23

~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~SSD训练

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

cd caffe之后,git checkout ssd

2.然后make -j8 && make pycaffe

会出现两个错误:

(1)解决caffe make报错:/usr/include/boost/property_tree/detail/json_parser_read.hpp:257:264: error: ‘type name’ declared as function returning an array:

解决方法:https://blog.youkuaiyun.com/allyli0022/article/details/62881238

(2)安装sudo apt-get install libatlas-base-dev(不然编译caffe时候出错/usr/bin/ld:cannot find -lcblas和latlas

3.放VOC07格式数据集,并修改文件按照https://blog.youkuaiyun.com/lilai619/article/details/53791420

(1)预训练模型:链接: https://pan.baidu.com/s/1IzLh-YcW52tseZUldcEEhg 提取码: rprn 

(2)修改create_list.sh和create_data的时候,$HOME改成caffe所在路径

(2)在score_ssd_pascal.py/ssd_pascal_webcam.py/ssd_pascal_video.py等对应脚本中添加                sys.path.insert(0,'/home/xxx/caffe/python'),,,,否则create_data.sh的时候报错放

(3.按3.链接继续修改,运行:python ./examples/ssd/ssd_pascal.py 2>&1 | tee ssd_train_log.txt 开始训练

------------------------------------------------------------------------------------------------

参数:

学习率base_lr,  batch_size ,max_iteration, snapshot(保存间隔)

save_dir(模型保存路径)

=================================

第二步的报上有人说升级gcc,升级后还是报错gcc version later than 5.3 are not supported;不过按上面的链接修改/usr/include/boost/property_tree/detail/json_parser_read.hpp就可以了

 

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值