caffe

1
classdef Layer < handle
2
  % Wrapper class of caffe::Layer in matlab
3
  
4
  properties (Access = private)
5
    hLayer_self
6
    attributes
7
    % attributes fields:
8
    %     hBlob_blobs
9
  end
10
  properties (SetAccess = private)
11
    params
12
  end
13
  
14
  methods
15
    function self = Layer(hLayer_layer)
16
      CHECK(is_valid_handle(hLayer_layer), 'invalid Layer handle');
17
      
18
      % setup self handle and attributes
19
      self.hLayer_self = hLayer_layer;
20
      self.attributes = caffe_('layer_get_attr', self.hLayer_self);
21
      
22
      % setup weights
23
      self.params = caffe.Blob.empty();
24
      for n = 1:length(self.attributes.hBlob_blobs)
25
        self.params(n) = caffe.Blob(self.attributes.hBlob_blobs(n));
26
      end
27
    end
28
    function layer_type = type(self)
29
      layer_type = caffe_('layer_get_type', self.hLayer_self);
30
    end
31
  end
32
end
33