caffe 0.1.0

data/ext/caffe/net.cc

@@ -0,0 +1,55 @@
+#include "net.hpp"
+#include "common.hpp"
+#include "blob.hpp"
+#include "util.hpp"
+#include <rice/Data_Type.hpp>
+#include <rice/Constructor.hpp>
+#include <rice/Module.hpp>
+#include <rice/String.hpp>
+#include <rice/Array.hpp>
+
+using namespace Rice;
+
+static Array getInputs(Object self) {
+    Net *net = from_ruby<Net *>(self);
+    const std::vector<Blob *> &vec = net -> input_blobs();
+    return mapArray(vec.begin(), vec.end(), objectNoGC<Blob>);
+}
+
+static Array getOutputs(Object self) {
+    Net *net = from_ruby<Net *>(self);
+    const std::vector<Blob *> &vec = net -> output_blobs();
+    return mapArray(vec.begin(), vec.end(), objectNoGC<Blob>);
+}
+
+static Object getBlobByName(Object self, String name) {
+    Net *net = from_ruby<Net *>(self);
+    Blob *blob = net -> blob_by_name(from_ruby<std::string>(name)).get();
+
+    if (blob) {
+        return objectNoGC(blob);
+    } else {
+        return Qnil;
+    }
+}
+
+static Object forward(Object self) {
+    Net *net = from_ruby<Net *>(self);
+    float loss = .0;
+    net -> Forward(NULL);
+    return to_ruby(loss);
+}
+
+void Init_net() {
+    Module rb_mCaffe = define_module("Caffe");
+
+    Data_Type<Net> rb_cNet = rb_mCaffe
+        .define_class<Net>("Net")
+        .define_constructor(Constructor<Net, std::string, caffe::Phase>())
+        .define_method("inputs", &getInputs)
+        .define_method("outputs", &getOutputs)
+        .define_method("blob", &getBlobByName)
+        .define_method("reshape!", &Net::Reshape)
+        .define_method("load_trained!", &Net::CopyTrainedLayersFromBinaryProto)
+        .define_method("forward!", &forward);
+}

data/ext/caffe/net.hpp

@@ -0,0 +1,10 @@
+#ifndef __NET
+#define __NET
+
+#include <caffe/caffe.hpp>
+
+typedef caffe::Net<float> Net;
+
+void Init_net(void);
+
+#endif

data/ext/caffe/util.hpp

@@ -0,0 +1,39 @@
+#ifndef __UTIL
+#define __UTIL
+
+#include <rice/Data_Type.hpp>
+#include <rice/Array.hpp>
+#include <vector>
+
+template<typename T>
+struct EmptyFreeFunction {
+    static void free(T *obj) {}
+};
+
+template<typename T>
+Rice::Data_Object<T> objectNoGC(T *obj) {
+    return Rice::Data_Object<T>(obj, Rice::Data_Type<T>::klass(),
+                                Rice::Default_Mark_Function<T>::mark,
+                                EmptyFreeFunction<T>::free);
+}
+
+template<typename Iter, typename Func>
+Rice::Array mapArray(Iter begin, Iter end, Func func) {
+    Rice::Array ret;
+    for (; begin != end; ++begin) {
+        ret.push(to_ruby(func(*begin)));
+    }
+    return ret;
+}
+
+template<typename T>
+std::vector<T> arrayToVector(Rice::Array arr) {
+    int n = arr.size();
+    std::vector<T> ret(n);
+    for (int i = 0; i < n; ++i) {
+        ret[i] = from_ruby<T>(arr[i]);
+    }
+    return ret;
+}
+
+#endif

data/lib/caffe.rb

@@ -0,0 +1,5 @@
+require 'caffe/version'
+require 'caffe/caffe.pb'
+require 'caffe/caffe'
+require 'caffe/common'
+require 'caffe/blob'

data/lib/caffe/blob.rb

@@ -0,0 +1,35 @@
+module Caffe
+  # Wrapper class for caffe::Blob<float>
+  class Blob
+    # Proxy class to access caffe::Blob<float> like Array / Enumerable
+    class Cursor
+      include Enumerable
+
+      alias count size
+    end
+
+    def [](index)
+      data[index]
+    end
+
+    def []=(index, x)
+      data[index] = x
+    end
+
+    def size
+      data.size
+    end
+
+    def copy_from!(x)
+      data.copy_from! x
+    end
+
+    def each(&blk)
+      data.each(&blk)
+    end
+
+    include Enumerable
+
+    alias count size
+  end
+end

data/lib/caffe/caffe.pb.rb

