Изменение в Prototext, но сеть не меняется? (LeNet) - PullRequest
Новая
       7

Изменение в Prototext, но сеть не меняется? (LeNet)

0 голосов
/ 28 августа 2018

эй, поэтому я немного поэкспериментировал с cafe с python и написал немного кода для чтения некоторой информации из моей сети (leNet, mnist)

Я определяю свой решатель и слой, который хочу проанализировать с помощью 

solver = caffe.get_solver(solver_prototxt)
layer = 'conv1'

и считайте карты функций и размер пакета с помощью 

numFm = solver.net.blobs[layer].shape[1] # Feature Maps
batchsize = solver.net.blobs[layer].shape[0]

Все работает отлично и выдает то, что я хочу (20 функций Карт, 64 изображения в мини-пакете), но я пытался немного изменить ситуацию и посмотреть, что произойдет, когда я изменю что-то в своей сети. Для этого я изменил lenet.prototxt с 

name: "LeNet"
layer {
 name: "data"
 type: "Input"
 top: "data"
  input_param { shape: { dim: 64 dim: 1 dim: 28 dim: 28 } }
}
layer {
  name: "conv1"
  type: "Convolution"
  bottom: "data"
  top: "conv1"
  param {
    lr_mult: 1
  }
  param {
   lr_mult: 2
  }
  convolution_param {
    num_output: 20
    kernel_size: 5
    stride: 1
    weight_filler {
       type: "xavier"
     }
     bias_filler {
      type: "constant"
    }
  }

до 

  ....
  convolution_param {
    num_output: 500
    kernel_size: 5
    stride: 1
    weight_filler {
       type: "xavier"
     }
     bias_filler {
      type: "constant"
    }
  }

поэтому я изменил номер карты объектов с 20 на 500, но после сохранения и повторного запуска кода я все равно получаю 20 в качестве числа карт объектов.

Есть идеи, что я должен изменить?

...