├── .gitignore ├── LICENSE ├── README.md ├── config_vipc.py ├── main.py ├── metrics ├── CD │ ├── LICENSE │ ├── README.md │ ├── __init__.py │ ├── chamfer3D │ │ ├── chamfer3D.cu │ │ ├── chamfer_cuda.cpp │ │ ├── dist_chamfer_3D.py │ │ ├── run_build.sh │ │ └── setup.py │ ├── chamfer_python.py │ ├── fscore.py │ └── unit_test.py ├── EMD │ ├── README.md │ ├── __init__.py │ ├── clean.sh │ ├── emd.cpp │ ├── emd_cuda.cu │ ├── emd_module.py │ ├── run_build.sh │ ├── run_compile.sh │ └── setup.py └── __init__.py ├── models ├── EGIInet.py ├── dec_net.py ├── encoders.py ├── layers │ ├── __init__.py │ ├── activation.py │ ├── attention.py │ ├── conv.py │ ├── drop.py │ ├── graph_conv.py │ ├── group.py │ ├── group_embed.py │ ├── helpers.py │ ├── kmeans.py │ ├── knn.py │ ├── local_aggregation.py │ ├── mlp.py │ ├── norm.py │ ├── padding.py │ ├── patch_embed.py │ ├── registry.py │ ├── subsample.py │ ├── upsampling.py │ └── weight_init.py └── pointnet2_batch │ ├── __init__.py │ ├── setup.py │ └── src │ ├── ball_query.cpp │ ├── ball_query_gpu.cu │ ├── ball_query_gpu.h │ ├── cuda_utils.h │ ├── group_points.cpp │ ├── group_points_gpu.cu │ ├── group_points_gpu.h │ ├── interpolate.cpp │ ├── interpolate_gpu.cu │ ├── interpolate_gpu.h │ ├── pointnet2_api.cpp │ ├── sampling.cpp │ ├── sampling_gpu.cu │ └── sampling_gpu.h ├── project_logs └── plane │ └── logs │ ├── 24-07-03-14-32-48 │ ├── test │ │ └── events.out.tfevents.1719988368.xuhang-MS-7D07 │ └── train │ │ └── events.out.tfevents.1719988368.xuhang-MS-7D07 │ ├── 24-07-03-14-33-07 │ ├── test │ │ └── events.out.tfevents.1719988387.xuhang-MS-7D07 │ └── train │ │ └── events.out.tfevents.1719988387.xuhang-MS-7D07 │ ├── 24-07-03-14-33-26 │ ├── test │ │ └── events.out.tfevents.1719988406.xuhang-MS-7D07 │ └── train │ │ └── events.out.tfevents.1719988406.xuhang-MS-7D07 │ └── 24-07-03-14-33-46 │ ├── test │ └── events.out.tfevents.1719988426.xuhang-MS-7D07 │ └── train │ └── events.out.tfevents.1719988426.xuhang-MS-7D07 ├── run ├── test.py └── train.py ├── test_list.txt ├── train_list.txt └── utils ├── ViPCdataloader.py ├── average_meter.py ├── furthestPointSampling ├── fps.py ├── sampling.cpp ├── sampling_gpu.cu ├── sampling_gpu.h └── setup.py ├── loss_utils.py └── schedular.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WHU-USI3DV/EGIInet/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WHU-USI3DV/EGIInet/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WHU-USI3DV/EGIInet/HEAD/README.md -------------------------------------------------------------------------------- /config_vipc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WHU-USI3DV/EGIInet/HEAD/config_vipc.py -------------------------------------------------------------------------------- /main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WHU-USI3DV/EGIInet/HEAD/main.py -------------------------------------------------------------------------------- /metrics/CD/LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WHU-USI3DV/EGIInet/HEAD/metrics/CD/LICENSE -------------------------------------------------------------------------------- /metrics/CD/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WHU-USI3DV/EGIInet/HEAD/metrics/CD/README.md -------------------------------------------------------------------------------- /metrics/CD/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WHU-USI3DV/EGIInet/HEAD/metrics/CD/__init__.py -------------------------------------------------------------------------------- /metrics/CD/chamfer3D/chamfer3D.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WHU-USI3DV/EGIInet/HEAD/metrics/CD/chamfer3D/chamfer3D.cu -------------------------------------------------------------------------------- /metrics/CD/chamfer3D/chamfer_cuda.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WHU-USI3DV/EGIInet/HEAD/metrics/CD/chamfer3D/chamfer_cuda.cpp -------------------------------------------------------------------------------- /metrics/CD/chamfer3D/dist_chamfer_3D.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WHU-USI3DV/EGIInet/HEAD/metrics/CD/chamfer3D/dist_chamfer_3D.py -------------------------------------------------------------------------------- /metrics/CD/chamfer3D/run_build.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WHU-USI3DV/EGIInet/HEAD/metrics/CD/chamfer3D/run_build.sh -------------------------------------------------------------------------------- /metrics/CD/chamfer3D/setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WHU-USI3DV/EGIInet/HEAD/metrics/CD/chamfer3D/setup.py -------------------------------------------------------------------------------- /metrics/CD/chamfer_python.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WHU-USI3DV/EGIInet/HEAD/metrics/CD/chamfer_python.py -------------------------------------------------------------------------------- /metrics/CD/fscore.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WHU-USI3DV/EGIInet/HEAD/metrics/CD/fscore.py -------------------------------------------------------------------------------- /metrics/CD/unit_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WHU-USI3DV/EGIInet/HEAD/metrics/CD/unit_test.py -------------------------------------------------------------------------------- /metrics/EMD/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WHU-USI3DV/EGIInet/HEAD/metrics/EMD/README.md -------------------------------------------------------------------------------- /metrics/EMD/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WHU-USI3DV/EGIInet/HEAD/metrics/EMD/__init__.py -------------------------------------------------------------------------------- /metrics/EMD/clean.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WHU-USI3DV/EGIInet/HEAD/metrics/EMD/clean.sh -------------------------------------------------------------------------------- /metrics/EMD/emd.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WHU-USI3DV/EGIInet/HEAD/metrics/EMD/emd.cpp -------------------------------------------------------------------------------- /metrics/EMD/emd_cuda.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WHU-USI3DV/EGIInet/HEAD/metrics/EMD/emd_cuda.cu -------------------------------------------------------------------------------- /metrics/EMD/emd_module.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WHU-USI3DV/EGIInet/HEAD/metrics/EMD/emd_module.py -------------------------------------------------------------------------------- /metrics/EMD/run_build.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WHU-USI3DV/EGIInet/HEAD/metrics/EMD/run_build.sh -------------------------------------------------------------------------------- /metrics/EMD/run_compile.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WHU-USI3DV/EGIInet/HEAD/metrics/EMD/run_compile.sh -------------------------------------------------------------------------------- /metrics/EMD/setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WHU-USI3DV/EGIInet/HEAD/metrics/EMD/setup.py -------------------------------------------------------------------------------- /metrics/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WHU-USI3DV/EGIInet/HEAD/metrics/__init__.py -------------------------------------------------------------------------------- /models/EGIInet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WHU-USI3DV/EGIInet/HEAD/models/EGIInet.py -------------------------------------------------------------------------------- /models/dec_net.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WHU-USI3DV/EGIInet/HEAD/models/dec_net.py -------------------------------------------------------------------------------- /models/encoders.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WHU-USI3DV/EGIInet/HEAD/models/encoders.py -------------------------------------------------------------------------------- /models/layers/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WHU-USI3DV/EGIInet/HEAD/models/layers/__init__.py -------------------------------------------------------------------------------- /models/layers/activation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WHU-USI3DV/EGIInet/HEAD/models/layers/activation.py -------------------------------------------------------------------------------- /models/layers/attention.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WHU-USI3DV/EGIInet/HEAD/models/layers/attention.py -------------------------------------------------------------------------------- /models/layers/conv.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WHU-USI3DV/EGIInet/HEAD/models/layers/conv.py -------------------------------------------------------------------------------- /models/layers/drop.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WHU-USI3DV/EGIInet/HEAD/models/layers/drop.py -------------------------------------------------------------------------------- /models/layers/graph_conv.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WHU-USI3DV/EGIInet/HEAD/models/layers/graph_conv.py -------------------------------------------------------------------------------- /models/layers/group.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WHU-USI3DV/EGIInet/HEAD/models/layers/group.py -------------------------------------------------------------------------------- /models/layers/group_embed.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WHU-USI3DV/EGIInet/HEAD/models/layers/group_embed.py -------------------------------------------------------------------------------- /models/layers/helpers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WHU-USI3DV/EGIInet/HEAD/models/layers/helpers.py -------------------------------------------------------------------------------- /models/layers/kmeans.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WHU-USI3DV/EGIInet/HEAD/models/layers/kmeans.py -------------------------------------------------------------------------------- /models/layers/knn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WHU-USI3DV/EGIInet/HEAD/models/layers/knn.py -------------------------------------------------------------------------------- /models/layers/local_aggregation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WHU-USI3DV/EGIInet/HEAD/models/layers/local_aggregation.py -------------------------------------------------------------------------------- /models/layers/mlp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WHU-USI3DV/EGIInet/HEAD/models/layers/mlp.py -------------------------------------------------------------------------------- /models/layers/norm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WHU-USI3DV/EGIInet/HEAD/models/layers/norm.py -------------------------------------------------------------------------------- /models/layers/padding.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WHU-USI3DV/EGIInet/HEAD/models/layers/padding.py -------------------------------------------------------------------------------- /models/layers/patch_embed.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WHU-USI3DV/EGIInet/HEAD/models/layers/patch_embed.py -------------------------------------------------------------------------------- /models/layers/registry.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WHU-USI3DV/EGIInet/HEAD/models/layers/registry.py -------------------------------------------------------------------------------- /models/layers/subsample.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WHU-USI3DV/EGIInet/HEAD/models/layers/subsample.py -------------------------------------------------------------------------------- /models/layers/upsampling.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WHU-USI3DV/EGIInet/HEAD/models/layers/upsampling.py -------------------------------------------------------------------------------- /models/layers/weight_init.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WHU-USI3DV/EGIInet/HEAD/models/layers/weight_init.py -------------------------------------------------------------------------------- /models/pointnet2_batch/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WHU-USI3DV/EGIInet/HEAD/models/pointnet2_batch/__init__.py -------------------------------------------------------------------------------- /models/pointnet2_batch/setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WHU-USI3DV/EGIInet/HEAD/models/pointnet2_batch/setup.py -------------------------------------------------------------------------------- /models/pointnet2_batch/src/ball_query.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WHU-USI3DV/EGIInet/HEAD/models/pointnet2_batch/src/ball_query.cpp -------------------------------------------------------------------------------- /models/pointnet2_batch/src/ball_query_gpu.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WHU-USI3DV/EGIInet/HEAD/models/pointnet2_batch/src/ball_query_gpu.cu -------------------------------------------------------------------------------- /models/pointnet2_batch/src/ball_query_gpu.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WHU-USI3DV/EGIInet/HEAD/models/pointnet2_batch/src/ball_query_gpu.h -------------------------------------------------------------------------------- /models/pointnet2_batch/src/cuda_utils.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WHU-USI3DV/EGIInet/HEAD/models/pointnet2_batch/src/cuda_utils.h -------------------------------------------------------------------------------- /models/pointnet2_batch/src/group_points.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WHU-USI3DV/EGIInet/HEAD/models/pointnet2_batch/src/group_points.cpp -------------------------------------------------------------------------------- /models/pointnet2_batch/src/group_points_gpu.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WHU-USI3DV/EGIInet/HEAD/models/pointnet2_batch/src/group_points_gpu.cu -------------------------------------------------------------------------------- /models/pointnet2_batch/src/group_points_gpu.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WHU-USI3DV/EGIInet/HEAD/models/pointnet2_batch/src/group_points_gpu.h -------------------------------------------------------------------------------- /models/pointnet2_batch/src/interpolate.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WHU-USI3DV/EGIInet/HEAD/models/pointnet2_batch/src/interpolate.cpp -------------------------------------------------------------------------------- /models/pointnet2_batch/src/interpolate_gpu.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WHU-USI3DV/EGIInet/HEAD/models/pointnet2_batch/src/interpolate_gpu.cu -------------------------------------------------------------------------------- /models/pointnet2_batch/src/interpolate_gpu.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WHU-USI3DV/EGIInet/HEAD/models/pointnet2_batch/src/interpolate_gpu.h -------------------------------------------------------------------------------- /models/pointnet2_batch/src/pointnet2_api.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WHU-USI3DV/EGIInet/HEAD/models/pointnet2_batch/src/pointnet2_api.cpp -------------------------------------------------------------------------------- /models/pointnet2_batch/src/sampling.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WHU-USI3DV/EGIInet/HEAD/models/pointnet2_batch/src/sampling.cpp -------------------------------------------------------------------------------- /models/pointnet2_batch/src/sampling_gpu.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WHU-USI3DV/EGIInet/HEAD/models/pointnet2_batch/src/sampling_gpu.cu -------------------------------------------------------------------------------- /models/pointnet2_batch/src/sampling_gpu.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WHU-USI3DV/EGIInet/HEAD/models/pointnet2_batch/src/sampling_gpu.h -------------------------------------------------------------------------------- /project_logs/plane/logs/24-07-03-14-32-48/test/events.out.tfevents.1719988368.xuhang-MS-7D07: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WHU-USI3DV/EGIInet/HEAD/project_logs/plane/logs/24-07-03-14-32-48/test/events.out.tfevents.1719988368.xuhang-MS-7D07 -------------------------------------------------------------------------------- /project_logs/plane/logs/24-07-03-14-32-48/train/events.out.tfevents.1719988368.xuhang-MS-7D07: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WHU-USI3DV/EGIInet/HEAD/project_logs/plane/logs/24-07-03-14-32-48/train/events.out.tfevents.1719988368.xuhang-MS-7D07 -------------------------------------------------------------------------------- /project_logs/plane/logs/24-07-03-14-33-07/test/events.out.tfevents.1719988387.xuhang-MS-7D07: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WHU-USI3DV/EGIInet/HEAD/project_logs/plane/logs/24-07-03-14-33-07/test/events.out.tfevents.1719988387.xuhang-MS-7D07 -------------------------------------------------------------------------------- /project_logs/plane/logs/24-07-03-14-33-07/train/events.out.tfevents.1719988387.xuhang-MS-7D07: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WHU-USI3DV/EGIInet/HEAD/project_logs/plane/logs/24-07-03-14-33-07/train/events.out.tfevents.1719988387.xuhang-MS-7D07 -------------------------------------------------------------------------------- /project_logs/plane/logs/24-07-03-14-33-26/test/events.out.tfevents.1719988406.xuhang-MS-7D07: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WHU-USI3DV/EGIInet/HEAD/project_logs/plane/logs/24-07-03-14-33-26/test/events.out.tfevents.1719988406.xuhang-MS-7D07 -------------------------------------------------------------------------------- /project_logs/plane/logs/24-07-03-14-33-26/train/events.out.tfevents.1719988406.xuhang-MS-7D07: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WHU-USI3DV/EGIInet/HEAD/project_logs/plane/logs/24-07-03-14-33-26/train/events.out.tfevents.1719988406.xuhang-MS-7D07 -------------------------------------------------------------------------------- /project_logs/plane/logs/24-07-03-14-33-46/test/events.out.tfevents.1719988426.xuhang-MS-7D07: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /project_logs/plane/logs/24-07-03-14-33-46/train/events.out.tfevents.1719988426.xuhang-MS-7D07: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /run/test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WHU-USI3DV/EGIInet/HEAD/run/test.py -------------------------------------------------------------------------------- /run/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WHU-USI3DV/EGIInet/HEAD/run/train.py -------------------------------------------------------------------------------- /test_list.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WHU-USI3DV/EGIInet/HEAD/test_list.txt -------------------------------------------------------------------------------- /train_list.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WHU-USI3DV/EGIInet/HEAD/train_list.txt -------------------------------------------------------------------------------- /utils/ViPCdataloader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WHU-USI3DV/EGIInet/HEAD/utils/ViPCdataloader.py -------------------------------------------------------------------------------- /utils/average_meter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WHU-USI3DV/EGIInet/HEAD/utils/average_meter.py -------------------------------------------------------------------------------- /utils/furthestPointSampling/fps.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WHU-USI3DV/EGIInet/HEAD/utils/furthestPointSampling/fps.py -------------------------------------------------------------------------------- /utils/furthestPointSampling/sampling.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WHU-USI3DV/EGIInet/HEAD/utils/furthestPointSampling/sampling.cpp -------------------------------------------------------------------------------- /utils/furthestPointSampling/sampling_gpu.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WHU-USI3DV/EGIInet/HEAD/utils/furthestPointSampling/sampling_gpu.cu -------------------------------------------------------------------------------- /utils/furthestPointSampling/sampling_gpu.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WHU-USI3DV/EGIInet/HEAD/utils/furthestPointSampling/sampling_gpu.h -------------------------------------------------------------------------------- /utils/furthestPointSampling/setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WHU-USI3DV/EGIInet/HEAD/utils/furthestPointSampling/setup.py -------------------------------------------------------------------------------- /utils/loss_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WHU-USI3DV/EGIInet/HEAD/utils/loss_utils.py -------------------------------------------------------------------------------- /utils/schedular.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WHU-USI3DV/EGIInet/HEAD/utils/schedular.py --------------------------------------------------------------------------------