Open learnbott opened 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?
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.
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.