FCN语义分割——直接加载图像数据
来源:互联网 发布:男士护肤 知乎 编辑:程序博客网 时间:2024/06/01 07:22
前言
在之前的一篇博客中我们使用了在Github上的代码进行语义分割,在进行训练的时候Github上给出的是mat文件类型的图像数据,这就很不方便了。这里经过查阅相关资料之后发现不需要进行格式的转换也可以进行训练的。
修改过程
这里还是用之前的博客里面提到的voc-fcn8s。打开train.prototxt和val.prototxt之后可以看到这样的层定义
layer { name: "data" type: "Python" top: "data" top: "label" python_param { module: "voc_layers" layer: "SBDDSegDataLayer" param_str: "{\'sbdd_dir\': \'../data/sbdd/dataset\', \'seed\': 1337, \'split\': \'train\', \'mean\': (104.00699, 116.66877, 122.67892)}" }}
里面的module指明了使用的python文件,也就是用来进行数据导入的。打开它我们可以看到在类SBDDSegDataLayer的成员函数load_label使用到了mat文件,如下
def load_label(self, idx): """ Load label image as 1 x height x width integer array of label indices. The leading singleton dimension is required by the loss. """ import scipy.io mat = scipy.io.loadmat('{}/cls/{}.mat'.format(self.sbdd_dir, idx)) label = mat['GTcls'][0]['Segmentation'][0].astype(np.uint8) label = label[np.newaxis, ...] return label
将其修改为
def load_label(self, idx): """ Load label image as 1 x height x width integer array of label indices. The leading singleton dimension is required by the loss. """ im = Image.open('{}/cls/{}.png'.format(self.sbdd_dir, idx)) label = np.array(im, dtype=np.uint8) label = label[np.newaxis, ...] return label
就可以直接进行加载图像数据了。
ref: caffe下fcn数据集制作的简化
PS 之前应为没有注意好图像的通道问题导致了CUBLAS_STATUS_MAPING_ERROR的问题
这里是因为本人私自加了im= im.convert(‘L’)。导致参数不匹配,出现错误,将其删除之后正常运行。类似的出现这种错误的原因主要为输入参数范围不匹配等造成的。
阅读全文
0 0
- FCN语义分割——直接加载图像数据
- FCN图像语义分割
- FCN语义分割——训练自己的数据
- 图像语义分割之FCN和CRF
- 图像语义分割之FCN和CRF
- 使用FCN做图像语义分割
- 图像语义分割之FCN和CRF
- 图像语义分割之FCN和CRF
- 图像语义分割之FCN和CRF
- 图像语义分割(1)- FCN
- 图像语义分割之FCN和CRF
- 图像语义分割之FCN和CRF
- 图像语义分割之FCN和CRF
- 图像语义分割之FCN和CRF
- keras版FCN网络进行图像语义分割--使用VOC2012数据
- fcn语义分割 论文
- FCN语义分割
- 使用FCN做图像语义分割(实践篇)
- Jzoj4746 树塔狂想曲
- HDU 1572 下沙小面的(2)
- C++ 严格弱序
- Spring的事务管理
- 正向代理、反向代理
- FCN语义分割——直接加载图像数据
- ios-GCD信号量
- Android超酷的加载效果
- Win10系统如何在右键菜单添加"管理员取得所有权
- OPENMV结合PIX飞控实现定点 循迹
- 进程简述
- python反射
- Class文件中的常量池详解(上)
- Django的基本操作流程总结