├── 3DHPA.yml ├── README.md ├── exps ├── method │ ├── __init__.py │ ├── comp_distance.py │ ├── count_part.py │ ├── datasets │ │ ├── partnet.py │ │ ├── partnet_debug.py │ │ └── partnet_vis.py │ ├── execute.py │ ├── hierarchical_eval.py │ ├── hierarchical_train.py │ ├── inference_hpa.sh │ ├── models │ │ ├── SAmodule.py │ │ ├── __init__.py │ │ ├── build_model.py │ │ ├── func.py │ │ ├── hierarchical_model.py │ │ ├── inference.py │ │ ├── losses.py │ │ ├── models.py │ │ ├── pointnet │ │ │ ├── __init__.py │ │ │ ├── layers.py │ │ │ ├── pointnet_cls.py │ │ │ ├── pointnet_part_seg.py │ │ │ ├── pointnet_sem_seg.py │ │ │ ├── pointnet_utils.py │ │ │ └── utils.py │ │ └── pointnet_util.py │ ├── scripts │ │ ├── __init__.py │ │ ├── d_utils.py │ │ ├── vis.py │ │ └── z_utils.py │ └── train_hpa.sh └── utils │ ├── anchor.pts │ ├── camera_at_2.blend │ ├── camera_centered.blend │ ├── cd │ ├── .gitignore │ ├── __init__.py │ ├── chamfer.py │ ├── chamfer_cuda.cpython-37m-x86_64-linux-gnu.so │ ├── cuda │ │ ├── chamfer.cpp │ │ ├── chamfer_kernel.cu │ │ └── chamfer_kernel_old.cu │ ├── setup.py │ └── test_chamfer.py │ ├── chamfer │ ├── .gitignore │ ├── __init__.py │ ├── chamfer.py │ ├── cuda │ │ ├── chamfer.cpp │ │ └── chamfer_kernel.cu │ ├── setup.py │ └── test_chamfer.py │ ├── colors.py │ ├── cube.mtl │ ├── cube.obj │ ├── cube.pts │ ├── data.npy │ ├── data.png.mtl │ ├── emd │ ├── .gitignore │ ├── __init__.py │ ├── cuda │ │ ├── emd.cpp │ │ └── emd_kernel.cu │ ├── emd.py │ ├── emd_ext.egg-info │ │ ├── PKG-INFO │ │ ├── SOURCES.txt │ │ ├── dependency_links.txt │ │ └── top_level.txt │ ├── setup.py │ └── test_emd_loss.py │ ├── gen_html_hierarchy_local.py │ ├── geometry_utils.py │ ├── object_centered.blend │ ├── quaternion.py │ ├── rand_cmap.py │ ├── render_blender.py │ └── render_using_blender.py ├── images ├── framework.png └── teaser.png └── prep_data ├── prepare_contact_points.py ├── prepare_shape.py └── pyquaternion ├── __init__.py └── quaternion.py /3DHPA.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pkudba/3DHPA/HEAD/3DHPA.yml -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pkudba/3DHPA/HEAD/README.md -------------------------------------------------------------------------------- /exps/method/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /exps/method/comp_distance.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pkudba/3DHPA/HEAD/exps/method/comp_distance.py -------------------------------------------------------------------------------- /exps/method/count_part.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pkudba/3DHPA/HEAD/exps/method/count_part.py -------------------------------------------------------------------------------- /exps/method/datasets/partnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pkudba/3DHPA/HEAD/exps/method/datasets/partnet.py -------------------------------------------------------------------------------- /exps/method/datasets/partnet_debug.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pkudba/3DHPA/HEAD/exps/method/datasets/partnet_debug.py -------------------------------------------------------------------------------- /exps/method/datasets/partnet_vis.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pkudba/3DHPA/HEAD/exps/method/datasets/partnet_vis.py -------------------------------------------------------------------------------- /exps/method/execute.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pkudba/3DHPA/HEAD/exps/method/execute.py -------------------------------------------------------------------------------- /exps/method/hierarchical_eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pkudba/3DHPA/HEAD/exps/method/hierarchical_eval.py -------------------------------------------------------------------------------- /exps/method/hierarchical_train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pkudba/3DHPA/HEAD/exps/method/hierarchical_train.py -------------------------------------------------------------------------------- /exps/method/inference_hpa.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pkudba/3DHPA/HEAD/exps/method/inference_hpa.sh -------------------------------------------------------------------------------- /exps/method/models/SAmodule.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pkudba/3DHPA/HEAD/exps/method/models/SAmodule.py -------------------------------------------------------------------------------- /exps/method/models/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /exps/method/models/build_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pkudba/3DHPA/HEAD/exps/method/models/build_model.py -------------------------------------------------------------------------------- /exps/method/models/func.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pkudba/3DHPA/HEAD/exps/method/models/func.py -------------------------------------------------------------------------------- /exps/method/models/hierarchical_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pkudba/3DHPA/HEAD/exps/method/models/hierarchical_model.py -------------------------------------------------------------------------------- /exps/method/models/inference.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pkudba/3DHPA/HEAD/exps/method/models/inference.py -------------------------------------------------------------------------------- /exps/method/models/losses.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pkudba/3DHPA/HEAD/exps/method/models/losses.py -------------------------------------------------------------------------------- /exps/method/models/models.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pkudba/3DHPA/HEAD/exps/method/models/models.py -------------------------------------------------------------------------------- /exps/method/models/pointnet/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /exps/method/models/pointnet/layers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pkudba/3DHPA/HEAD/exps/method/models/pointnet/layers.py -------------------------------------------------------------------------------- /exps/method/models/pointnet/pointnet_cls.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pkudba/3DHPA/HEAD/exps/method/models/pointnet/pointnet_cls.py -------------------------------------------------------------------------------- /exps/method/models/pointnet/pointnet_part_seg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pkudba/3DHPA/HEAD/exps/method/models/pointnet/pointnet_part_seg.py -------------------------------------------------------------------------------- /exps/method/models/pointnet/pointnet_sem_seg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pkudba/3DHPA/HEAD/exps/method/models/pointnet/pointnet_sem_seg.py -------------------------------------------------------------------------------- /exps/method/models/pointnet/pointnet_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pkudba/3DHPA/HEAD/exps/method/models/pointnet/pointnet_utils.py -------------------------------------------------------------------------------- /exps/method/models/pointnet/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pkudba/3DHPA/HEAD/exps/method/models/pointnet/utils.py -------------------------------------------------------------------------------- /exps/method/models/pointnet_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pkudba/3DHPA/HEAD/exps/method/models/pointnet_util.py -------------------------------------------------------------------------------- /exps/method/scripts/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /exps/method/scripts/d_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pkudba/3DHPA/HEAD/exps/method/scripts/d_utils.py -------------------------------------------------------------------------------- /exps/method/scripts/vis.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pkudba/3DHPA/HEAD/exps/method/scripts/vis.py -------------------------------------------------------------------------------- /exps/method/scripts/z_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pkudba/3DHPA/HEAD/exps/method/scripts/z_utils.py -------------------------------------------------------------------------------- /exps/method/train_hpa.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pkudba/3DHPA/HEAD/exps/method/train_hpa.sh -------------------------------------------------------------------------------- /exps/utils/anchor.pts: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pkudba/3DHPA/HEAD/exps/utils/anchor.pts -------------------------------------------------------------------------------- /exps/utils/camera_at_2.blend: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pkudba/3DHPA/HEAD/exps/utils/camera_at_2.blend -------------------------------------------------------------------------------- /exps/utils/camera_centered.blend: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pkudba/3DHPA/HEAD/exps/utils/camera_centered.blend -------------------------------------------------------------------------------- /exps/utils/cd/.gitignore: -------------------------------------------------------------------------------- 1 | build 2 | dist 3 | __pycache__ 4 | chamfer_ext.egg-info 5 | -------------------------------------------------------------------------------- /exps/utils/cd/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /exps/utils/cd/chamfer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pkudba/3DHPA/HEAD/exps/utils/cd/chamfer.py -------------------------------------------------------------------------------- /exps/utils/cd/chamfer_cuda.cpython-37m-x86_64-linux-gnu.so: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pkudba/3DHPA/HEAD/exps/utils/cd/chamfer_cuda.cpython-37m-x86_64-linux-gnu.so -------------------------------------------------------------------------------- /exps/utils/cd/cuda/chamfer.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pkudba/3DHPA/HEAD/exps/utils/cd/cuda/chamfer.cpp -------------------------------------------------------------------------------- /exps/utils/cd/cuda/chamfer_kernel.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pkudba/3DHPA/HEAD/exps/utils/cd/cuda/chamfer_kernel.cu -------------------------------------------------------------------------------- /exps/utils/cd/cuda/chamfer_kernel_old.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pkudba/3DHPA/HEAD/exps/utils/cd/cuda/chamfer_kernel_old.cu -------------------------------------------------------------------------------- /exps/utils/cd/setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pkudba/3DHPA/HEAD/exps/utils/cd/setup.py -------------------------------------------------------------------------------- /exps/utils/cd/test_chamfer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pkudba/3DHPA/HEAD/exps/utils/cd/test_chamfer.py -------------------------------------------------------------------------------- /exps/utils/chamfer/.gitignore: -------------------------------------------------------------------------------- 1 | build 2 | dist 3 | __pycache__ 4 | chamfer_ext.egg-info 5 | -------------------------------------------------------------------------------- /exps/utils/chamfer/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /exps/utils/chamfer/chamfer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pkudba/3DHPA/HEAD/exps/utils/chamfer/chamfer.py -------------------------------------------------------------------------------- /exps/utils/chamfer/cuda/chamfer.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pkudba/3DHPA/HEAD/exps/utils/chamfer/cuda/chamfer.cpp -------------------------------------------------------------------------------- /exps/utils/chamfer/cuda/chamfer_kernel.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pkudba/3DHPA/HEAD/exps/utils/chamfer/cuda/chamfer_kernel.cu -------------------------------------------------------------------------------- /exps/utils/chamfer/setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pkudba/3DHPA/HEAD/exps/utils/chamfer/setup.py -------------------------------------------------------------------------------- /exps/utils/chamfer/test_chamfer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pkudba/3DHPA/HEAD/exps/utils/chamfer/test_chamfer.py -------------------------------------------------------------------------------- /exps/utils/colors.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pkudba/3DHPA/HEAD/exps/utils/colors.py -------------------------------------------------------------------------------- /exps/utils/cube.mtl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pkudba/3DHPA/HEAD/exps/utils/cube.mtl -------------------------------------------------------------------------------- /exps/utils/cube.obj: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pkudba/3DHPA/HEAD/exps/utils/cube.obj -------------------------------------------------------------------------------- /exps/utils/cube.pts: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pkudba/3DHPA/HEAD/exps/utils/cube.pts -------------------------------------------------------------------------------- /exps/utils/data.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pkudba/3DHPA/HEAD/exps/utils/data.npy -------------------------------------------------------------------------------- /exps/utils/data.png.mtl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pkudba/3DHPA/HEAD/exps/utils/data.png.mtl -------------------------------------------------------------------------------- /exps/utils/emd/.gitignore: -------------------------------------------------------------------------------- 1 | build 2 | dist 3 | __pycache__ 4 | chamfer_ext.egg-info 5 | -------------------------------------------------------------------------------- /exps/utils/emd/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /exps/utils/emd/cuda/emd.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pkudba/3DHPA/HEAD/exps/utils/emd/cuda/emd.cpp -------------------------------------------------------------------------------- /exps/utils/emd/cuda/emd_kernel.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pkudba/3DHPA/HEAD/exps/utils/emd/cuda/emd_kernel.cu -------------------------------------------------------------------------------- /exps/utils/emd/emd.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pkudba/3DHPA/HEAD/exps/utils/emd/emd.py -------------------------------------------------------------------------------- /exps/utils/emd/emd_ext.egg-info/PKG-INFO: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pkudba/3DHPA/HEAD/exps/utils/emd/emd_ext.egg-info/PKG-INFO -------------------------------------------------------------------------------- /exps/utils/emd/emd_ext.egg-info/SOURCES.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pkudba/3DHPA/HEAD/exps/utils/emd/emd_ext.egg-info/SOURCES.txt -------------------------------------------------------------------------------- /exps/utils/emd/emd_ext.egg-info/dependency_links.txt: -------------------------------------------------------------------------------- 1 | 2 | -------------------------------------------------------------------------------- /exps/utils/emd/emd_ext.egg-info/top_level.txt: -------------------------------------------------------------------------------- 1 | emd_cuda 2 | -------------------------------------------------------------------------------- /exps/utils/emd/setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pkudba/3DHPA/HEAD/exps/utils/emd/setup.py -------------------------------------------------------------------------------- /exps/utils/emd/test_emd_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pkudba/3DHPA/HEAD/exps/utils/emd/test_emd_loss.py -------------------------------------------------------------------------------- /exps/utils/gen_html_hierarchy_local.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pkudba/3DHPA/HEAD/exps/utils/gen_html_hierarchy_local.py -------------------------------------------------------------------------------- /exps/utils/geometry_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pkudba/3DHPA/HEAD/exps/utils/geometry_utils.py -------------------------------------------------------------------------------- /exps/utils/object_centered.blend: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pkudba/3DHPA/HEAD/exps/utils/object_centered.blend -------------------------------------------------------------------------------- /exps/utils/quaternion.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pkudba/3DHPA/HEAD/exps/utils/quaternion.py -------------------------------------------------------------------------------- /exps/utils/rand_cmap.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pkudba/3DHPA/HEAD/exps/utils/rand_cmap.py -------------------------------------------------------------------------------- /exps/utils/render_blender.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pkudba/3DHPA/HEAD/exps/utils/render_blender.py -------------------------------------------------------------------------------- /exps/utils/render_using_blender.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pkudba/3DHPA/HEAD/exps/utils/render_using_blender.py -------------------------------------------------------------------------------- /images/framework.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pkudba/3DHPA/HEAD/images/framework.png -------------------------------------------------------------------------------- /images/teaser.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pkudba/3DHPA/HEAD/images/teaser.png -------------------------------------------------------------------------------- /prep_data/prepare_contact_points.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pkudba/3DHPA/HEAD/prep_data/prepare_contact_points.py -------------------------------------------------------------------------------- /prep_data/prepare_shape.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pkudba/3DHPA/HEAD/prep_data/prepare_shape.py -------------------------------------------------------------------------------- /prep_data/pyquaternion/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pkudba/3DHPA/HEAD/prep_data/pyquaternion/__init__.py -------------------------------------------------------------------------------- /prep_data/pyquaternion/quaternion.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pkudba/3DHPA/HEAD/prep_data/pyquaternion/quaternion.py --------------------------------------------------------------------------------