├── .gitignore ├── .gitmodules ├── LICENSE ├── README.md ├── config ├── pointgroup_default_scannet.yaml └── pointgroup_run1_scannet.yaml ├── data └── scannetv2_inst.py ├── dataset └── scannetv2 │ ├── prepare_data_inst.py │ ├── prepare_data_inst_gttxt.py │ └── scannet_util.py ├── doc ├── overview.png └── scannet_benchmark.png ├── lib └── pointgroup_ops │ ├── functions │ └── pointgroup_ops.py │ ├── setup.py │ └── src │ ├── bfs_cluster │ ├── bfs_cluster.cpp │ ├── bfs_cluster.cu │ └── bfs_cluster.h │ ├── cuda.cu │ ├── cuda_utils.h │ ├── datatype │ ├── datatype.cpp │ └── datatype.h │ ├── get_iou │ ├── get_iou.cpp │ ├── get_iou.cu │ └── get_iou.h │ ├── pointgroup_ops.cpp │ ├── pointgroup_ops.h │ ├── pointgroup_ops_api.cpp │ ├── roipool │ ├── roipool.cpp │ ├── roipool.cu │ └── roipool.h │ ├── sec_mean │ ├── sec_mean.cpp │ ├── sec_mean.cu │ └── sec_mean.h │ └── voxelize │ ├── voxelize.cpp │ ├── voxelize.cu │ └── voxelize.h ├── model └── pointgroup │ └── pointgroup.py ├── requirements.txt ├── test.py ├── train.py └── util ├── config.py ├── eval.py ├── log.py ├── utils.py ├── utils_3d.py └── visualize.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dvlab-research/PointGroup/HEAD/.gitignore -------------------------------------------------------------------------------- /.gitmodules: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dvlab-research/PointGroup/HEAD/.gitmodules -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dvlab-research/PointGroup/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dvlab-research/PointGroup/HEAD/README.md -------------------------------------------------------------------------------- /config/pointgroup_default_scannet.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dvlab-research/PointGroup/HEAD/config/pointgroup_default_scannet.yaml -------------------------------------------------------------------------------- /config/pointgroup_run1_scannet.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dvlab-research/PointGroup/HEAD/config/pointgroup_run1_scannet.yaml -------------------------------------------------------------------------------- /data/scannetv2_inst.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dvlab-research/PointGroup/HEAD/data/scannetv2_inst.py -------------------------------------------------------------------------------- /dataset/scannetv2/prepare_data_inst.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dvlab-research/PointGroup/HEAD/dataset/scannetv2/prepare_data_inst.py -------------------------------------------------------------------------------- /dataset/scannetv2/prepare_data_inst_gttxt.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dvlab-research/PointGroup/HEAD/dataset/scannetv2/prepare_data_inst_gttxt.py -------------------------------------------------------------------------------- /dataset/scannetv2/scannet_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dvlab-research/PointGroup/HEAD/dataset/scannetv2/scannet_util.py -------------------------------------------------------------------------------- /doc/overview.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dvlab-research/PointGroup/HEAD/doc/overview.png -------------------------------------------------------------------------------- /doc/scannet_benchmark.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dvlab-research/PointGroup/HEAD/doc/scannet_benchmark.png -------------------------------------------------------------------------------- /lib/pointgroup_ops/functions/pointgroup_ops.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dvlab-research/PointGroup/HEAD/lib/pointgroup_ops/functions/pointgroup_ops.py -------------------------------------------------------------------------------- /lib/pointgroup_ops/setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dvlab-research/PointGroup/HEAD/lib/pointgroup_ops/setup.py -------------------------------------------------------------------------------- /lib/pointgroup_ops/src/bfs_cluster/bfs_cluster.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dvlab-research/PointGroup/HEAD/lib/pointgroup_ops/src/bfs_cluster/bfs_cluster.cpp -------------------------------------------------------------------------------- /lib/pointgroup_ops/src/bfs_cluster/bfs_cluster.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dvlab-research/PointGroup/HEAD/lib/pointgroup_ops/src/bfs_cluster/bfs_cluster.cu -------------------------------------------------------------------------------- /lib/pointgroup_ops/src/bfs_cluster/bfs_cluster.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dvlab-research/PointGroup/HEAD/lib/pointgroup_ops/src/bfs_cluster/bfs_cluster.h -------------------------------------------------------------------------------- /lib/pointgroup_ops/src/cuda.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dvlab-research/PointGroup/HEAD/lib/pointgroup_ops/src/cuda.cu -------------------------------------------------------------------------------- /lib/pointgroup_ops/src/cuda_utils.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dvlab-research/PointGroup/HEAD/lib/pointgroup_ops/src/cuda_utils.h -------------------------------------------------------------------------------- /lib/pointgroup_ops/src/datatype/datatype.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dvlab-research/PointGroup/HEAD/lib/pointgroup_ops/src/datatype/datatype.cpp -------------------------------------------------------------------------------- /lib/pointgroup_ops/src/datatype/datatype.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dvlab-research/PointGroup/HEAD/lib/pointgroup_ops/src/datatype/datatype.h -------------------------------------------------------------------------------- /lib/pointgroup_ops/src/get_iou/get_iou.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dvlab-research/PointGroup/HEAD/lib/pointgroup_ops/src/get_iou/get_iou.cpp -------------------------------------------------------------------------------- /lib/pointgroup_ops/src/get_iou/get_iou.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dvlab-research/PointGroup/HEAD/lib/pointgroup_ops/src/get_iou/get_iou.cu -------------------------------------------------------------------------------- /lib/pointgroup_ops/src/get_iou/get_iou.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dvlab-research/PointGroup/HEAD/lib/pointgroup_ops/src/get_iou/get_iou.h -------------------------------------------------------------------------------- /lib/pointgroup_ops/src/pointgroup_ops.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dvlab-research/PointGroup/HEAD/lib/pointgroup_ops/src/pointgroup_ops.cpp -------------------------------------------------------------------------------- /lib/pointgroup_ops/src/pointgroup_ops.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dvlab-research/PointGroup/HEAD/lib/pointgroup_ops/src/pointgroup_ops.h -------------------------------------------------------------------------------- /lib/pointgroup_ops/src/pointgroup_ops_api.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dvlab-research/PointGroup/HEAD/lib/pointgroup_ops/src/pointgroup_ops_api.cpp -------------------------------------------------------------------------------- /lib/pointgroup_ops/src/roipool/roipool.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dvlab-research/PointGroup/HEAD/lib/pointgroup_ops/src/roipool/roipool.cpp -------------------------------------------------------------------------------- /lib/pointgroup_ops/src/roipool/roipool.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dvlab-research/PointGroup/HEAD/lib/pointgroup_ops/src/roipool/roipool.cu -------------------------------------------------------------------------------- /lib/pointgroup_ops/src/roipool/roipool.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dvlab-research/PointGroup/HEAD/lib/pointgroup_ops/src/roipool/roipool.h -------------------------------------------------------------------------------- /lib/pointgroup_ops/src/sec_mean/sec_mean.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dvlab-research/PointGroup/HEAD/lib/pointgroup_ops/src/sec_mean/sec_mean.cpp -------------------------------------------------------------------------------- /lib/pointgroup_ops/src/sec_mean/sec_mean.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dvlab-research/PointGroup/HEAD/lib/pointgroup_ops/src/sec_mean/sec_mean.cu -------------------------------------------------------------------------------- /lib/pointgroup_ops/src/sec_mean/sec_mean.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dvlab-research/PointGroup/HEAD/lib/pointgroup_ops/src/sec_mean/sec_mean.h -------------------------------------------------------------------------------- /lib/pointgroup_ops/src/voxelize/voxelize.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dvlab-research/PointGroup/HEAD/lib/pointgroup_ops/src/voxelize/voxelize.cpp -------------------------------------------------------------------------------- /lib/pointgroup_ops/src/voxelize/voxelize.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dvlab-research/PointGroup/HEAD/lib/pointgroup_ops/src/voxelize/voxelize.cu -------------------------------------------------------------------------------- /lib/pointgroup_ops/src/voxelize/voxelize.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dvlab-research/PointGroup/HEAD/lib/pointgroup_ops/src/voxelize/voxelize.h -------------------------------------------------------------------------------- /model/pointgroup/pointgroup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dvlab-research/PointGroup/HEAD/model/pointgroup/pointgroup.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- 1 | torch==1.1 2 | cmake>=3.13.2 3 | plyfile 4 | tensorboardX 5 | pyyaml 6 | scipy 7 | -------------------------------------------------------------------------------- /test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dvlab-research/PointGroup/HEAD/test.py -------------------------------------------------------------------------------- /train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dvlab-research/PointGroup/HEAD/train.py -------------------------------------------------------------------------------- /util/config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dvlab-research/PointGroup/HEAD/util/config.py -------------------------------------------------------------------------------- /util/eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dvlab-research/PointGroup/HEAD/util/eval.py -------------------------------------------------------------------------------- /util/log.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dvlab-research/PointGroup/HEAD/util/log.py -------------------------------------------------------------------------------- /util/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dvlab-research/PointGroup/HEAD/util/utils.py -------------------------------------------------------------------------------- /util/utils_3d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dvlab-research/PointGroup/HEAD/util/utils_3d.py -------------------------------------------------------------------------------- /util/visualize.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dvlab-research/PointGroup/HEAD/util/visualize.py --------------------------------------------------------------------------------