caffe添加自己的数据输入层

整体思路:

  1. 阅读caffe数据输入层各个类之间的继承关系,确定当前类需要继承的父类以及所需参数的设置。
  2. 编写zzq_data.cpp
  3. 在layer_factory.cpp中完成注册;
  4. 在caffe.proto中声明参数类型;
  5. 编译。

继承关系:

继承自基类的成员变量:
protected: Blob<Dtype> prefetch_data_; Blob<Dtype> prefetch_label_; Blob<Dtype> transformed_data_;
用于保存预读取的数据,标签,以及转换过的数据。

具体操作参考:https://blog.csdn.net/tsb831211/article/details/52760240
好累,不想写了。。。

posted @ 2018-05-17 13:33  恩zzq我是  阅读(688)  评论(0编辑  收藏  举报