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
}
}

转载于:https://www.cnblogs.com/kisstea/p/5315544.html

caffe修改hdf5的datalayer相关推荐

caffe中HDF5层及数据生成
HDF 文件结构包括一个file id(文件号).至少一个 data descriptor (数据描述符).没有或多个 data element(数据内容)数据内容. file id (文件号)是一个 ...
caffe修改需要的东西 6：40
https://blog.csdn.net/zhaishengfu/article/details/51971768?locationNum=3&fps=1
Ubuntu17.10 下配置caffe 仅CPU i386可以直接apt install caffe-cpu，但是怎么运行mnist代码我懵逼了...
Ubuntu16.04下配置caffe(仅CPU) 参考:http://blog.csdn.net/zt_1995/article/details/56283249 第二次配置caffe环境,依旧把之 ...
fatal error: hdf5.h: No such file or directory
在编译caffe的时候,可能会出现各种找不到文件的情况.这种情况的一种解决方式是改变Makefile.config中的路径. 比如,在编译caffe的时候提醒找不到hdf5.h或者hdf5_hl.h ...
深度学习框架caffe及py-faster-rcnn详细配置安装过程
深度学习框架caffe及py-faster-rcnn详细配置安装过程配置环境: ubuntu14.04 CUDA7.5 1.准备工作安装vim.python-pip.git sudo apt-ge ...
安装caffe碰到的坑
解决办法是依据出现错误的顺序而给出的,为了方便,可以直接先执行所有解决办法后再安装caffe. 1. ./include/caffe/common.hpp:5:27: fatal error: gfl ...
Caffe+Kubuntu16.04_X64+CUDA 8.0配置
前言: 经过尝试过几次Caffe,theano,MxNet之后,很长时间没有进行caffe的更新,此次在Ubuntu16.04下安装Caffe,折腾了一天时间,终于安装成功. 参考链接:Caffe+U ...
【caffe-windows】Linux至Windows平台的caffe移植
1.前言主要参考两篇博客以及很多论坛解决细节问题: http://www.cnblogs.com/trantor/p/4570097.html https://initialneil.wordpre ...
20160512关于mac安装caffe的记录
记得2015年在mac系统上安装过一次caffe,非常顺利,但是最近群里许多同学反映mac安装caffe出现了各种问题,同时我也在帮助别人安装caffe的时候也遇到了一些坑,不再像以前这么顺利了.估计 ...

caffe修改hdf5的datalayer

caffe修改hdf5的datalayer相关推荐

最新文章

热门文章