RobertCsordas / RFCN-tensorflow

RFCN implementation in TensorFlow
292 stars 137 forks source link

batch_size larger than 1 #23

Open learnbott opened 6 years ago

learnbott commented 6 years ago

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.

jacklonghui commented 6 years ago

@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?

RobertCsordas commented 6 years ago

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.