caffe添加自己的数据输入层
整体思路:
- 阅读caffe数据输入层各个类之间的继承关系,确定当前类需要继承的父类以及所需参数的设置。
- 编写zzq_data.cpp
- 在layer_factory.cpp中完成注册;
- 在caffe.proto中声明参数类型;
- 编译。
继承关系:
继承自基类的成员变量:
protected: Blob<Dtype> prefetch_data_; Blob<Dtype> prefetch_label_; Blob<Dtype> transformed_data_;
用于保存预读取的数据,标签,以及转换过的数据。
具体操作参考:https://blog.csdn.net/tsb831211/article/details/52760240
好累,不想写了。。。
CV小蜡肉