Neural network caffe prototxt中没有从hdf5读取数据

我正试图通过给一个包含所有元素的水滴喂食来测试我的caffe模型。因此，我通过以下方式形成hdf5文件：

import h5py, os
import numpy as np

SIZE = 227 # fixed size to all images

X = np.ones((1, 3, SIZE, SIZE), dtype='f8')

with h5py.File('test_idty.h5','w') as H:
    H.create_dataset('img', data=X ) 
with open('test_h5_idty_list.txt','w') as L:
    L.write( '/home/wei/deep_metric/test_idty.h5' )

然后，我将caffe协议更改为：

layer{
  name:"data"
  type:"HDF5Data"
  top:"img"
  include:{
    phase:TEST
  }
  hdf5_data_param{
    source:"/home/wei/deep_metric/test_h5_idty_list.txt"
    batch_size:1
  }
}

然后，我尝试通过以下方式确保正确输入数据：

net = caffe.Net(Model,Pretrained,caffe.TEST)
data = net.blobs['img'].data.copy()

然而，这给了我矩阵中所有的零。 你知道怎么解决吗

---

谢谢

为了让

“HDF5Data”

层读取它的第一批数据，您需要首先调用

net.forward（）

。一旦向前传递完成，层的顶部将读取文件中的数据。

@ShaiThank你太感谢了！我想我错过了前进功能。现在可以了。