search

youmi-zym / GO-SLAM

[ICCV2023] GO-SLAM: Global Optimization for Consistent 3D Instant Reconstruction
Apache License 2.0
344 stars 30 forks source link

droid_backends #33

Open jc19970622 opened 4 months ago

jc19970622 commented 4 months ago

import torch import torch.nn.functional as F import droid_backends

anyone know what is the droid_backends?

coarsedamavand commented 2 months ago

It's from the setup.py file: https://github.com/youmi-zym/GO-SLAM/blob/e954a850bf52aced47bb40d7637e840edf0a73ce/setup.py#L7-L31

the setup.py file creates a setuptools.Extension for the c++ and CUDA files in the 'src/lib' subdirectory: https://github.com/youmi-zym/GO-SLAM/tree/e954a850bf52aced47bb40d7637e840edf0a73ce/src/lib

Read more here: https://pytorch.org/docs/stable/cpp_extension.html#torch.utils.cpp_extension.CUDAExtension