The caffe-net-upgrade from lcskrishna

lcskrishna / caffe-net-upgrade Goto Github PK

View Code? Open in Web Editor NEW

Upgrade deprecated caffemodel to latest caffemodel (V1LayerParameters to LayerParameters) without installing caffe

License: MIT License

CMake 5.19% C++ 94.81%

caffe-net-upgrade's People

Contributors

Stargazers

Watchers

caffe-net-upgrade's Issues

upgrade prototxt

input_shape is not upgraded to an input Layer

Input Prototxt:

name: "TestConv"
input: "data"
input_shape {
  dim: 1
  dim: 4
  dim: 224
  dim: 224
}
layer {
  name: "conv1/7x7_s2"
  type: "Convolution"
  bottom: "data"
  top: "conv1/7x7_s2"
  param {
    lr_mult: 1
    decay_mult: 1
  }
  convolution_param {
    num_output: 64
    bias_term: false
    pad: 3
    kernel_size: 7
    stride: 2
    weight_filler {
      type: "xavier"
      std: 0.1
    }
  }
}

Expected (From the official Caffe upgrade tool):

name: "TestConv"
layer {
  name: "input"
  type: "Input"
  top: "data"
  input_param {
    shape {
      dim: 1
      dim: 4
      dim: 224
      dim: 224
    }
  }
}
layer {
  name: "conv1/7x7_s2"
  type: "Convolution"
  bottom: "data"
  top: "conv1/7x7_s2"
  param {
    lr_mult: 1
    decay_mult: 1
  }
  convolution_param {
    num_output: 64
    bias_term: false
    pad: 3
    kernel_size: 7
    stride: 2
    weight_filler {
      type: "xavier"
      std: 0.1
    }
  }
}

Observed:

Loading prototxt file ... 
INFO: Reading the prototxt file from : /home/ian/test_conv.prototxt
INFO: prototxt read successful 
INFO: Network loaded is : TestConv
INFO: This network doesn't require upgrade

Recommend Projects