我们可以将Concate图层用于两种不同大小的输入图像吗?

时间:2017-03-02 11:03:47

标签: machine-learning neural-network deep-learning caffe

我为我的网络实施多尺度输入

input1(WxH)    -->Network----| 
                             |-->SoftmaxWithLoss
input2(W/2xH/2)-->Network----|
                             ^
label(WxH)-------------------|

该方案非常简单,只使用相同的网络架构,但图像比例不同。我可以使用Concate图层完成该任务吗?特别是,图层原型将是

layer {
  name: "data"
  type: "HDF5Data"
  top: "input1"
  top: "input2"
  top: "label"
  include {
    phase: TRAIN 
  }
  hdf5_data_param {
    source: "./list2.txt"
    batch_size: 2
  }
}

layer {
  name: "concat"
  type: "Concat" 
  bottom: "input1"
  bottom: "input2"
  top: "data"
}

0 个答案:

没有答案