fcn网络训练代码_FCN网络训练 菜鸟版

本文详细介绍了如何在Ubuntu 14.04环境下,使用Caffe训练全卷积网络FCN。从下载Caffe和FCN代码开始,通过编译Caffe、设置PYTHONPATH、下载数据集、预训练模型,到修改solver.prototxt和solve.py文件,以及最终的模型训练和单张图片测试。过程中特别提到了针对不同FCN模型版本权重移植的注意事项。
摘要由CSDN通过智能技术生成

论文全名:Fully Convolutional Networks for Semantic Segmentation

全卷积神经网络 FCN代码运行详解:

运行平台:

Ubuntu 14.04 + cudnn7

1.代码准备:

下载新版caffe:   https://github.com/BVLC/caffe

下载fcn代码:      https://github.com/shelhamer/fcn.berkeleyvision.org

将下载后的压缩包解压 将解压后的代码丢进服务器

2.编辑caffe和python接口:

cd进入caffe所在路径

以个人路径示例:/home/my/caffe-master/caffe-master

在该目录下,执行

cp Makefile.config.example Makefile.config

vim Makefile.config(这句代码根据自己情况选择,如果需要修改相关设定,就使用这句,需要注意的是,将WITH_PYTHON_LAYER :=1前面的#去掉

如果使用cudnn,就把use cudann前面的#去掉

我这边安装的是openbla,所以我的设置为BLAS:=open)

make all -j8 //8代表线程数量,可以加快编译速度

make test -j8 //编译测试需要的文件

make runtest //开始运行测试例子,这一句貌似有没有都行以上结束后开始编译python接口

在下载的caffe源码包中,有个叫python的文件夹,cd进入这个文件夹

以我个人路径为例cd至 /home/wangkun/caffe-master/caffe-master/python

编译python:

make pycaffe

#测试是否成功

cd caffe-folder/python

python

import caffe

#如果上述命令未报错,说明成功

#添加caffe/python 到python path变量

vim ~/.bashrc

#set the caffe PYTHONPATH

export PYTHONPATH=/path/to/caffe/python:$PYTHONPATH

!如果是采用的Anaconda python,切记编译python接口在最后执行。

!如果是系统自带的python在make pycaffe可能出现python/caffe/_caffe.hpp:8:31: fatal error: numpy/arra

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值