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_;
用于保存预读取的数据,标签,以及转换过的数据。

猜你喜欢

转载自www.cnblogs.com/zzq-123456/p/9050617.html
今日推荐