deep-learning - 在 pycaffe 中使用 net.forward 时从 lmdb 的第一个转发

Question

我正在使用 pycaffe，我的训练和测试数据是 LMDB 格式。

我已经像这样创建了我的网络：net = caffe.Net('train.prototxt', 'c.caffemodel', caffe.TEST)

当您调用 net.forward 时，您会隐式地逐批浏览 LMDB 测试数据库。我的问题是如何从 LMDB 开始并在前 n 批测试数据上测试我的网络？

谢谢

Answer 1

不确定它是否仍然相关，但您需要更改输入层的数据，就像这样net.params['data'][0].data[...] =net.params['data'][0].data[n,:,:,:]