Caffe deploy.prototxt具有不同維度的多個輸入

Question

我在Caffe中有一個網絡，它接受多個圖像輸入（可能具有不同的維度）並將它們用作網絡初始部分的單獨blob。

對於培訓階段，我通過創建HDF5數據庫來實現這一點，如文檔中建議的那樣。

但是，對於部署階段，我需要用訓練階段用輸入層替換數據層，並指定輸入blob的維度。

對於單個blob，這是通過input_param的shape屬性完成的。在這種情況下，我怎麼能做到這一點，我有多個blob，可能有不同的尺寸？

謝謝。

Answer 1

如果您在caffe.proto仔細觀察你會發現input和input_shape有repeated屬性：這意味着你可以有幾個等項

// DEPRECATED. See InputParameter. The input blobs to the network. 
repeated string input = 3; 
// DEPRECATED. See InputParameter. The shape of the input blobs. 
repeated BlobShape input_shape = 8; 

：

name: "AnAmazingNetWithMultipleInputs_ThankYouShai_IwillForeverBeInYourDebt" 
input: "first_input_1x3x127x127" 
input_shape { dim: 1 dim: 3 dim: 127 dim: 127 } 
input: "second_input_2x4x224x224" 
input_shape { dim: 2 dim: 4 dim: 224 dim: 224 } 
# I hope you can take it from here ;) 

---

如果您在的相關部分看得更近，你會注意到這種形式的「輸入形狀聲明」是DEPRECATED。
一個更好的辦法是使用"Input"層：

layer { 
    type: "Input" 
    name: "one_input_layer" 
    top: "first_input_1x3x127x127" 
    shape { dim: 1 dim: 3 dim: 127 dim: 127 } 
    top: "second_input_2x4x224x224" 
    shape { dim: 2 dim: 4 dim: 224 dim: 224 } 
} 

你也可以有不同的"Input"層對於每個輸入，如果你發現它更容易理解/閱讀。