caffe GPU版本使用fcn分割自己的图像

最新推荐文章于 2024-07-12 15:12:25 发布

只要简单点

最新推荐文章于 2024-07-12 15:12:25 发布

阅读量4.2k

点赞数 3

分类专栏：总结文章标签： caffe fcn 数据集 label

本文链接：https://blog.csdn.net/a244513086/article/details/72520630

版权

本文记录了使用Caffe-GPU版本的FCN进行图像分割的全过程，包括环境配置、数据集制作、模型训练和图像分割。重点介绍了如何修改prototxt文件以适应自定义数据集，以及解决训练过程中遇到的问题。

摘要由CSDN通过智能技术生成

折腾好久终于把caffe的fcn图像分割调通，在此记录一下整个过程

非常感谢supe_king的几篇博客http://blog.csdn.net/supe_king/article/details/54142973

下面介绍正题

使用caffe-gpu版本需要先安装python2.7或者python3.5，安装cuda7.5或者cuda8.0，最后相应版本的cudnn，安装方法为：
将cudnn解压到cuda对应的目录中，cuda目录加入环境变量

caffe制作自己的数据集

1 训练集

fcn.berkeleyvision.org-master\data\sbdd\dataset\img\    原始图片
fcn.berkeleyvision.org-master\data\sbdd\dataset\cls\    label图片
fcn.berkeleyvision.org-master\data\sbdd\dataset\train.txt   图片名称列表

2 测试集

fcn.berkeleyvision.org-master\data\pascal\VOC2011\JPEGImages\ 测试用原始图片
fcn.berkeleyvision.org-master\data\pascal\VOC2011\SegmentationClass\ label图片
fcn.berkeleyvision.org-master\data\pascal\VOC2011\ImageSets\Se