├── README.md ├── dataset ├── __init__.py ├── bhcp_dataloader.py ├── data_utils.py └── shapenet_seg_dataloader.py ├── demo_data ├── bike │ ├── 0.off │ ├── 1.off │ └── 2.off ├── chair │ ├── 0.off │ ├── 1.off │ └── 2.off ├── plane │ ├── 0.off │ ├── 1.off │ └── 2.off ├── realscan_chair │ ├── 0.off │ ├── 1.off │ └── 2.off └── shape_corres │ ├── src_and_query_pts │ ├── bike_query_pts.off │ ├── bike_src_pts.off │ ├── chair_query_pts.off │ ├── chair_src_pts.off │ ├── plane_query_pts.off │ └── plane_src_pts.off │ └── tgt_pts │ ├── bike_tgt_pts_0.off │ ├── bike_tgt_pts_1.off │ ├── bike_tgt_pts_2.off │ ├── chair_tgt_pts_0.off │ ├── chair_tgt_pts_1.off │ ├── chair_tgt_pts_2.off │ ├── plane_tgt_pts_0.off │ ├── plane_tgt_pts_1.off │ └── plane_tgt_pts_2.off ├── docs ├── glab.css ├── images │ ├── data_ico.jpg │ ├── paper_thumbnail.jpg │ ├── paper_thumbnail_ori.jpg │ ├── pipeline.jpg │ └── teaser.jpg └── index.html ├── evaluate ├── evaluate_corres_accuracy.py └── evaluate_label_transfer_accuracy.py ├── figs └── teaser.png ├── models ├── __init__.py ├── chamfer_distance.py └── pointnet2_structure_point_net.py ├── pointnet2 ├── __init__.py ├── _ext-src │ ├── include │ │ ├── ball_query.h │ │ ├── cuda_utils.h │ │ ├── group_points.h │ │ ├── interpolate.h │ │ ├── sampling.h │ │ └── utils.h │ └── src │ │ ├── ball_query.cpp │ │ ├── ball_query_gpu.cu │ │ ├── bindings.cpp │ │ ├── group_points.cpp │ │ ├── group_points_gpu.cu │ │ ├── interpolate.cpp │ │ ├── interpolate_gpu.cu │ │ ├── sampling.cpp │ │ └── sampling_gpu.cu ├── setup.py └── utils │ ├── __init__.py │ ├── etw_pytorch_utils │ ├── __init__.py │ ├── persistent_dataloader.py │ ├── pytorch_utils.py │ ├── sacred_trainer.py │ ├── seq.py │ ├── visdom_observer.py │ └── viz.py │ ├── linalg_utils.py │ ├── pointnet2_modules.py │ └── pointnet2_utils.py ├── requirements.txt ├── test ├── test_shape_correspondence.py └── test_structure_points.py ├── train ├── __init__.py ├── train_structure_points.py └── train_structure_points_for_label_transfer.py └── utils ├── __init__.py ├── check_points_utils.py ├── logutils.py ├── mesh_utils.py └── point_cloud_utils.py /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NolenChen/3DStructurePoints/HEAD/README.md -------------------------------------------------------------------------------- /dataset/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NolenChen/3DStructurePoints/HEAD/dataset/__init__.py -------------------------------------------------------------------------------- /dataset/bhcp_dataloader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NolenChen/3DStructurePoints/HEAD/dataset/bhcp_dataloader.py -------------------------------------------------------------------------------- /dataset/data_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NolenChen/3DStructurePoints/HEAD/dataset/data_utils.py -------------------------------------------------------------------------------- /dataset/shapenet_seg_dataloader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NolenChen/3DStructurePoints/HEAD/dataset/shapenet_seg_dataloader.py -------------------------------------------------------------------------------- /demo_data/bike/0.off: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NolenChen/3DStructurePoints/HEAD/demo_data/bike/0.off -------------------------------------------------------------------------------- /demo_data/bike/1.off: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NolenChen/3DStructurePoints/HEAD/demo_data/bike/1.off -------------------------------------------------------------------------------- /demo_data/bike/2.off: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NolenChen/3DStructurePoints/HEAD/demo_data/bike/2.off -------------------------------------------------------------------------------- /demo_data/chair/0.off: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NolenChen/3DStructurePoints/HEAD/demo_data/chair/0.off -------------------------------------------------------------------------------- /demo_data/chair/1.off: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NolenChen/3DStructurePoints/HEAD/demo_data/chair/1.off -------------------------------------------------------------------------------- /demo_data/chair/2.off: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NolenChen/3DStructurePoints/HEAD/demo_data/chair/2.off -------------------------------------------------------------------------------- /demo_data/plane/0.off: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NolenChen/3DStructurePoints/HEAD/demo_data/plane/0.off -------------------------------------------------------------------------------- /demo_data/plane/1.off: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NolenChen/3DStructurePoints/HEAD/demo_data/plane/1.off -------------------------------------------------------------------------------- /demo_data/plane/2.off: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NolenChen/3DStructurePoints/HEAD/demo_data/plane/2.off -------------------------------------------------------------------------------- /demo_data/realscan_chair/0.off: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NolenChen/3DStructurePoints/HEAD/demo_data/realscan_chair/0.off -------------------------------------------------------------------------------- /demo_data/realscan_chair/1.off: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NolenChen/3DStructurePoints/HEAD/demo_data/realscan_chair/1.off -------------------------------------------------------------------------------- /demo_data/realscan_chair/2.off: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NolenChen/3DStructurePoints/HEAD/demo_data/realscan_chair/2.off -------------------------------------------------------------------------------- /demo_data/shape_corres/src_and_query_pts/bike_query_pts.off: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NolenChen/3DStructurePoints/HEAD/demo_data/shape_corres/src_and_query_pts/bike_query_pts.off -------------------------------------------------------------------------------- /demo_data/shape_corres/src_and_query_pts/bike_src_pts.off: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NolenChen/3DStructurePoints/HEAD/demo_data/shape_corres/src_and_query_pts/bike_src_pts.off -------------------------------------------------------------------------------- /demo_data/shape_corres/src_and_query_pts/chair_query_pts.off: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NolenChen/3DStructurePoints/HEAD/demo_data/shape_corres/src_and_query_pts/chair_query_pts.off -------------------------------------------------------------------------------- /demo_data/shape_corres/src_and_query_pts/chair_src_pts.off: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NolenChen/3DStructurePoints/HEAD/demo_data/shape_corres/src_and_query_pts/chair_src_pts.off -------------------------------------------------------------------------------- /demo_data/shape_corres/src_and_query_pts/plane_query_pts.off: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NolenChen/3DStructurePoints/HEAD/demo_data/shape_corres/src_and_query_pts/plane_query_pts.off -------------------------------------------------------------------------------- /demo_data/shape_corres/src_and_query_pts/plane_src_pts.off: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NolenChen/3DStructurePoints/HEAD/demo_data/shape_corres/src_and_query_pts/plane_src_pts.off -------------------------------------------------------------------------------- /demo_data/shape_corres/tgt_pts/bike_tgt_pts_0.off: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NolenChen/3DStructurePoints/HEAD/demo_data/shape_corres/tgt_pts/bike_tgt_pts_0.off -------------------------------------------------------------------------------- /demo_data/shape_corres/tgt_pts/bike_tgt_pts_1.off: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NolenChen/3DStructurePoints/HEAD/demo_data/shape_corres/tgt_pts/bike_tgt_pts_1.off -------------------------------------------------------------------------------- /demo_data/shape_corres/tgt_pts/bike_tgt_pts_2.off: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NolenChen/3DStructurePoints/HEAD/demo_data/shape_corres/tgt_pts/bike_tgt_pts_2.off -------------------------------------------------------------------------------- /demo_data/shape_corres/tgt_pts/chair_tgt_pts_0.off: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NolenChen/3DStructurePoints/HEAD/demo_data/shape_corres/tgt_pts/chair_tgt_pts_0.off -------------------------------------------------------------------------------- /demo_data/shape_corres/tgt_pts/chair_tgt_pts_1.off: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NolenChen/3DStructurePoints/HEAD/demo_data/shape_corres/tgt_pts/chair_tgt_pts_1.off -------------------------------------------------------------------------------- /demo_data/shape_corres/tgt_pts/chair_tgt_pts_2.off: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NolenChen/3DStructurePoints/HEAD/demo_data/shape_corres/tgt_pts/chair_tgt_pts_2.off -------------------------------------------------------------------------------- /demo_data/shape_corres/tgt_pts/plane_tgt_pts_0.off: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NolenChen/3DStructurePoints/HEAD/demo_data/shape_corres/tgt_pts/plane_tgt_pts_0.off -------------------------------------------------------------------------------- /demo_data/shape_corres/tgt_pts/plane_tgt_pts_1.off: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NolenChen/3DStructurePoints/HEAD/demo_data/shape_corres/tgt_pts/plane_tgt_pts_1.off -------------------------------------------------------------------------------- /demo_data/shape_corres/tgt_pts/plane_tgt_pts_2.off: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NolenChen/3DStructurePoints/HEAD/demo_data/shape_corres/tgt_pts/plane_tgt_pts_2.off -------------------------------------------------------------------------------- /docs/glab.css: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NolenChen/3DStructurePoints/HEAD/docs/glab.css -------------------------------------------------------------------------------- /docs/images/data_ico.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NolenChen/3DStructurePoints/HEAD/docs/images/data_ico.jpg -------------------------------------------------------------------------------- /docs/images/paper_thumbnail.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NolenChen/3DStructurePoints/HEAD/docs/images/paper_thumbnail.jpg -------------------------------------------------------------------------------- /docs/images/paper_thumbnail_ori.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NolenChen/3DStructurePoints/HEAD/docs/images/paper_thumbnail_ori.jpg -------------------------------------------------------------------------------- /docs/images/pipeline.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NolenChen/3DStructurePoints/HEAD/docs/images/pipeline.jpg -------------------------------------------------------------------------------- /docs/images/teaser.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NolenChen/3DStructurePoints/HEAD/docs/images/teaser.jpg -------------------------------------------------------------------------------- /docs/index.html: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NolenChen/3DStructurePoints/HEAD/docs/index.html -------------------------------------------------------------------------------- /evaluate/evaluate_corres_accuracy.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NolenChen/3DStructurePoints/HEAD/evaluate/evaluate_corres_accuracy.py -------------------------------------------------------------------------------- /evaluate/evaluate_label_transfer_accuracy.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NolenChen/3DStructurePoints/HEAD/evaluate/evaluate_label_transfer_accuracy.py -------------------------------------------------------------------------------- /figs/teaser.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NolenChen/3DStructurePoints/HEAD/figs/teaser.png -------------------------------------------------------------------------------- /models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NolenChen/3DStructurePoints/HEAD/models/__init__.py -------------------------------------------------------------------------------- /models/chamfer_distance.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NolenChen/3DStructurePoints/HEAD/models/chamfer_distance.py -------------------------------------------------------------------------------- /models/pointnet2_structure_point_net.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NolenChen/3DStructurePoints/HEAD/models/pointnet2_structure_point_net.py -------------------------------------------------------------------------------- /pointnet2/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NolenChen/3DStructurePoints/HEAD/pointnet2/__init__.py -------------------------------------------------------------------------------- /pointnet2/_ext-src/include/ball_query.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NolenChen/3DStructurePoints/HEAD/pointnet2/_ext-src/include/ball_query.h -------------------------------------------------------------------------------- /pointnet2/_ext-src/include/cuda_utils.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NolenChen/3DStructurePoints/HEAD/pointnet2/_ext-src/include/cuda_utils.h -------------------------------------------------------------------------------- /pointnet2/_ext-src/include/group_points.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NolenChen/3DStructurePoints/HEAD/pointnet2/_ext-src/include/group_points.h -------------------------------------------------------------------------------- /pointnet2/_ext-src/include/interpolate.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NolenChen/3DStructurePoints/HEAD/pointnet2/_ext-src/include/interpolate.h -------------------------------------------------------------------------------- /pointnet2/_ext-src/include/sampling.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NolenChen/3DStructurePoints/HEAD/pointnet2/_ext-src/include/sampling.h -------------------------------------------------------------------------------- /pointnet2/_ext-src/include/utils.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NolenChen/3DStructurePoints/HEAD/pointnet2/_ext-src/include/utils.h -------------------------------------------------------------------------------- /pointnet2/_ext-src/src/ball_query.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NolenChen/3DStructurePoints/HEAD/pointnet2/_ext-src/src/ball_query.cpp -------------------------------------------------------------------------------- /pointnet2/_ext-src/src/ball_query_gpu.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NolenChen/3DStructurePoints/HEAD/pointnet2/_ext-src/src/ball_query_gpu.cu -------------------------------------------------------------------------------- /pointnet2/_ext-src/src/bindings.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NolenChen/3DStructurePoints/HEAD/pointnet2/_ext-src/src/bindings.cpp -------------------------------------------------------------------------------- /pointnet2/_ext-src/src/group_points.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NolenChen/3DStructurePoints/HEAD/pointnet2/_ext-src/src/group_points.cpp -------------------------------------------------------------------------------- /pointnet2/_ext-src/src/group_points_gpu.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NolenChen/3DStructurePoints/HEAD/pointnet2/_ext-src/src/group_points_gpu.cu -------------------------------------------------------------------------------- /pointnet2/_ext-src/src/interpolate.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NolenChen/3DStructurePoints/HEAD/pointnet2/_ext-src/src/interpolate.cpp -------------------------------------------------------------------------------- /pointnet2/_ext-src/src/interpolate_gpu.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NolenChen/3DStructurePoints/HEAD/pointnet2/_ext-src/src/interpolate_gpu.cu -------------------------------------------------------------------------------- /pointnet2/_ext-src/src/sampling.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NolenChen/3DStructurePoints/HEAD/pointnet2/_ext-src/src/sampling.cpp -------------------------------------------------------------------------------- /pointnet2/_ext-src/src/sampling_gpu.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NolenChen/3DStructurePoints/HEAD/pointnet2/_ext-src/src/sampling_gpu.cu -------------------------------------------------------------------------------- /pointnet2/setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NolenChen/3DStructurePoints/HEAD/pointnet2/setup.py -------------------------------------------------------------------------------- /pointnet2/utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NolenChen/3DStructurePoints/HEAD/pointnet2/utils/__init__.py -------------------------------------------------------------------------------- /pointnet2/utils/etw_pytorch_utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NolenChen/3DStructurePoints/HEAD/pointnet2/utils/etw_pytorch_utils/__init__.py -------------------------------------------------------------------------------- /pointnet2/utils/etw_pytorch_utils/persistent_dataloader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NolenChen/3DStructurePoints/HEAD/pointnet2/utils/etw_pytorch_utils/persistent_dataloader.py -------------------------------------------------------------------------------- /pointnet2/utils/etw_pytorch_utils/pytorch_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NolenChen/3DStructurePoints/HEAD/pointnet2/utils/etw_pytorch_utils/pytorch_utils.py -------------------------------------------------------------------------------- /pointnet2/utils/etw_pytorch_utils/sacred_trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NolenChen/3DStructurePoints/HEAD/pointnet2/utils/etw_pytorch_utils/sacred_trainer.py -------------------------------------------------------------------------------- /pointnet2/utils/etw_pytorch_utils/seq.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NolenChen/3DStructurePoints/HEAD/pointnet2/utils/etw_pytorch_utils/seq.py -------------------------------------------------------------------------------- /pointnet2/utils/etw_pytorch_utils/visdom_observer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NolenChen/3DStructurePoints/HEAD/pointnet2/utils/etw_pytorch_utils/visdom_observer.py -------------------------------------------------------------------------------- /pointnet2/utils/etw_pytorch_utils/viz.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NolenChen/3DStructurePoints/HEAD/pointnet2/utils/etw_pytorch_utils/viz.py -------------------------------------------------------------------------------- /pointnet2/utils/linalg_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NolenChen/3DStructurePoints/HEAD/pointnet2/utils/linalg_utils.py -------------------------------------------------------------------------------- /pointnet2/utils/pointnet2_modules.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NolenChen/3DStructurePoints/HEAD/pointnet2/utils/pointnet2_modules.py -------------------------------------------------------------------------------- /pointnet2/utils/pointnet2_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NolenChen/3DStructurePoints/HEAD/pointnet2/utils/pointnet2_utils.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NolenChen/3DStructurePoints/HEAD/requirements.txt -------------------------------------------------------------------------------- /test/test_shape_correspondence.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NolenChen/3DStructurePoints/HEAD/test/test_shape_correspondence.py -------------------------------------------------------------------------------- /test/test_structure_points.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NolenChen/3DStructurePoints/HEAD/test/test_structure_points.py -------------------------------------------------------------------------------- /train/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /train/train_structure_points.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NolenChen/3DStructurePoints/HEAD/train/train_structure_points.py -------------------------------------------------------------------------------- /train/train_structure_points_for_label_transfer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NolenChen/3DStructurePoints/HEAD/train/train_structure_points_for_label_transfer.py -------------------------------------------------------------------------------- /utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NolenChen/3DStructurePoints/HEAD/utils/__init__.py -------------------------------------------------------------------------------- /utils/check_points_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NolenChen/3DStructurePoints/HEAD/utils/check_points_utils.py -------------------------------------------------------------------------------- /utils/logutils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NolenChen/3DStructurePoints/HEAD/utils/logutils.py -------------------------------------------------------------------------------- /utils/mesh_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NolenChen/3DStructurePoints/HEAD/utils/mesh_utils.py -------------------------------------------------------------------------------- /utils/point_cloud_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NolenChen/3DStructurePoints/HEAD/utils/point_cloud_utils.py --------------------------------------------------------------------------------