Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

batch_size larger than 1 #23

Open
learnbott opened this issue Nov 8, 2017 · 2 comments
Open

batch_size larger than 1 #23

learnbott opened this issue Nov 8, 2017 · 2 comments

Comments

@learnbott
Copy link

learnbott commented Nov 8, 2017

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.

@Compilerxb
Copy link

@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
Copy link
Owner

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.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

3 participants