search

openxrlab / xrmocap

OpenXRLab Multi-view Motion Capture Toolbox and Benchmark
https://xrmocap.readthedocs.io/
Other
348 stars 42 forks source link

Surpport batch_size>1? #124

Closed Dragon2938734 closed 1 year ago

Dragon2938734 commented 1 year ago

excusme, does this code surpport batch_size>1? for example, in position_encoding.py code: pixel_camera = torch.bmm(xy1.flatten(1, 2).repeat(views, 1, 1), torch.inverse(K).transpose(2, 1)) will get error if batch_size>1, because the dim 0 do not equal. Should i replace ".repeat(views, 1, 1)" to "" .repeat(batch*views, 1, 1)"?

github-actions[bot] commented 1 year ago

Hi @Dragon2938734, welcome to commit your first issue! 你好 @Dragon2938734，非常欢迎首次提交你的问题！

wqyin commented 1 year ago

Sorry, currently we only support batch_size = 1. MVP processes all the views for each frame at a time. Please feel free to modify the network to fit your use case better.