Forward from the first of lmdb when using net.forward in pycaffe

问题

I am using pycaffe and my train and test data is in LMDB format.

I have created my net like this: net = caffe.Net('train.prototxt', 'c.caffemodel', caffe.TEST)

when you call net.forward, implicitly you walk through the LMDB test database one by one batches. My question is how can I start from the beginning of LMDB and test my network on the first n batches of the test data?

Thanks

回答1:

not sure it still relevant but you would need to change the data of input layer ,something like this net.params['data'][0].data[...] =net.params['data'][0].data[n,:,:,:]

来源：https://stackoverflow.com/questions/40122308/forward-from-the-first-of-lmdb-when-using-net-forward-in-pycaffe