rsy6318 / ddf-loss Goto Github PK

Here we provide the examples for three tasks, including shape reconstruction, rigid registration and scene flow estimation.

pytorch             #1.10.0+cu111
pytorch3d           #0.6.2
open3d
trimesh
point-cloud-utils

python exp_shape_reconstruction.py

The reconstructed point cloud and trained model is saved at example/shape_reconstruction.

run python exp_registration.py to get the RE and TE.

(Note the rotation matrix here is the transpose of the GT rotation matrix.)

run python exp_scene_flow.py to get the EPE, ACC-0.05, ACC-0.1, and Outliers.