code for Fast Point Cloud Registration with Optimal Transport

Related tags

Deep Learning robot
Overview

robot

This is the repository for the paper "Accurate Point Cloud Registration with Robust Optimal Transport".

We are in the process of refactoring the code and will update this repository after that.

For now, we temporally relink the repository to https://github.com/uncbiag/shapmagn , which contains our development code. Stay tuned!

Comments
  • python pointnet2/lib/setup.py install

    python pointnet2/lib/setup.py install

    Thank you for your excellent work. when i using the code, there are some errors. first i use ubuntu,cuda11.3,pytorch1.8, it shows :

    /home/lh/Nasdata/DJstudents/yyang17/Code/robot/pointnet2/lib/src/ball_query.cpp:3:10: fatal error: THC/THC.h: 没有那个文件或目录
        3 | #include <THC/THC.h>
    
    

    In order to solve this , i tried to use cuda10.2, pytorch1.6,but there still has some errors:

    
      _File "/home/lh/anaconda3/envs/yy_mesh_1/lib/python3.8/site-packages/setuptools/_distutils/dist.py", line 992, in run_command
        cmd_obj.run()
      File "/home/lh/anaconda3/envs/yy_mesh_1/lib/python3.8/site-packages/setuptools/command/build_ext.py", line 79, in run
        _build_ext.run(self)
      File "/home/lh/anaconda3/envs/yy_mesh_1/lib/python3.8/site-packages/setuptools/_distutils/command/build_ext.py", line 346, in run
        self.build_extensions()
      File "/home/lh/anaconda3/envs/yy_mesh_1/lib/python3.8/site-packages/torch/utils/cpp_extension.py", line 653, in build_extensions
        build_ext.build_extensions(self)
      File "/home/lh/anaconda3/envs/yy_mesh_1/lib/python3.8/site-packages/setuptools/_distutils/command/build_ext.py", line 466, in build_extensions
        self._build_extensions_serial()
      File "/home/lh/anaconda3/envs/yy_mesh_1/lib/python3.8/site-packages/setuptools/_distutils/command/build_ext.py", line 492, in _build_extensions_serial
        self.build_extension(ext)
      File "/home/lh/anaconda3/envs/yy_mesh_1/lib/python3.8/site-packages/setuptools/command/build_ext.py", line 202, in build_extension
        _build_ext.build_extension(self, ext)
      File "/home/lh/anaconda3/envs/yy_mesh_1/lib/python3.8/site-packages/setuptools/_distutils/command/build_ext.py", line 547, in build_extension
        objects = self.compiler.compile(
      File "/home/lh/anaconda3/envs/yy_mesh_1/lib/python3.8/site-packages/torch/utils/cpp_extension.py", line 473, in unix_wrap_ninja_compile
        _write_ninja_file_and_compile_objects(
      File "/home/lh/anaconda3/envs/yy_mesh_1/lib/python3.8/site-packages/torch/utils/cpp_extension.py", line 1233, in _write_ninja_file_and_compile_objects
        _run_ninja_build(
      File "/home/lh/anaconda3/envs/yy_mesh_1/lib/python3.8/site-packages/torch/utils/cpp_extension.py", line 1538, in _run_ninja_build
        raise RuntimeError(message) from e
    RuntimeError: Error compiling objects for extension_
    

    Can you share your version of pytorch, cuda & pykeops( I used version of 1.5, 2.1 also come out errors)

    opened by yangyang117 3
  • Translation between DIRLab and PVT coordinates

    Translation between DIRLab and PVT coordinates

    Hi @hbgtjxzbbx thank you for sharing the code for your great work!

    Could you please guide me on how to precisely transform the DIRLab landmarks to the coordinate system of the PVT clouds or the other way round?

    Thanks!

    opened by alexbgl 1
  • Registration results

    Registration results

    Hello, thanks for your excellent work! I observed that the proposed method can be used for rigid point cloud registration and found partial_prealign_reg.py in the demo. However, the registration results seems to contain only flowed points and I could not find the transformation matrix (i.e., rotation and translation). Could you please how can I get the transformation matrix?

    Thanks in advance.

    opened by FanLu97 1
  • Commands/settings for reproducing paper results on PVT

    Commands/settings for reproducing paper results on PVT

    Hi @hbgtjxzbbx,

    thanks again for sharing code and data of your interesting work!

    I would like to reproduce your experiments on the PVT1010 dataset in the NeurIPS paper. Specifically, I would like to reproduce D-Robot raw/spline/LDDMM with and without postprocessing, i.e. last three rows of Tab. 1 in the main paper and first and last row in Tab. 4 of the supplementary. Could you provide the required task_setting.json files and the corresponding commands? This would be a great help.

    Many thanks for considering my request!

    opened by alexbgl 0
  • Issue for visualization

    Issue for visualization

    Thanks for such an excellent work. When I tried to visualize the point cloud, the following issue was raised:

    [xcb] Unknown request in queue while appending request
    [xcb] Most likely this is a multi-threaded client and XInitThreads has not been called
    [xcb] Aborting, sorry about that.
    python3: ../../src/xcb_io.c:151: append_pending_request: Assertion `!xcb_xlib_unknown_req_pending' failed.
    

    It looks like the issue happened on the 'p.write_frame()' function.

    Hope for your answer. Thanks。

    opened by leihouyeung 7
Owner
null
GeoTransformer - Geometric Transformer for Fast and Robust Point Cloud Registration

Geometric Transformer for Fast and Robust Point Cloud Registration PyTorch imple

Zheng Qin 220 Jan 5, 2023
Registration Loss Learning for Deep Probabilistic Point Set Registration

RLLReg This repository contains a Pytorch implementation of the point set registration method RLLReg. Details about the method can be found in the 3DV

Felix Järemo Lawin 35 Nov 2, 2022
Point Cloud Denoising input segmentation output raw point-cloud valid/clear fog rain de-noised Abstract Lidar sensors are frequently used in environme

Point Cloud Denoising input segmentation output raw point-cloud valid/clear fog rain de-noised Abstract Lidar sensors are frequently used in environme

null 75 Nov 24, 2022
Transport Mode detection - can detect the mode of transport with the help of features such as acceeration,jerk etc

title emoji colorFrom colorTo sdk app_file pinned Transport_Mode_Detector ?? purple yellow gradio app.py false Configuration title: string Display tit

Nishant Rajadhyaksha 3 Jan 16, 2022
Code for paper "Vocabulary Learning via Optimal Transport for Neural Machine Translation"

**Codebase and data are uploaded in progress. ** VOLT(-py) is a vocabulary learning codebase that allows researchers and developers to automaticaly ge

null 416 Jan 9, 2023
[PyTorch] Official implementation of CVPR2021 paper "PointDSC: Robust Point Cloud Registration using Deep Spatial Consistency". https://arxiv.org/abs/2103.05465

PointDSC repository PyTorch implementation of PointDSC for CVPR'2021 paper "PointDSC: Robust Point Cloud Registration using Deep Spatial Consistency",

null 153 Dec 14, 2022
MVP Benchmark for Multi-View Partial Point Cloud Completion and Registration

MVP Benchmark: Multi-View Partial Point Clouds for Completion and Registration [NEWS] 2021-07-12 [NEW ?? ] The submission on Codalab starts! 2021-07-1

PL 93 Dec 21, 2022
Compute descriptors for 3D point cloud registration using a multi scale sparse voxel architecture

MS-SVConv : 3D Point Cloud Registration with Multi-Scale Architecture and Self-supervised Fine-tuning Compute features for 3D point cloud registration

null 42 Jul 25, 2022
You Only Hypothesize Once: Point Cloud Registration with Rotation-equivariant Descriptors

You Only Hypothesize Once: Point Cloud Registration with Rotation-equivariant Descriptors In this paper, we propose a novel local descriptor-based fra

Haiping Wang 80 Dec 15, 2022
Point Cloud Registration using Representative Overlapping Points.

Point Cloud Registration using Representative Overlapping Points (ROPNet) Abstract 3D point cloud registration is a fundamental task in robotics and c

ZhuLifa 36 Dec 16, 2022
PyTorch implementation of NeurIPS 2021 paper: "CoFiNet: Reliable Coarse-to-fine Correspondences for Robust Point Cloud Registration"

PyTorch implementation of NeurIPS 2021 paper: "CoFiNet: Reliable Coarse-to-fine Correspondences for Robust Point Cloud Registration"

null 76 Jan 3, 2023
Point Cloud Registration Network

PCRNet: Point Cloud Registration Network using PointNet Encoding Source Code Author: Vinit Sarode and Xueqian Li Paper | Website | Video | Pytorch Imp

ViNiT SaRoDe 59 Nov 19, 2022
POT : Python Optimal Transport

POT: Python Optimal Transport This open source Python library provide several solvers for optimization problems related to Optimal Transport for signa

Python Optimal Transport 1.7k Dec 31, 2022
Official implementation of our CVPR2021 paper "OTA: Optimal Transport Assignment for Object Detection" in Pytorch.

OTA: Optimal Transport Assignment for Object Detection This project provides an implementation for our CVPR2021 paper "OTA: Optimal Transport Assignme

null 217 Jan 3, 2023
Official implementation of NLOS-OT: Passive Non-Line-of-Sight Imaging Using Optimal Transport (IEEE TIP, accepted)

NLOS-OT Official implementation of NLOS-OT: Passive Non-Line-of-Sight Imaging Using Optimal Transport (IEEE TIP, accepted) Description In this reposit

Ruixu Geng(耿瑞旭) 16 Dec 16, 2022
Universal Probability Distributions with Optimal Transport and Convex Optimization

Sylvester normalizing flows for variational inference Pytorch implementation of Sylvester normalizing flows, based on our paper: Sylvester normalizing

Rianne van den Berg 172 Dec 13, 2022
Style-based Point Generator with Adversarial Rendering for Point Cloud Completion (CVPR 2021)

Style-based Point Generator with Adversarial Rendering for Point Cloud Completion (CVPR 2021) An efficient PyTorch library for Point Cloud Completion.

Microsoft 119 Jan 2, 2023
Implementation of the "PSTNet: Point Spatio-Temporal Convolution on Point Cloud Sequences" paper.

PSTNet: Point Spatio-Temporal Convolution on Point Cloud Sequences Introduction Point cloud sequences are irregular and unordered in the spatial dimen

Hehe Fan 63 Dec 9, 2022
Implementation of the "Point 4D Transformer Networks for Spatio-Temporal Modeling in Point Cloud Videos" paper.

Point 4D Transformer Networks for Spatio-Temporal Modeling in Point Cloud Videos Introduction Point cloud videos exhibit irregularities and lack of or

Hehe Fan 101 Dec 29, 2022