テストフェーズ中にデータブロブの平均を計算する

Question

データとラベルを持つhdf5レイヤを使用しています。

layer { 
    name: "data" 
    type: "HDF5Data" 
    top: "data" 
    top: "label" 
    include { 
    phase: TEST 
    } 
    hdf5_data_param { 
    source: "./list.txt" 
    batch_size: 8 
    shuffle: true 
    } 
} 

テストフェーズでは、テストセットから8つのイメージがロードされ、ネットワークに供給されます。テスト段階で各画像の平均値を出力したいと思います。それをCAFFEで使用することは可能ですか？どのレイヤーを使用すべきですか？

Answer 1

私はあなたが"Reduction"層

layer { 
    type: "Reduction" 
    name: "img_mean" 
    bottom: "data" 
    top: "img_mean" 
    reduction_param { 
    operation: MEAN 
    axis: 0 # to reduce the entire batch, or axis: 1 for per-image mean 
    } 
}