search

yihongXU / TransCenter

This is the official implementation of TransCenter (TPAMI). The code and pretrained models are now available here: https://gitlab.inria.fr/yixu/TransCenter_official.
https://team.inria.fr/robotlearn/transcenter-transformers-with-dense-queriesfor-multiple-object-tracking/
Other
108 stars 7 forks source link

关于减少显存占用 #6

Closed 1145284121 closed 2 years ago

1145284121 commented 2 years ago

感谢开源， 作者有试过输入小一点分辨率的图片，或者其他减少显存占用的思路，实验效果如何吗 （）当前的模型在batch_size调到1依然需要超过12G显存）

yihongXU commented 2 years ago

你好，我们没有试过减小输入分辨率。你可以尝试分辨率长宽减半。另外，我们将会在近期推出更轻便的版本，敬请期待，祝好。