search

slothfulxtx / MBPTrack3D

[ICCV2023] MBPTrack: Improving 3D Point Cloud Tracking with Memory Networks and Box Priors
24 stars 6 forks source link

Is it possible to run MBPTrack on CPU? #3

Closed MaxTeselkin closed 1 year ago

MaxTeselkin commented 1 year ago

Hi, I need to run MBPTrack on CPU. As far as I understood, your code (main.py) supports only GPU inference. I guess MBPTrack is not very computationally expensive, is it possible to run it on CPU?

slothfulxtx commented 1 year ago

Hi, thanks for your interst. Yes, you can implement MBPTrack on CPU with some minor modification, because our code does not rely on GPU specific env or code. For example, you may remove some code related to pynvml, which is used for finding the GPU that is not occupied.

slothfulxtx commented 1 year ago

Sorry, I've just noted that pointnet-ops may not support CPU env, you can replace these code with pytorch3d. All these point cloud operators have been implemented in pytorch3d.