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

Can tomp process multiple images at once as batches? #419

Closed
ninono12345 opened this issue Jan 24, 2024 · 2 comments
Closed

Can tomp process multiple images at once as batches? #419

ninono12345 opened this issue Jan 24, 2024 · 2 comments

Comments

@ninono12345
Copy link

Hello @martin-danelljan I want to ask you for advice, since you have been more active.

Is it possible so that for example if I'm tracking 4 objects at once, that I could send into track a tensor with 4 batches, instead, so tracking woud do all computations for all 4 objects at once, instead passing through neural networks 4 separate times?

I tried just to simulate 2 frame tracking to do torch.cat with 2 image tensors (the same frame to be tracked 2 times, next time I would use different ones), didn't work... Tried to torch.cat inside classify_target all variables that enter neural networks, also didn't work... Now I'm studying the algorithm, but I wanted to ask you in theory is this possible, or would it be too much work?

Thank you

@2006pmach
Copy link
Collaborator

Have a look at TaMOs our most recent tracker. It is designed to track multiple objects at the same time more efficiently.

@ninono12345
Copy link
Author

ninono12345 commented May 19, 2024

Sorry, didn't forgot to reply, already figured it out. I concatenate the input tensors and very slightly change predict_cls_bbreg_filters_parallel, and tomp.py classify target

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

2 participants