repo_description: [CVPR2022] Geometric Transformer for Fast and Robust Point Cloud Registration
repo_topics: attention-mechanism computer-vision deep-learning neural-network point-cloud point-cloud-registration pytorch registration transformer
code: from setuptools import setup
  from torch.utils.cpp_extension import BuildExtension, CUDAExtension
  
  
  setup(
      name='geotransformer',
      version='1.0.0',
      ext_modules=[
          CUDAExtension(
              name='geotransformer.ext',
              sources=[
                  'geotransformer/extensions/extra/cloud/cloud.cpp',
                  'geotransformer/extensions/cpu/grid_subsampling/grid_subsampling.cpp',
                  'geotransformer/extensions/cpu/grid_subsampling/grid_subsampling_cpu.cpp',
                  'geotransformer/extensions/cpu/radius_neighbors/radius_neighbors.cpp',
                  'geotransformer/extensions/cpu/radius_neighbors/radius_neighbors_cpu.cpp',
                  'geotransformer/extensions/pybind.cpp',
              ],
          ),
      ],
      cmdclass={'build_ext': BuildExtension},
  )
  