最近用caffe做回歸，用到hdf5類型數(shù)據(jù)，由于要對(duì)數(shù)據(jù)做argument，所以自己繼承了一個(gè)hdf5的類。

可以參考?http://blog.csdn.net/kuaitoukid/article/details/41865803（這里是新建一個(gè)layer，我這里是繼承一個(gè)）

1.自己新建一個(gè)類繼承hdf5 ??

class HDF5DataLDLayer : public HDF5DataLayer<Dtype>

2.同時(shí)復(fù)制相對(duì)應(yīng)的cu文件和cpp文件，進(jìn)行自己需要的argument修改（我把成員變量和函數(shù)都復(fù)制下來(lái)了，然后添加了自己需要的成員變量，修改了部分成員函數(shù)的實(shí)現(xiàn)）。

3.在proto文件使用對(duì)應(yīng)的layer就好了

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

??

轉(zhuǎn)載于:https://www.cnblogs.com/kisstea/p/5315544.html

總結(jié)

以上是生活随笔為你收集整理的caffe修改hdf5的datalayer的全部?jī)?nèi)容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網(wǎng)站內(nèi)容還不錯(cuò)，歡迎將生活随笔推薦給好友。