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

时间:2016-10-19 04:44:50

标签: deep-learning caffe pycaffe lmdb

我正在使用pycaffe,我的火车和测试数据是LMDB格式。

我创建了这样的网: net = caffe.Net('train.prototxt','c.caffemodel',caffe.TEST)

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

由于

1 个答案:

答案 0 :(得分:0)

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