caffe修改hdf5的datalayer

最近用caffe做回归,用到hdf5类型数据,由于要对数据做argument,所以自己继承了一个hdf5的类。

可以参考 http://blog.csdn.net/kuaitoukid/article/details/41865803(这里是新建一个layer,我这里是继承一个)

1.自己新建一个类继承hdf5   

class HDF5DataLDLayer : public HDF5DataLayer<Dtype>

2.同时复制相对应的cu文件和cpp文件,进行自己需要的argument修改(我把成员变量和函数都复制下来了,然后添加了自己需要的成员变量,修改了部分成员函数的实现)。

3.在proto文件使用对应的layer就好了

layer {
name: "data"
type: "HDF5DataLD"
top: "data"
top: "label"
include {
phase: TRAIN
}
}

  

posted @ 2016-03-24 14:50  kisstea  阅读(483)  评论(0编辑  收藏  举报