@@ -0,0 +1,902 @@
+# encoding: utf-8
+
+##
+# This file is auto-generated. DO NOT EDIT!
+#
+require 'protobuf/message'
+
+module Caffe
+
+  ##
+  # Enum Classes
+  #
+  class Phase < ::Protobuf::Enum
+    define :TRAIN, 0
+    define :TEST, 1
+  end
+
+
+  ##
+  # Message Classes
+  #
+  class BlobShape < ::Protobuf::Message; end
+  class BlobProto < ::Protobuf::Message; end
+  class BlobProtoVector < ::Protobuf::Message; end
+  class Datum < ::Protobuf::Message; end
+  class FillerParameter < ::Protobuf::Message
+    class VarianceNorm < ::Protobuf::Enum
+      define :FAN_IN, 0
+      define :FAN_OUT, 1
+      define :AVERAGE, 2
+    end
+
+  end
+
+  class NetParameter < ::Protobuf::Message; end
+  class SolverParameter < ::Protobuf::Message
+    class SnapshotFormat < ::Protobuf::Enum
+      define :HDF5, 0
+      define :BINARYPROTO, 1
+    end
+
+    class SolverMode < ::Protobuf::Enum
+      define :CPU, 0
+      define :GPU, 1
+    end
+
+    class SolverType < ::Protobuf::Enum
+      define :SGD, 0
+      define :NESTEROV, 1
+      define :ADAGRAD, 2
+      define :RMSPROP, 3
+      define :ADADELTA, 4
+      define :ADAM, 5
+    end
+
+  end
+
+  class SolverState < ::Protobuf::Message; end
+  class NetState < ::Protobuf::Message; end
+  class NetStateRule < ::Protobuf::Message; end
+  class ParamSpec < ::Protobuf::Message
+    class DimCheckMode < ::Protobuf::Enum
+      define :STRICT, 0
+      define :PERMISSIVE, 1
+    end
+
+  end
+
+  class LayerParameter < ::Protobuf::Message; end
+  class TransformationParameter < ::Protobuf::Message; end
+  class LossParameter < ::Protobuf::Message
+    class NormalizationMode < ::Protobuf::Enum
+      define :FULL, 0
+      define :VALID, 1
+      define :BATCH_SIZE, 2
+      define :NONE, 3
+    end
+
+  end
+
+  class AccuracyParameter < ::Protobuf::Message; end
+  class ArgMaxParameter < ::Protobuf::Message; end
+  class ConcatParameter < ::Protobuf::Message; end
+  class BatchNormParameter < ::Protobuf::Message; end
+  class BiasParameter < ::Protobuf::Message; end
+  class ContrastiveLossParameter < ::Protobuf::Message; end
+  class ConvolutionParameter < ::Protobuf::Message
+    class Engine < ::Protobuf::Enum
+      define :DEFAULT, 0
+      define :CAFFE, 1
+      define :CUDNN, 2
+    end
+
+  end
+
+  class CropParameter < ::Protobuf::Message; end
+  class DataParameter < ::Protobuf::Message
+    class DB < ::Protobuf::Enum
+      define :LEVELDB, 0
+      define :LMDB, 1
+    end
+
+  end
+
+  class DropoutParameter < ::Protobuf::Message; end
+  class DummyDataParameter < ::Protobuf::Message; end
+  class EltwiseParameter < ::Protobuf::Message
+    class EltwiseOp < ::Protobuf::Enum
+      define :PROD, 0
+      define :SUM, 1
+      define :MAX, 2
+    end
+
+  end
+
+  class ELUParameter < ::Protobuf::Message; end
+  class EmbedParameter < ::Protobuf::Message; end
+  class ExpParameter < ::Protobuf::Message; end
+  class FlattenParameter < ::Protobuf::Message; end
+  class HDF5DataParameter < ::Protobuf::Message; end
+  class HDF5OutputParameter < ::Protobuf::Message; end
+  class HingeLossParameter < ::Protobuf::Message
+    class Norm < ::Protobuf::Enum
+      define :L1, 1
+      define :L2, 2
+    end
+
+  end
+
+  class ImageDataParameter < ::Protobuf::Message; end
+  class InfogainLossParameter < ::Protobuf::Message; end
+  class InnerProductParameter < ::Protobuf::Message; end
+  class InputParameter < ::Protobuf::Message; end
+  class LogParameter < ::Protobuf::Message; end
+  class LRNParameter < ::Protobuf::Message
+    class NormRegion < ::Protobuf::Enum
+      define :ACROSS_CHANNELS, 0
+      define :WITHIN_CHANNEL, 1
+    end
+
+    class Engine < ::Protobuf::Enum
+      define :DEFAULT, 0
+      define :CAFFE, 1
+      define :CUDNN, 2
+    end
+
+  end
+
+  class MemoryDataParameter < ::Protobuf::Message; end
+  class MVNParameter < ::Protobuf::Message; end
+  class ParameterParameter < ::Protobuf::Message; end
+  class PoolingParameter < ::Protobuf::Message
+    class PoolMethod < ::Protobuf::Enum
+      define :MAX, 0
+      define :AVE, 1
+      define :STOCHASTIC, 2
+    end
+
+    class Engine < ::Protobuf::Enum
+      define :DEFAULT, 0
+      define :CAFFE, 1
+      define :CUDNN, 2
+    end
+
+  end
+
+  class PowerParameter < ::Protobuf::Message; end
+  class PythonParameter < ::Protobuf::Message; end
+  class RecurrentParameter < ::Protobuf::Message; end
+  class ReductionParameter < ::Protobuf::Message
+    class ReductionOp < ::Protobuf::Enum
+      define :SUM, 1
+      define :ASUM, 2
+      define :SUMSQ, 3
+      define :MEAN, 4
+    end
+
+  end
+
+  class ReLUParameter < ::Protobuf::Message
+    class Engine < ::Protobuf::Enum
+      define :DEFAULT, 0
+      define :CAFFE, 1
+      define :CUDNN, 2
+    end
+
+  end
+
+  class ReshapeParameter < ::Protobuf::Message; end
+  class ScaleParameter < ::Protobuf::Message; end
+  class SigmoidParameter < ::Protobuf::Message
+    class Engine < ::Protobuf::Enum
+      define :DEFAULT, 0
+      define :CAFFE, 1
+      define :CUDNN, 2
+    end
+
+  end
+
+  class SliceParameter < ::Protobuf::Message; end
+  class SoftmaxParameter < ::Protobuf::Message
+    class Engine < ::Protobuf::Enum
+      define :DEFAULT, 0
+      define :CAFFE, 1
+      define :CUDNN, 2
+    end
+
+  end
+
+  class TanHParameter < ::Protobuf::Message
+    class Engine < ::Protobuf::Enum
+      define :DEFAULT, 0
+      define :CAFFE, 1
+      define :CUDNN, 2
+    end
+
+  end
+
+  class TileParameter < ::Protobuf::Message; end
+  class ThresholdParameter < ::Protobuf::Message; end
+  class WindowDataParameter < ::Protobuf::Message; end
+  class SPPParameter < ::Protobuf::Message
+    class PoolMethod < ::Protobuf::Enum
+      define :MAX, 0
+      define :AVE, 1
+      define :STOCHASTIC, 2
+    end
+
+    class Engine < ::Protobuf::Enum
+      define :DEFAULT, 0
+      define :CAFFE, 1
+      define :CUDNN, 2
+    end
+
+  end
+
+  class V1LayerParameter < ::Protobuf::Message
+    class LayerType < ::Protobuf::Enum
+      define :NONE, 0
+      define :ABSVAL, 35
+      define :ACCURACY, 1
+      define :ARGMAX, 30
+      define :BNLL, 2
+      define :CONCAT, 3
+      define :CONTRASTIVE_LOSS, 37
+      define :CONVOLUTION, 4
+      define :DATA, 5
+      define :DECONVOLUTION, 39
+      define :DROPOUT, 6
+      define :DUMMY_DATA, 32
+      define :EUCLIDEAN_LOSS, 7
+      define :ELTWISE, 25
+      define :EXP, 38
+      define :FLATTEN, 8
+      define :HDF5_DATA, 9
+      define :HDF5_OUTPUT, 10
+      define :HINGE_LOSS, 28
+      define :IM2COL, 11
+      define :IMAGE_DATA, 12
+      define :INFOGAIN_LOSS, 13
+      define :INNER_PRODUCT, 14
+      define :LRN, 15
+      define :MEMORY_DATA, 29
+      define :MULTINOMIAL_LOGISTIC_LOSS, 16
+      define :MVN, 34
+      define :POOLING, 17
+      define :POWER, 26
+      define :RELU, 18
+      define :SIGMOID, 19
+      define :SIGMOID_CROSS_ENTROPY_LOSS, 27
+      define :SILENCE, 36
+      define :SOFTMAX, 20
+      define :SOFTMAX_LOSS, 21
+      define :SPLIT, 22
+      define :SLICE, 33
+      define :TANH, 23
+      define :WINDOW_DATA, 24
+      define :THRESHOLD, 31
+    end
+
+    class DimCheckMode < ::Protobuf::Enum
+      define :STRICT, 0
+      define :PERMISSIVE, 1
+    end
+
+  end
+
+  class V0LayerParameter < ::Protobuf::Message
+    class PoolMethod < ::Protobuf::Enum
+      define :MAX, 0
+      define :AVE, 1
+      define :STOCHASTIC, 2
+    end
+
+  end
+
+  class PReLUParameter < ::Protobuf::Message; end
+
+
+  ##
+  # Message Fields
+  #
+  class BlobShape
+    repeated :int64, :dim, 1, :packed => true
+  end
+
+  class BlobProto
+    optional ::Caffe::BlobShape, :shape, 7
+    repeated :float, :data, 5, :packed => true
+    repeated :float, :diff, 6, :packed => true
+    repeated :double, :double_data, 8, :packed => true
+    repeated :double, :double_diff, 9, :packed => true
+    optional :int32, :num, 1, :default => 0
+    optional :int32, :channels, 2, :default => 0
+    optional :int32, :height, 3, :default => 0
+    optional :int32, :width, 4, :default => 0
+  end
+
+  class BlobProtoVector
+    repeated ::Caffe::BlobProto, :blobs, 1
+  end
+
+  class Datum
+    optional :int32, :channels, 1
+    optional :int32, :height, 2
+    optional :int32, :width, 3
+    optional :bytes, :data, 4
+    optional :int32, :label, 5
+    repeated :float, :float_data, 6
+    optional :bool, :encoded, 7, :default => false
+  end
+
+  class FillerParameter
+    optional :string, :type, 1, :default => "constant"
+    optional :float, :value, 2, :default => 0
+    optional :float, :min, 3, :default => 0
+    optional :float, :max, 4, :default => 1
+    optional :float, :mean, 5, :default => 0
+    optional :float, :std, 6, :default => 1
+    optional :int32, :sparse, 7, :default => -1
+    optional ::Caffe::FillerParameter::VarianceNorm, :variance_norm, 8, :default => ::Caffe::FillerParameter::VarianceNorm::FAN_IN
+  end
+
+  class NetParameter
+    optional :string, :name, 1
+    repeated :string, :input, 3
+    repeated ::Caffe::BlobShape, :input_shape, 8
+    repeated :int32, :input_dim, 4
+    optional :bool, :force_backward, 5, :default => false
+    optional ::Caffe::NetState, :state, 6
+    optional :bool, :debug_info, 7, :default => false
+    repeated ::Caffe::LayerParameter, :layer, 100
+    repeated ::Caffe::V1LayerParameter, :layers, 2
+  end
+
+  class SolverParameter
+    optional :string, :net, 24
+    optional ::Caffe::NetParameter, :net_param, 25
+    optional :string, :train_net, 1
+    repeated :string, :test_net, 2
+    optional ::Caffe::NetParameter, :train_net_param, 21
+    repeated ::Caffe::NetParameter, :test_net_param, 22
+    optional ::Caffe::NetState, :train_state, 26
+    repeated ::Caffe::NetState, :test_state, 27
+    repeated :int32, :test_iter, 3
+    optional :int32, :test_interval, 4, :default => 0
+    optional :bool, :test_compute_loss, 19, :default => false
+    optional :bool, :test_initialization, 32, :default => true
+    optional :float, :base_lr, 5
+    optional :int32, :display, 6
+    optional :int32, :average_loss, 33, :default => 1
+    optional :int32, :max_iter, 7
+    optional :int32, :iter_size, 36, :default => 1
+    optional :string, :lr_policy, 8
+    optional :float, :gamma, 9
+    optional :float, :power, 10
+    optional :float, :momentum, 11
+    optional :float, :weight_decay, 12
+    optional :string, :regularization_type, 29, :default => "L2"
+    optional :int32, :stepsize, 13
+    repeated :int32, :stepvalue, 34
+    optional :float, :clip_gradients, 35, :default => -1
+    optional :int32, :snapshot, 14, :default => 0
+    optional :string, :snapshot_prefix, 15
+    optional :bool, :snapshot_diff, 16, :default => false
+    optional ::Caffe::SolverParameter::SnapshotFormat, :snapshot_format, 37, :default => ::Caffe::SolverParameter::SnapshotFormat::BINARYPROTO
+    optional ::Caffe::SolverParameter::SolverMode, :solver_mode, 17, :default => ::Caffe::SolverParameter::SolverMode::GPU
+    optional :int32, :device_id, 18, :default => 0
+    optional :int64, :random_seed, 20, :default => -1
+    optional :string, :type, 40, :default => "SGD"
+    optional :float, :delta, 31, :default => 1e-08
+    optional :float, :momentum2, 39, :default => 0.999
+    optional :float, :rms_decay, 38, :default => 0.99
+    optional :bool, :debug_info, 23, :default => false
+    optional :bool, :snapshot_after_train, 28, :default => true
+    optional ::Caffe::SolverParameter::SolverType, :solver_type, 30, :default => ::Caffe::SolverParameter::SolverType::SGD
+    optional :bool, :layer_wise_reduce, 41, :default => true
+  end
+
+  class SolverState
+    optional :int32, :iter, 1
+    optional :string, :learned_net, 2
+    repeated ::Caffe::BlobProto, :history, 3
+    optional :int32, :current_step, 4, :default => 0
+  end
+
+  class NetState
+    optional ::Caffe::Phase, :phase, 1, :default => ::Caffe::Phase::TEST
+    optional :int32, :level, 2, :default => 0
+    repeated :string, :stage, 3
+  end
+
+  class NetStateRule
+    optional ::Caffe::Phase, :phase, 1
+    optional :int32, :min_level, 2
+    optional :int32, :max_level, 3
+    repeated :string, :stage, 4
+    repeated :string, :not_stage, 5
+  end
+
+  class ParamSpec
+    optional :string, :name, 1
+    optional ::Caffe::ParamSpec::DimCheckMode, :share_mode, 2
+    optional :float, :lr_mult, 3, :default => 1
+    optional :float, :decay_mult, 4, :default => 1
+  end
+
+  class LayerParameter
+    optional :string, :name, 1
+    optional :string, :type, 2
+    repeated :string, :bottom, 3
+    repeated :string, :top, 4
+    optional ::Caffe::Phase, :phase, 10
+    repeated :float, :loss_weight, 5
+    repeated ::Caffe::ParamSpec, :param, 6
+    repeated ::Caffe::BlobProto, :blobs, 7
+    repeated :bool, :propagate_down, 11
+    repeated ::Caffe::NetStateRule, :include, 8
+    repeated ::Caffe::NetStateRule, :exclude, 9
+    optional ::Caffe::TransformationParameter, :transform_param, 100
+    optional ::Caffe::LossParameter, :loss_param, 101
+    optional ::Caffe::AccuracyParameter, :accuracy_param, 102
+    optional ::Caffe::ArgMaxParameter, :argmax_param, 103
+    optional ::Caffe::BatchNormParameter, :batch_norm_param, 139
+    optional ::Caffe::BiasParameter, :bias_param, 141
+    optional ::Caffe::ConcatParameter, :concat_param, 104
+    optional ::Caffe::ContrastiveLossParameter, :contrastive_loss_param, 105
+    optional ::Caffe::ConvolutionParameter, :convolution_param, 106
+    optional ::Caffe::CropParameter, :crop_param, 144
+    optional ::Caffe::DataParameter, :data_param, 107
+    optional ::Caffe::DropoutParameter, :dropout_param, 108
+    optional ::Caffe::DummyDataParameter, :dummy_data_param, 109
+    optional ::Caffe::EltwiseParameter, :eltwise_param, 110
+    optional ::Caffe::ELUParameter, :elu_param, 140
+    optional ::Caffe::EmbedParameter, :embed_param, 137
+    optional ::Caffe::ExpParameter, :exp_param, 111
+    optional ::Caffe::FlattenParameter, :flatten_param, 135
+    optional ::Caffe::HDF5DataParameter, :hdf5_data_param, 112
+    optional ::Caffe::HDF5OutputParameter, :hdf5_output_param, 113
+    optional ::Caffe::HingeLossParameter, :hinge_loss_param, 114
+    optional ::Caffe::ImageDataParameter, :image_data_param, 115
+    optional ::Caffe::InfogainLossParameter, :infogain_loss_param, 116
+    optional ::Caffe::InnerProductParameter, :inner_product_param, 117
+    optional ::Caffe::InputParameter, :input_param, 143
+    optional ::Caffe::LogParameter, :log_param, 134
+    optional ::Caffe::LRNParameter, :lrn_param, 118
+    optional ::Caffe::MemoryDataParameter, :memory_data_param, 119
+    optional ::Caffe::MVNParameter, :mvn_param, 120
+    optional ::Caffe::ParameterParameter, :parameter_param, 145
+    optional ::Caffe::PoolingParameter, :pooling_param, 121
+    optional ::Caffe::PowerParameter, :power_param, 122
+    optional ::Caffe::PReLUParameter, :prelu_param, 131
+    optional ::Caffe::PythonParameter, :python_param, 130
+    optional ::Caffe::RecurrentParameter, :recurrent_param, 146
+    optional ::Caffe::ReductionParameter, :reduction_param, 136
+    optional ::Caffe::ReLUParameter, :relu_param, 123
+    optional ::Caffe::ReshapeParameter, :reshape_param, 133
+    optional ::Caffe::ScaleParameter, :scale_param, 142
+    optional ::Caffe::SigmoidParameter, :sigmoid_param, 124
+    optional ::Caffe::SoftmaxParameter, :softmax_param, 125
+    optional ::Caffe::SPPParameter, :spp_param, 132
+    optional ::Caffe::SliceParameter, :slice_param, 126
+    optional ::Caffe::TanHParameter, :tanh_param, 127
+    optional ::Caffe::ThresholdParameter, :threshold_param, 128
+    optional ::Caffe::TileParameter, :tile_param, 138
+    optional ::Caffe::WindowDataParameter, :window_data_param, 129
+  end
+
+  class TransformationParameter
+    optional :float, :scale, 1, :default => 1
+    optional :bool, :mirror, 2, :default => false
+    optional :uint32, :crop_size, 3, :default => 0
+    optional :string, :mean_file, 4
+    repeated :float, :mean_value, 5
+    optional :bool, :force_color, 6, :default => false
+    optional :bool, :force_gray, 7, :default => false
+  end
+
+  class LossParameter
+    optional :int32, :ignore_label, 1
+    optional ::Caffe::LossParameter::NormalizationMode, :normalization, 3, :default => ::Caffe::LossParameter::NormalizationMode::VALID
+    optional :bool, :normalize, 2
+  end
+
+  class AccuracyParameter
+    optional :uint32, :top_k, 1, :default => 1
+    optional :int32, :axis, 2, :default => 1
+    optional :int32, :ignore_label, 3
+  end
+
+  class ArgMaxParameter
+    optional :bool, :out_max_val, 1, :default => false
+    optional :uint32, :top_k, 2, :default => 1
+    optional :int32, :axis, 3
+  end
+
+  class ConcatParameter
+    optional :int32, :axis, 2, :default => 1
+    optional :uint32, :concat_dim, 1, :default => 1
+  end
+
+  class BatchNormParameter
+    optional :bool, :use_global_stats, 1
+    optional :float, :moving_average_fraction, 2, :default => 0.999
+    optional :float, :eps, 3, :default => 1e-05
+  end
+
+  class BiasParameter
+    optional :int32, :axis, 1, :default => 1
+    optional :int32, :num_axes, 2, :default => 1
+    optional ::Caffe::FillerParameter, :filler, 3
+  end
+
+  class ContrastiveLossParameter
+    optional :float, :margin, 1, :default => 1
+    optional :bool, :legacy_version, 2, :default => false
+  end
+
+  class ConvolutionParameter
+    optional :uint32, :num_output, 1
+    optional :bool, :bias_term, 2, :default => true
+    repeated :uint32, :pad, 3
+    repeated :uint32, :kernel_size, 4
+    repeated :uint32, :stride, 6
+    repeated :uint32, :dilation, 18
+    optional :uint32, :pad_h, 9, :default => 0
+    optional :uint32, :pad_w, 10, :default => 0
+    optional :uint32, :kernel_h, 11
+    optional :uint32, :kernel_w, 12
+    optional :uint32, :stride_h, 13
+    optional :uint32, :stride_w, 14
+    optional :uint32, :group, 5, :default => 1
+    optional ::Caffe::FillerParameter, :weight_filler, 7
+    optional ::Caffe::FillerParameter, :bias_filler, 8
+    optional ::Caffe::ConvolutionParameter::Engine, :engine, 15, :default => ::Caffe::ConvolutionParameter::Engine::DEFAULT
+    optional :int32, :axis, 16, :default => 1
+    optional :bool, :force_nd_im2col, 17, :default => false
+  end
+
+  class CropParameter
+    optional :int32, :axis, 1, :default => 2
+    repeated :uint32, :offset, 2
+  end
+
+  class DataParameter
+    optional :string, :source, 1
+    optional :uint32, :batch_size, 4
+    optional :uint32, :rand_skip, 7, :default => 0
+    optional ::Caffe::DataParameter::DB, :backend, 8, :default => ::Caffe::DataParameter::DB::LEVELDB
+    optional :float, :scale, 2, :default => 1
+    optional :string, :mean_file, 3
+    optional :uint32, :crop_size, 5, :default => 0
+    optional :bool, :mirror, 6, :default => false
+    optional :bool, :force_encoded_color, 9, :default => false
+    optional :uint32, :prefetch, 10, :default => 4
+  end
+
+  class DropoutParameter
+    optional :float, :dropout_ratio, 1, :default => 0.5
+  end
+
+  class DummyDataParameter
+    repeated ::Caffe::FillerParameter, :data_filler, 1
+    repeated ::Caffe::BlobShape, :shape, 6
+    repeated :uint32, :num, 2
+    repeated :uint32, :channels, 3
+    repeated :uint32, :height, 4
+    repeated :uint32, :width, 5
+  end
+
+  class EltwiseParameter
+    optional ::Caffe::EltwiseParameter::EltwiseOp, :operation, 1, :default => ::Caffe::EltwiseParameter::EltwiseOp::SUM
+    repeated :float, :coeff, 2
+    optional :bool, :stable_prod_grad, 3, :default => true
+  end
+
+  class ELUParameter
+    optional :float, :alpha, 1, :default => 1
+  end
+
+  class EmbedParameter
+    optional :uint32, :num_output, 1
+    optional :uint32, :input_dim, 2
+    optional :bool, :bias_term, 3, :default => true
+    optional ::Caffe::FillerParameter, :weight_filler, 4
+    optional ::Caffe::FillerParameter, :bias_filler, 5
+  end
+
+  class ExpParameter
+    optional :float, :base, 1, :default => -1
+    optional :float, :scale, 2, :default => 1
+    optional :float, :shift, 3, :default => 0
+  end
+
+  class FlattenParameter
+    optional :int32, :axis, 1, :default => 1
+    optional :int32, :end_axis, 2, :default => -1
+  end
+
+  class HDF5DataParameter
+    optional :string, :source, 1
+    optional :uint32, :batch_size, 2
+    optional :bool, :shuffle, 3, :default => false
+  end
+
+  class HDF5OutputParameter
+    optional :string, :file_name, 1
+  end
+
+  class HingeLossParameter
+    optional ::Caffe::HingeLossParameter::Norm, :norm, 1, :default => ::Caffe::HingeLossParameter::Norm::L1
+  end
+
+  class ImageDataParameter
+    optional :string, :source, 1
+    optional :uint32, :batch_size, 4, :default => 1
+    optional :uint32, :rand_skip, 7, :default => 0
+    optional :bool, :shuffle, 8, :default => false
+    optional :uint32, :new_height, 9, :default => 0
+    optional :uint32, :new_width, 10, :default => 0
+    optional :bool, :is_color, 11, :default => true
+    optional :float, :scale, 2, :default => 1
+    optional :string, :mean_file, 3
+    optional :uint32, :crop_size, 5, :default => 0
+    optional :bool, :mirror, 6, :default => false
+    optional :string, :root_folder, 12
+  end
+
+  class InfogainLossParameter
+    optional :string, :source, 1
+  end
+
+  class InnerProductParameter
+    optional :uint32, :num_output, 1
+    optional :bool, :bias_term, 2, :default => true
+    optional ::Caffe::FillerParameter, :weight_filler, 3
+    optional ::Caffe::FillerParameter, :bias_filler, 4
+    optional :int32, :axis, 5, :default => 1
+    optional :bool, :transpose, 6, :default => false
+  end
+
+  class InputParameter
+    repeated ::Caffe::BlobShape, :shape, 1
+  end
+
+  class LogParameter
+    optional :float, :base, 1, :default => -1
+    optional :float, :scale, 2, :default => 1
+    optional :float, :shift, 3, :default => 0
+  end
+
+  class LRNParameter
+    optional :uint32, :local_size, 1, :default => 5
+    optional :float, :alpha, 2, :default => 1
+    optional :float, :beta, 3, :default => 0.75
+    optional ::Caffe::LRNParameter::NormRegion, :norm_region, 4, :default => ::Caffe::LRNParameter::NormRegion::ACROSS_CHANNELS
+    optional :float, :k, 5, :default => 1
+    optional ::Caffe::LRNParameter::Engine, :engine, 6, :default => ::Caffe::LRNParameter::Engine::DEFAULT
+  end
+
+  class MemoryDataParameter
+    optional :uint32, :batch_size, 1
+    optional :uint32, :channels, 2
+    optional :uint32, :height, 3
+    optional :uint32, :width, 4
+  end
+
+  class MVNParameter
+    optional :bool, :normalize_variance, 1, :default => true
+    optional :bool, :across_channels, 2, :default => false
+    optional :float, :eps, 3, :default => 1e-09
+  end
+
+  class ParameterParameter
+    optional ::Caffe::BlobShape, :shape, 1
+  end
+
+  class PoolingParameter
+    optional ::Caffe::PoolingParameter::PoolMethod, :pool, 1, :default => ::Caffe::PoolingParameter::PoolMethod::MAX
+    optional :uint32, :pad, 4, :default => 0
+    optional :uint32, :pad_h, 9, :default => 0
+    optional :uint32, :pad_w, 10, :default => 0
+    optional :uint32, :kernel_size, 2
+    optional :uint32, :kernel_h, 5
+    optional :uint32, :kernel_w, 6
+    optional :uint32, :stride, 3, :default => 1
+    optional :uint32, :stride_h, 7
+    optional :uint32, :stride_w, 8
+    optional ::Caffe::PoolingParameter::Engine, :engine, 11, :default => ::Caffe::PoolingParameter::Engine::DEFAULT
+    optional :bool, :global_pooling, 12, :default => false
+  end
+
+  class PowerParameter
+    optional :float, :power, 1, :default => 1
+    optional :float, :scale, 2, :default => 1
+    optional :float, :shift, 3, :default => 0
+  end
+
+  class PythonParameter
+    optional :string, :module, 1
+    optional :string, :layer, 2
+    optional :string, :param_str, 3
+    optional :bool, :share_in_parallel, 4, :default => false
+  end
+
+  class RecurrentParameter
+    optional :uint32, :num_output, 1, :default => 0
+    optional ::Caffe::FillerParameter, :weight_filler, 2
+    optional ::Caffe::FillerParameter, :bias_filler, 3
+    optional :bool, :debug_info, 4, :default => false
+    optional :bool, :expose_hidden, 5, :default => false
+  end
+
+  class ReductionParameter
+    optional ::Caffe::ReductionParameter::ReductionOp, :operation, 1, :default => ::Caffe::ReductionParameter::ReductionOp::SUM
+    optional :int32, :axis, 2, :default => 0
+    optional :float, :coeff, 3, :default => 1
+  end
+
+  class ReLUParameter
+    optional :float, :negative_slope, 1, :default => 0
+    optional ::Caffe::ReLUParameter::Engine, :engine, 2, :default => ::Caffe::ReLUParameter::Engine::DEFAULT
+  end
+
+  class ReshapeParameter
+    optional ::Caffe::BlobShape, :shape, 1
+    optional :int32, :axis, 2, :default => 0
+    optional :int32, :num_axes, 3, :default => -1
+  end
+
+  class ScaleParameter
+    optional :int32, :axis, 1, :default => 1
+    optional :int32, :num_axes, 2, :default => 1
+    optional ::Caffe::FillerParameter, :filler, 3
+    optional :bool, :bias_term, 4, :default => false
+    optional ::Caffe::FillerParameter, :bias_filler, 5
+  end
+
+  class SigmoidParameter
+    optional ::Caffe::SigmoidParameter::Engine, :engine, 1, :default => ::Caffe::SigmoidParameter::Engine::DEFAULT
+  end
+
+  class SliceParameter
+    optional :int32, :axis, 3, :default => 1
+    repeated :uint32, :slice_point, 2
+    optional :uint32, :slice_dim, 1, :default => 1
+  end
+
+  class SoftmaxParameter
+    optional ::Caffe::SoftmaxParameter::Engine, :engine, 1, :default => ::Caffe::SoftmaxParameter::Engine::DEFAULT
+    optional :int32, :axis, 2, :default => 1
+  end
+
+  class TanHParameter
+    optional ::Caffe::TanHParameter::Engine, :engine, 1, :default => ::Caffe::TanHParameter::Engine::DEFAULT
+  end
+
+  class TileParameter
+    optional :int32, :axis, 1, :default => 1
+    optional :int32, :tiles, 2
+  end
+
+  class ThresholdParameter
+    optional :float, :threshold, 1, :default => 0
+  end
+
+  class WindowDataParameter
+    optional :string, :source, 1
+    optional :float, :scale, 2, :default => 1
+    optional :string, :mean_file, 3
+    optional :uint32, :batch_size, 4
+    optional :uint32, :crop_size, 5, :default => 0
+    optional :bool, :mirror, 6, :default => false
+    optional :float, :fg_threshold, 7, :default => 0.5
+    optional :float, :bg_threshold, 8, :default => 0.5
+    optional :float, :fg_fraction, 9, :default => 0.25
+    optional :uint32, :context_pad, 10, :default => 0
+    optional :string, :crop_mode, 11, :default => "warp"
+    optional :bool, :cache_images, 12, :default => false
+    optional :string, :root_folder, 13
+  end
+
+  class SPPParameter
+    optional :uint32, :pyramid_height, 1
+    optional ::Caffe::SPPParameter::PoolMethod, :pool, 2, :default => ::Caffe::SPPParameter::PoolMethod::MAX
+    optional ::Caffe::SPPParameter::Engine, :engine, 6, :default => ::Caffe::SPPParameter::Engine::DEFAULT
+  end
+
+  class V1LayerParameter
+    repeated :string, :bottom, 2
+    repeated :string, :top, 3
+    optional :string, :name, 4
+    repeated ::Caffe::NetStateRule, :include, 32
+    repeated ::Caffe::NetStateRule, :exclude, 33
+    optional ::Caffe::V1LayerParameter::LayerType, :type, 5
+    repeated ::Caffe::BlobProto, :blobs, 6
+    repeated :string, :param, 1001
+    repeated ::Caffe::V1LayerParameter::DimCheckMode, :blob_share_mode, 1002
+    repeated :float, :blobs_lr, 7
+    repeated :float, :weight_decay, 8
+    repeated :float, :loss_weight, 35
+    optional ::Caffe::AccuracyParameter, :accuracy_param, 27
+    optional ::Caffe::ArgMaxParameter, :argmax_param, 23
+    optional ::Caffe::ConcatParameter, :concat_param, 9
+    optional ::Caffe::ContrastiveLossParameter, :contrastive_loss_param, 40
+    optional ::Caffe::ConvolutionParameter, :convolution_param, 10
+    optional ::Caffe::DataParameter, :data_param, 11
+    optional ::Caffe::DropoutParameter, :dropout_param, 12
+    optional ::Caffe::DummyDataParameter, :dummy_data_param, 26
+    optional ::Caffe::EltwiseParameter, :eltwise_param, 24
+    optional ::Caffe::ExpParameter, :exp_param, 41
+    optional ::Caffe::HDF5DataParameter, :hdf5_data_param, 13
+    optional ::Caffe::HDF5OutputParameter, :hdf5_output_param, 14
+    optional ::Caffe::HingeLossParameter, :hinge_loss_param, 29
+    optional ::Caffe::ImageDataParameter, :image_data_param, 15
+    optional ::Caffe::InfogainLossParameter, :infogain_loss_param, 16
+    optional ::Caffe::InnerProductParameter, :inner_product_param, 17
+    optional ::Caffe::LRNParameter, :lrn_param, 18
+    optional ::Caffe::MemoryDataParameter, :memory_data_param, 22
+    optional ::Caffe::MVNParameter, :mvn_param, 34
+    optional ::Caffe::PoolingParameter, :pooling_param, 19
+    optional ::Caffe::PowerParameter, :power_param, 21
+    optional ::Caffe::ReLUParameter, :relu_param, 30
+    optional ::Caffe::SigmoidParameter, :sigmoid_param, 38
+    optional ::Caffe::SoftmaxParameter, :softmax_param, 39
+    optional ::Caffe::SliceParameter, :slice_param, 31
+    optional ::Caffe::TanHParameter, :tanh_param, 37
+    optional ::Caffe::ThresholdParameter, :threshold_param, 25
+    optional ::Caffe::WindowDataParameter, :window_data_param, 20
+    optional ::Caffe::TransformationParameter, :transform_param, 36
+    optional ::Caffe::LossParameter, :loss_param, 42
+    optional ::Caffe::V0LayerParameter, :layer, 1
+  end
+
+  class V0LayerParameter
+    optional :string, :name, 1
+    optional :string, :type, 2
+    optional :uint32, :num_output, 3
+    optional :bool, :biasterm, 4, :default => true
+    optional ::Caffe::FillerParameter, :weight_filler, 5
+    optional ::Caffe::FillerParameter, :bias_filler, 6
+    optional :uint32, :pad, 7, :default => 0
+    optional :uint32, :kernelsize, 8
+    optional :uint32, :group, 9, :default => 1
+    optional :uint32, :stride, 10, :default => 1
+    optional ::Caffe::V0LayerParameter::PoolMethod, :pool, 11, :default => ::Caffe::V0LayerParameter::PoolMethod::MAX
+    optional :float, :dropout_ratio, 12, :default => 0.5
+    optional :uint32, :local_size, 13, :default => 5
+    optional :float, :alpha, 14, :default => 1
+    optional :float, :beta, 15, :default => 0.75
+    optional :float, :k, 22, :default => 1
+    optional :string, :source, 16
+    optional :float, :scale, 17, :default => 1
+    optional :string, :meanfile, 18
+    optional :uint32, :batchsize, 19
+    optional :uint32, :cropsize, 20, :default => 0
+    optional :bool, :mirror, 21, :default => false
+    repeated ::Caffe::BlobProto, :blobs, 50
+    repeated :float, :blobs_lr, 51
+    repeated :float, :weight_decay, 52
+    optional :uint32, :rand_skip, 53, :default => 0
+    optional :float, :det_fg_threshold, 54, :default => 0.5
+    optional :float, :det_bg_threshold, 55, :default => 0.5
+    optional :float, :det_fg_fraction, 56, :default => 0.25
+    optional :uint32, :det_context_pad, 58, :default => 0
+    optional :string, :det_crop_mode, 59, :default => "warp"
+    optional :int32, :new_num, 60, :default => 0
+    optional :int32, :new_channels, 61, :default => 0
+    optional :int32, :new_height, 62, :default => 0
+    optional :int32, :new_width, 63, :default => 0
+    optional :bool, :shuffle_images, 64, :default => false
+    optional :uint32, :concat_dim, 65, :default => 1
+    optional ::Caffe::HDF5OutputParameter, :hdf5_output_param, 1001
+  end
+
+  class PReLUParameter
+    optional ::Caffe::FillerParameter, :filler, 1
+    optional :bool, :channel_shared, 2, :default => false
+  end
+
+end
+