我正在使用包含数据和标签的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中可以实现吗？我应该使用哪个层？

回答：

我想你正在寻找"Reduction"层

layer {  type: "Reduction"  name: "img_mean"  bottom: "data"  top: "img_mean"  reduction_param {    operation: MEAN    axis: 0  # 用于计算整个批次的平均值，或axis: 1用于计算每张图片的平均值  }}