├── README.md ├── compile.sh ├── models ├── SegGCN_s3dis.py └── SegGCN_scannet.py ├── s3dis_seg ├── eval_s3dis_radius.sh ├── evaluate_s3dis_with_overlap.py ├── getLabelColor.m ├── labelcolor_labelid.mat ├── s3dis_block2scene_index.py ├── s3dis_config.py ├── s3dis_config_kpconv.py ├── s3dis_config_resnet.py ├── s3dis_gt_voxelization_sample.m ├── save_weights_as_numpy.py ├── train_s3dis.py ├── train_s3dis_resnet.sh └── visualize.m ├── scannet_seg ├── evaluate_scannet_withoverlap.py ├── getiou.m ├── scannet_block2scene_index.py ├── scannet_config.py ├── scannet_config_SegGCN.py ├── scannet_merge.m ├── scannet_merge_benchmarkEval.m └── train_scannet.py ├── tf_ops ├── buildkernel │ ├── tf_buildkernel.cpp │ ├── tf_buildkernel.py │ ├── tf_buildkernel_compile.sh │ └── tf_buildkernel_gpu.cu ├── convolution │ ├── tf_conv3d.cpp │ ├── tf_conv3d.py │ ├── tf_conv3d_compile.sh │ └── tf_conv3d_gpu.cu ├── nnquery │ ├── tf_nnquery.cpp │ ├── tf_nnquery.py │ ├── tf_nnquery_compile.sh │ └── tf_nnquery_gpu.cu ├── pooling │ ├── tf_pool3d.cpp │ ├── tf_pool3d.py │ ├── tf_pool3d_compile.sh │ └── tf_pool3d_gpu.cu ├── sampling │ ├── tf_sample.cpp │ ├── tf_sample.py │ ├── tf_sample_compile.sh │ └── tf_sample_gpu.cu ├── unpooling │ ├── tf_unpool3d.cpp │ ├── tf_unpool3d.py │ ├── tf_unpool3d_compile.sh │ └── tf_unpool3d_gpu.cu └── weight_helper │ ├── tf_dist2weight.cpp │ ├── tf_dist2weight_compile.sh │ ├── tf_dist2weight_gpu.cu │ ├── tf_weight_op_test.py │ └── weight_helper.py └── utils ├── data_util.py ├── sph3gcn_util.py ├── sph3gcn_util_fuzzy.py ├── sph3gcn_util_old.py ├── tf_unpool ├── __pycache__ │ └── tf_unpool3d.cpython-35.pyc ├── tf_unpool3d.py ├── tf_unpool3d_mean_test.py └── tf_unpool3d_weighted_test.py └── tf_unpool3d.py /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EnyaHermite/SegGCN/HEAD/README.md -------------------------------------------------------------------------------- /compile.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EnyaHermite/SegGCN/HEAD/compile.sh -------------------------------------------------------------------------------- /models/SegGCN_s3dis.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EnyaHermite/SegGCN/HEAD/models/SegGCN_s3dis.py -------------------------------------------------------------------------------- /models/SegGCN_scannet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EnyaHermite/SegGCN/HEAD/models/SegGCN_scannet.py -------------------------------------------------------------------------------- /s3dis_seg/eval_s3dis_radius.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EnyaHermite/SegGCN/HEAD/s3dis_seg/eval_s3dis_radius.sh -------------------------------------------------------------------------------- /s3dis_seg/evaluate_s3dis_with_overlap.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EnyaHermite/SegGCN/HEAD/s3dis_seg/evaluate_s3dis_with_overlap.py -------------------------------------------------------------------------------- /s3dis_seg/getLabelColor.m: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EnyaHermite/SegGCN/HEAD/s3dis_seg/getLabelColor.m -------------------------------------------------------------------------------- /s3dis_seg/labelcolor_labelid.mat: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EnyaHermite/SegGCN/HEAD/s3dis_seg/labelcolor_labelid.mat -------------------------------------------------------------------------------- /s3dis_seg/s3dis_block2scene_index.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EnyaHermite/SegGCN/HEAD/s3dis_seg/s3dis_block2scene_index.py -------------------------------------------------------------------------------- /s3dis_seg/s3dis_config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EnyaHermite/SegGCN/HEAD/s3dis_seg/s3dis_config.py -------------------------------------------------------------------------------- /s3dis_seg/s3dis_config_kpconv.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EnyaHermite/SegGCN/HEAD/s3dis_seg/s3dis_config_kpconv.py -------------------------------------------------------------------------------- /s3dis_seg/s3dis_config_resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EnyaHermite/SegGCN/HEAD/s3dis_seg/s3dis_config_resnet.py -------------------------------------------------------------------------------- /s3dis_seg/s3dis_gt_voxelization_sample.m: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EnyaHermite/SegGCN/HEAD/s3dis_seg/s3dis_gt_voxelization_sample.m -------------------------------------------------------------------------------- /s3dis_seg/save_weights_as_numpy.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EnyaHermite/SegGCN/HEAD/s3dis_seg/save_weights_as_numpy.py -------------------------------------------------------------------------------- /s3dis_seg/train_s3dis.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EnyaHermite/SegGCN/HEAD/s3dis_seg/train_s3dis.py -------------------------------------------------------------------------------- /s3dis_seg/train_s3dis_resnet.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EnyaHermite/SegGCN/HEAD/s3dis_seg/train_s3dis_resnet.sh -------------------------------------------------------------------------------- /s3dis_seg/visualize.m: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EnyaHermite/SegGCN/HEAD/s3dis_seg/visualize.m -------------------------------------------------------------------------------- /scannet_seg/evaluate_scannet_withoverlap.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EnyaHermite/SegGCN/HEAD/scannet_seg/evaluate_scannet_withoverlap.py -------------------------------------------------------------------------------- /scannet_seg/getiou.m: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EnyaHermite/SegGCN/HEAD/scannet_seg/getiou.m -------------------------------------------------------------------------------- /scannet_seg/scannet_block2scene_index.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EnyaHermite/SegGCN/HEAD/scannet_seg/scannet_block2scene_index.py -------------------------------------------------------------------------------- /scannet_seg/scannet_config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EnyaHermite/SegGCN/HEAD/scannet_seg/scannet_config.py -------------------------------------------------------------------------------- /scannet_seg/scannet_config_SegGCN.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EnyaHermite/SegGCN/HEAD/scannet_seg/scannet_config_SegGCN.py -------------------------------------------------------------------------------- /scannet_seg/scannet_merge.m: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EnyaHermite/SegGCN/HEAD/scannet_seg/scannet_merge.m -------------------------------------------------------------------------------- /scannet_seg/scannet_merge_benchmarkEval.m: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EnyaHermite/SegGCN/HEAD/scannet_seg/scannet_merge_benchmarkEval.m -------------------------------------------------------------------------------- /scannet_seg/train_scannet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EnyaHermite/SegGCN/HEAD/scannet_seg/train_scannet.py -------------------------------------------------------------------------------- /tf_ops/buildkernel/tf_buildkernel.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EnyaHermite/SegGCN/HEAD/tf_ops/buildkernel/tf_buildkernel.cpp -------------------------------------------------------------------------------- /tf_ops/buildkernel/tf_buildkernel.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EnyaHermite/SegGCN/HEAD/tf_ops/buildkernel/tf_buildkernel.py -------------------------------------------------------------------------------- /tf_ops/buildkernel/tf_buildkernel_compile.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EnyaHermite/SegGCN/HEAD/tf_ops/buildkernel/tf_buildkernel_compile.sh -------------------------------------------------------------------------------- /tf_ops/buildkernel/tf_buildkernel_gpu.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EnyaHermite/SegGCN/HEAD/tf_ops/buildkernel/tf_buildkernel_gpu.cu -------------------------------------------------------------------------------- /tf_ops/convolution/tf_conv3d.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EnyaHermite/SegGCN/HEAD/tf_ops/convolution/tf_conv3d.cpp -------------------------------------------------------------------------------- /tf_ops/convolution/tf_conv3d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EnyaHermite/SegGCN/HEAD/tf_ops/convolution/tf_conv3d.py -------------------------------------------------------------------------------- /tf_ops/convolution/tf_conv3d_compile.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EnyaHermite/SegGCN/HEAD/tf_ops/convolution/tf_conv3d_compile.sh -------------------------------------------------------------------------------- /tf_ops/convolution/tf_conv3d_gpu.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EnyaHermite/SegGCN/HEAD/tf_ops/convolution/tf_conv3d_gpu.cu -------------------------------------------------------------------------------- /tf_ops/nnquery/tf_nnquery.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EnyaHermite/SegGCN/HEAD/tf_ops/nnquery/tf_nnquery.cpp -------------------------------------------------------------------------------- /tf_ops/nnquery/tf_nnquery.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EnyaHermite/SegGCN/HEAD/tf_ops/nnquery/tf_nnquery.py -------------------------------------------------------------------------------- /tf_ops/nnquery/tf_nnquery_compile.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EnyaHermite/SegGCN/HEAD/tf_ops/nnquery/tf_nnquery_compile.sh -------------------------------------------------------------------------------- /tf_ops/nnquery/tf_nnquery_gpu.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EnyaHermite/SegGCN/HEAD/tf_ops/nnquery/tf_nnquery_gpu.cu -------------------------------------------------------------------------------- /tf_ops/pooling/tf_pool3d.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EnyaHermite/SegGCN/HEAD/tf_ops/pooling/tf_pool3d.cpp -------------------------------------------------------------------------------- /tf_ops/pooling/tf_pool3d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EnyaHermite/SegGCN/HEAD/tf_ops/pooling/tf_pool3d.py -------------------------------------------------------------------------------- /tf_ops/pooling/tf_pool3d_compile.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EnyaHermite/SegGCN/HEAD/tf_ops/pooling/tf_pool3d_compile.sh -------------------------------------------------------------------------------- /tf_ops/pooling/tf_pool3d_gpu.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EnyaHermite/SegGCN/HEAD/tf_ops/pooling/tf_pool3d_gpu.cu -------------------------------------------------------------------------------- /tf_ops/sampling/tf_sample.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EnyaHermite/SegGCN/HEAD/tf_ops/sampling/tf_sample.cpp -------------------------------------------------------------------------------- /tf_ops/sampling/tf_sample.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EnyaHermite/SegGCN/HEAD/tf_ops/sampling/tf_sample.py -------------------------------------------------------------------------------- /tf_ops/sampling/tf_sample_compile.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EnyaHermite/SegGCN/HEAD/tf_ops/sampling/tf_sample_compile.sh -------------------------------------------------------------------------------- /tf_ops/sampling/tf_sample_gpu.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EnyaHermite/SegGCN/HEAD/tf_ops/sampling/tf_sample_gpu.cu -------------------------------------------------------------------------------- /tf_ops/unpooling/tf_unpool3d.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EnyaHermite/SegGCN/HEAD/tf_ops/unpooling/tf_unpool3d.cpp -------------------------------------------------------------------------------- /tf_ops/unpooling/tf_unpool3d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EnyaHermite/SegGCN/HEAD/tf_ops/unpooling/tf_unpool3d.py -------------------------------------------------------------------------------- /tf_ops/unpooling/tf_unpool3d_compile.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EnyaHermite/SegGCN/HEAD/tf_ops/unpooling/tf_unpool3d_compile.sh -------------------------------------------------------------------------------- /tf_ops/unpooling/tf_unpool3d_gpu.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EnyaHermite/SegGCN/HEAD/tf_ops/unpooling/tf_unpool3d_gpu.cu -------------------------------------------------------------------------------- /tf_ops/weight_helper/tf_dist2weight.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EnyaHermite/SegGCN/HEAD/tf_ops/weight_helper/tf_dist2weight.cpp -------------------------------------------------------------------------------- /tf_ops/weight_helper/tf_dist2weight_compile.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EnyaHermite/SegGCN/HEAD/tf_ops/weight_helper/tf_dist2weight_compile.sh -------------------------------------------------------------------------------- /tf_ops/weight_helper/tf_dist2weight_gpu.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EnyaHermite/SegGCN/HEAD/tf_ops/weight_helper/tf_dist2weight_gpu.cu -------------------------------------------------------------------------------- /tf_ops/weight_helper/tf_weight_op_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EnyaHermite/SegGCN/HEAD/tf_ops/weight_helper/tf_weight_op_test.py -------------------------------------------------------------------------------- /tf_ops/weight_helper/weight_helper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EnyaHermite/SegGCN/HEAD/tf_ops/weight_helper/weight_helper.py -------------------------------------------------------------------------------- /utils/data_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EnyaHermite/SegGCN/HEAD/utils/data_util.py -------------------------------------------------------------------------------- /utils/sph3gcn_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EnyaHermite/SegGCN/HEAD/utils/sph3gcn_util.py -------------------------------------------------------------------------------- /utils/sph3gcn_util_fuzzy.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EnyaHermite/SegGCN/HEAD/utils/sph3gcn_util_fuzzy.py -------------------------------------------------------------------------------- /utils/sph3gcn_util_old.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EnyaHermite/SegGCN/HEAD/utils/sph3gcn_util_old.py -------------------------------------------------------------------------------- /utils/tf_unpool/__pycache__/tf_unpool3d.cpython-35.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EnyaHermite/SegGCN/HEAD/utils/tf_unpool/__pycache__/tf_unpool3d.cpython-35.pyc -------------------------------------------------------------------------------- /utils/tf_unpool/tf_unpool3d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EnyaHermite/SegGCN/HEAD/utils/tf_unpool/tf_unpool3d.py -------------------------------------------------------------------------------- /utils/tf_unpool/tf_unpool3d_mean_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EnyaHermite/SegGCN/HEAD/utils/tf_unpool/tf_unpool3d_mean_test.py -------------------------------------------------------------------------------- /utils/tf_unpool/tf_unpool3d_weighted_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EnyaHermite/SegGCN/HEAD/utils/tf_unpool/tf_unpool3d_weighted_test.py -------------------------------------------------------------------------------- /utils/tf_unpool3d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EnyaHermite/SegGCN/HEAD/utils/tf_unpool3d.py --------------------------------------------------------------------------------