RFCN-tensorflow icon indicating copy to clipboard operation
RFCN-tensorflow copied to clipboard

batch_size larger than 1

Open learnbott opened this issue 8 years ago • 2 comments

How can the repo most easily be modified to have minibatch sizes larger than 1 image? I am willing to put some time into it, but need a little direction.

learnbott avatar Nov 08 '17 00:11 learnbott

@learnbott hi,I have the same question, how do I modify the increase of batch in this project? I looked for a long time, but I didn't find any method, and the default batch was 1.Have you solved the problem, can you help me?

Compilerxb avatar Nov 16 '17 12:11 Compilerxb

Supporting batch of >1 is not trivial because of varying number of bounding boxes per image, and varying image size. Boxes could be represented by TensorArrays, or a padded tensor and a count vector. The end of the network must probably be replicated many times and the output of the neural part should be split between them among the batch dimension.

RobertCsordas avatar Jan 08 '18 13:01 RobertCsordas