├── LICENSE ├── README.md ├── config └── base.yaml ├── dependencies..sh ├── doc └── overview.png ├── lib ├── pointnet2 │ ├── __pycache__ │ │ ├── pointnet2_modules.cpython-36.pyc │ │ ├── pointnet2_utils.cpython-36.pyc │ │ └── pytorch_utils.cpython-36.pyc │ ├── _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 │ ├── pointnet2_modules.py │ ├── pointnet2_test.py │ ├── pointnet2_utils.py │ ├── pytorch_utils.py │ └── setup.py └── sphericalmap_utils │ ├── _ext_src │ ├── include │ │ ├── cuda_utils.h │ │ ├── sphericalmap.h │ │ └── utils.h │ └── src │ │ ├── bindings.cpp │ │ ├── sphericalmap.cpp │ │ └── sphericalmap_cuda.cu │ ├── setup.py │ └── smap_utils.py ├── model ├── PN2.py ├── VI_Net.py ├── layer.py ├── loss.py └── module.py ├── provider └── dataset.py ├── test.py ├── train.py └── utils ├── CATRE_evaluation_utils.py ├── data_utils.py ├── draw_utils.py ├── evaluation_utils.py ├── rotation_utils.py ├── solver.py └── spherical_utils.py /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JiehongLin/VI-Net/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JiehongLin/VI-Net/HEAD/README.md -------------------------------------------------------------------------------- /config/base.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JiehongLin/VI-Net/HEAD/config/base.yaml -------------------------------------------------------------------------------- /dependencies..sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JiehongLin/VI-Net/HEAD/dependencies..sh -------------------------------------------------------------------------------- /doc/overview.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JiehongLin/VI-Net/HEAD/doc/overview.png -------------------------------------------------------------------------------- /lib/pointnet2/__pycache__/pointnet2_modules.cpython-36.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JiehongLin/VI-Net/HEAD/lib/pointnet2/__pycache__/pointnet2_modules.cpython-36.pyc -------------------------------------------------------------------------------- /lib/pointnet2/__pycache__/pointnet2_utils.cpython-36.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JiehongLin/VI-Net/HEAD/lib/pointnet2/__pycache__/pointnet2_utils.cpython-36.pyc -------------------------------------------------------------------------------- /lib/pointnet2/__pycache__/pytorch_utils.cpython-36.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JiehongLin/VI-Net/HEAD/lib/pointnet2/__pycache__/pytorch_utils.cpython-36.pyc -------------------------------------------------------------------------------- /lib/pointnet2/_ext_src/include/ball_query.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JiehongLin/VI-Net/HEAD/lib/pointnet2/_ext_src/include/ball_query.h -------------------------------------------------------------------------------- /lib/pointnet2/_ext_src/include/cuda_utils.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JiehongLin/VI-Net/HEAD/lib/pointnet2/_ext_src/include/cuda_utils.h -------------------------------------------------------------------------------- /lib/pointnet2/_ext_src/include/group_points.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JiehongLin/VI-Net/HEAD/lib/pointnet2/_ext_src/include/group_points.h -------------------------------------------------------------------------------- /lib/pointnet2/_ext_src/include/interpolate.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JiehongLin/VI-Net/HEAD/lib/pointnet2/_ext_src/include/interpolate.h -------------------------------------------------------------------------------- /lib/pointnet2/_ext_src/include/sampling.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JiehongLin/VI-Net/HEAD/lib/pointnet2/_ext_src/include/sampling.h -------------------------------------------------------------------------------- /lib/pointnet2/_ext_src/include/utils.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JiehongLin/VI-Net/HEAD/lib/pointnet2/_ext_src/include/utils.h -------------------------------------------------------------------------------- /lib/pointnet2/_ext_src/src/ball_query.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JiehongLin/VI-Net/HEAD/lib/pointnet2/_ext_src/src/ball_query.cpp -------------------------------------------------------------------------------- /lib/pointnet2/_ext_src/src/ball_query_gpu.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JiehongLin/VI-Net/HEAD/lib/pointnet2/_ext_src/src/ball_query_gpu.cu -------------------------------------------------------------------------------- /lib/pointnet2/_ext_src/src/bindings.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JiehongLin/VI-Net/HEAD/lib/pointnet2/_ext_src/src/bindings.cpp -------------------------------------------------------------------------------- /lib/pointnet2/_ext_src/src/group_points.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JiehongLin/VI-Net/HEAD/lib/pointnet2/_ext_src/src/group_points.cpp -------------------------------------------------------------------------------- /lib/pointnet2/_ext_src/src/group_points_gpu.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JiehongLin/VI-Net/HEAD/lib/pointnet2/_ext_src/src/group_points_gpu.cu -------------------------------------------------------------------------------- /lib/pointnet2/_ext_src/src/interpolate.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JiehongLin/VI-Net/HEAD/lib/pointnet2/_ext_src/src/interpolate.cpp -------------------------------------------------------------------------------- /lib/pointnet2/_ext_src/src/interpolate_gpu.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JiehongLin/VI-Net/HEAD/lib/pointnet2/_ext_src/src/interpolate_gpu.cu -------------------------------------------------------------------------------- /lib/pointnet2/_ext_src/src/sampling.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JiehongLin/VI-Net/HEAD/lib/pointnet2/_ext_src/src/sampling.cpp -------------------------------------------------------------------------------- /lib/pointnet2/_ext_src/src/sampling_gpu.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JiehongLin/VI-Net/HEAD/lib/pointnet2/_ext_src/src/sampling_gpu.cu -------------------------------------------------------------------------------- /lib/pointnet2/pointnet2_modules.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JiehongLin/VI-Net/HEAD/lib/pointnet2/pointnet2_modules.py -------------------------------------------------------------------------------- /lib/pointnet2/pointnet2_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JiehongLin/VI-Net/HEAD/lib/pointnet2/pointnet2_test.py -------------------------------------------------------------------------------- /lib/pointnet2/pointnet2_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JiehongLin/VI-Net/HEAD/lib/pointnet2/pointnet2_utils.py -------------------------------------------------------------------------------- /lib/pointnet2/pytorch_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JiehongLin/VI-Net/HEAD/lib/pointnet2/pytorch_utils.py -------------------------------------------------------------------------------- /lib/pointnet2/setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JiehongLin/VI-Net/HEAD/lib/pointnet2/setup.py -------------------------------------------------------------------------------- /lib/sphericalmap_utils/_ext_src/include/cuda_utils.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JiehongLin/VI-Net/HEAD/lib/sphericalmap_utils/_ext_src/include/cuda_utils.h -------------------------------------------------------------------------------- /lib/sphericalmap_utils/_ext_src/include/sphericalmap.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JiehongLin/VI-Net/HEAD/lib/sphericalmap_utils/_ext_src/include/sphericalmap.h -------------------------------------------------------------------------------- /lib/sphericalmap_utils/_ext_src/include/utils.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JiehongLin/VI-Net/HEAD/lib/sphericalmap_utils/_ext_src/include/utils.h -------------------------------------------------------------------------------- /lib/sphericalmap_utils/_ext_src/src/bindings.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JiehongLin/VI-Net/HEAD/lib/sphericalmap_utils/_ext_src/src/bindings.cpp -------------------------------------------------------------------------------- /lib/sphericalmap_utils/_ext_src/src/sphericalmap.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JiehongLin/VI-Net/HEAD/lib/sphericalmap_utils/_ext_src/src/sphericalmap.cpp -------------------------------------------------------------------------------- /lib/sphericalmap_utils/_ext_src/src/sphericalmap_cuda.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JiehongLin/VI-Net/HEAD/lib/sphericalmap_utils/_ext_src/src/sphericalmap_cuda.cu -------------------------------------------------------------------------------- /lib/sphericalmap_utils/setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JiehongLin/VI-Net/HEAD/lib/sphericalmap_utils/setup.py -------------------------------------------------------------------------------- /lib/sphericalmap_utils/smap_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JiehongLin/VI-Net/HEAD/lib/sphericalmap_utils/smap_utils.py -------------------------------------------------------------------------------- /model/PN2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JiehongLin/VI-Net/HEAD/model/PN2.py -------------------------------------------------------------------------------- /model/VI_Net.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JiehongLin/VI-Net/HEAD/model/VI_Net.py -------------------------------------------------------------------------------- /model/layer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JiehongLin/VI-Net/HEAD/model/layer.py -------------------------------------------------------------------------------- /model/loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JiehongLin/VI-Net/HEAD/model/loss.py -------------------------------------------------------------------------------- /model/module.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JiehongLin/VI-Net/HEAD/model/module.py -------------------------------------------------------------------------------- /provider/dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JiehongLin/VI-Net/HEAD/provider/dataset.py -------------------------------------------------------------------------------- /test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JiehongLin/VI-Net/HEAD/test.py -------------------------------------------------------------------------------- /train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JiehongLin/VI-Net/HEAD/train.py -------------------------------------------------------------------------------- /utils/CATRE_evaluation_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JiehongLin/VI-Net/HEAD/utils/CATRE_evaluation_utils.py -------------------------------------------------------------------------------- /utils/data_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JiehongLin/VI-Net/HEAD/utils/data_utils.py -------------------------------------------------------------------------------- /utils/draw_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JiehongLin/VI-Net/HEAD/utils/draw_utils.py -------------------------------------------------------------------------------- /utils/evaluation_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JiehongLin/VI-Net/HEAD/utils/evaluation_utils.py -------------------------------------------------------------------------------- /utils/rotation_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JiehongLin/VI-Net/HEAD/utils/rotation_utils.py -------------------------------------------------------------------------------- /utils/solver.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JiehongLin/VI-Net/HEAD/utils/solver.py -------------------------------------------------------------------------------- /utils/spherical_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JiehongLin/VI-Net/HEAD/utils/spherical_utils.py --------------------------------------------------------------------------------