添加caffe layers

最新推荐文章于 2024-10-07 16:36:30 发布

TonyYang1995

最新推荐文章于 2024-10-07 16:36:30 发布

阅读量547

点赞数

分类专栏： caffe 文章标签： caffe c++

本文链接：https://blog.csdn.net/TonyYang1995/article/details/51848492

版权

caffe 的C++ 接口对于train 非常的方便，但是对于perdict 而言就不太友好，我们只能看到loss 。但是对于multilabel 问题，比如输出的perdict 和label 是一副图片的话，用c++ 就不太方便，所以我们要在caffe layers 中添加一些自己的layer 来实现可视化。

在一开始添加的时候，会遇到各种各样的bug，这个mark 下以备不时之需。

比如说，我要添加一个visual layer，把perdict 的图片输出出来。
在caffe/include/caffe/layers/ 下添加visual_layer.hpp
在 caffe/src/caffe/layers/ 下添加visual_layer.cpp
修改 caffe/src/proto 的caffe.proto 下修改相应的属性。

message LayerParameter {
...
optional VisualParameter visual_param = 1000; 这里添加的是caffe 的type，数字在message 上有题型下一个可用的数字是多少，这个因人而异
}

....
添加多一个message
message VisualParameter {
  optional string save_folder = 1;
  optional uint32 save_interval = 2 [default = 2];
  optional float scale = 3 [default = 1];
}

visual_layer.hpp

#ifndef CAFFE_VISUAL_LAYER_HPP_
#define CAFFE_VISUAL_LAYER_HPP_

#include <stdio.h>
#include <vector>

#include <opencv2/core/core.hpp>
#include <opencv2/highgui/highgui.hpp>
#include <opencv2/highgui/highgui_c.h>
#include <opencv2/imgproc/imgproc.hpp>

#include "caffe/blob.hpp"
#include "caffe/layer.hpp"
#include "caffe/proto/caffe.pb.h"

#include "caffe/layers/loss_layer.hpp"

namespace caffe {


template <typename Dtype>
class VisualLayer : public Layer<Dtype> {
 public:

  explicit VisualLayer(const LayerParameter& param)
      : Layer<Dtype>(param) {}
  virtual void LayerSetUp(const vector<Blob<Dtype>*>& bottom,
      const vector<Blob<Dtype>*>& top);
  virtual void Reshape(const vector<Blob<Dtype>*>& bottom,
      const vector<Blob<Dtype>*>& top) {
      //vector<int> top_shape(0); 
      //top[0]->Reshape(top_shape);
  };

  virtual inline const char