├── .gitignore ├── README.md ├── og3d_src ├── configs │ ├── nr3d_gtlabel_model.yaml │ ├── nr3d_gtlabelpcd_mix_model.yaml │ ├── nr3d_gtpcd_model.yaml │ ├── pcd_classifier.yaml │ ├── scanrefer_gtlabel_model.yaml │ ├── scanrefer_gtlabelpcd_mix_model.yaml │ ├── sr3d_gtlabel_model.yaml │ └── sr3d_gtlabelpcd_mix_model.yaml ├── data │ ├── common.py │ ├── gtlabel_dataset.py │ ├── gtlabelpcd_dataset.py │ ├── gtpcd_dataset.py │ └── utterances.py ├── model │ ├── backbone │ │ └── point_net_pp.py │ ├── cmt_module.py │ ├── external_tools │ │ ├── __init__.py │ │ └── pointnet2 │ │ │ ├── README.txt │ │ │ ├── _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.egg-info │ │ │ ├── PKG-INFO │ │ │ ├── SOURCES.txt │ │ │ ├── dependency_links.txt │ │ │ └── top_level.txt │ │ │ ├── pointnet2_modules.py │ │ │ ├── pointnet2_test.py │ │ │ ├── pointnet2_utils.py │ │ │ ├── pytorch_utils.py │ │ │ └── setup.py │ ├── mmt_module.py │ ├── obj_encoder.py │ ├── referit3d_net.py │ ├── referit3d_net_mix.py │ └── txt_encoder.py ├── optim │ ├── __init__.py │ ├── adamw.py │ ├── lookahead.py │ ├── misc.py │ ├── radam.py │ ├── ralamb.py │ ├── rangerlars.py │ └── sched.py ├── parser.py ├── train.py ├── train_mix.py ├── train_pcd_backbone.py └── utils │ ├── __init__.py │ ├── distributed.py │ ├── logger.py │ ├── misc.py │ └── save.py ├── preprocess ├── prepare_nr3d_annos.py ├── prepare_scanrefer_annos.py └── scannetv2 │ ├── README.md │ ├── precompute_instance_bboxes.py │ ├── precompute_instance_colors.py │ ├── precompute_pred_instance_colors.py │ ├── precompute_scene_locs.py │ ├── prepare_scannet_data.py │ └── scannetv2_label_glovefts.py └── requirements.txt /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cshizhe/vil3dref/HEAD/.gitignore -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cshizhe/vil3dref/HEAD/README.md -------------------------------------------------------------------------------- /og3d_src/configs/nr3d_gtlabel_model.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cshizhe/vil3dref/HEAD/og3d_src/configs/nr3d_gtlabel_model.yaml -------------------------------------------------------------------------------- /og3d_src/configs/nr3d_gtlabelpcd_mix_model.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cshizhe/vil3dref/HEAD/og3d_src/configs/nr3d_gtlabelpcd_mix_model.yaml -------------------------------------------------------------------------------- /og3d_src/configs/nr3d_gtpcd_model.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cshizhe/vil3dref/HEAD/og3d_src/configs/nr3d_gtpcd_model.yaml -------------------------------------------------------------------------------- /og3d_src/configs/pcd_classifier.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cshizhe/vil3dref/HEAD/og3d_src/configs/pcd_classifier.yaml -------------------------------------------------------------------------------- /og3d_src/configs/scanrefer_gtlabel_model.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cshizhe/vil3dref/HEAD/og3d_src/configs/scanrefer_gtlabel_model.yaml -------------------------------------------------------------------------------- /og3d_src/configs/scanrefer_gtlabelpcd_mix_model.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cshizhe/vil3dref/HEAD/og3d_src/configs/scanrefer_gtlabelpcd_mix_model.yaml -------------------------------------------------------------------------------- /og3d_src/configs/sr3d_gtlabel_model.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cshizhe/vil3dref/HEAD/og3d_src/configs/sr3d_gtlabel_model.yaml -------------------------------------------------------------------------------- /og3d_src/configs/sr3d_gtlabelpcd_mix_model.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cshizhe/vil3dref/HEAD/og3d_src/configs/sr3d_gtlabelpcd_mix_model.yaml -------------------------------------------------------------------------------- /og3d_src/data/common.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cshizhe/vil3dref/HEAD/og3d_src/data/common.py -------------------------------------------------------------------------------- /og3d_src/data/gtlabel_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cshizhe/vil3dref/HEAD/og3d_src/data/gtlabel_dataset.py -------------------------------------------------------------------------------- /og3d_src/data/gtlabelpcd_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cshizhe/vil3dref/HEAD/og3d_src/data/gtlabelpcd_dataset.py -------------------------------------------------------------------------------- /og3d_src/data/gtpcd_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cshizhe/vil3dref/HEAD/og3d_src/data/gtpcd_dataset.py -------------------------------------------------------------------------------- /og3d_src/data/utterances.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cshizhe/vil3dref/HEAD/og3d_src/data/utterances.py -------------------------------------------------------------------------------- /og3d_src/model/backbone/point_net_pp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cshizhe/vil3dref/HEAD/og3d_src/model/backbone/point_net_pp.py -------------------------------------------------------------------------------- /og3d_src/model/cmt_module.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cshizhe/vil3dref/HEAD/og3d_src/model/cmt_module.py -------------------------------------------------------------------------------- /og3d_src/model/external_tools/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /og3d_src/model/external_tools/pointnet2/README.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cshizhe/vil3dref/HEAD/og3d_src/model/external_tools/pointnet2/README.txt -------------------------------------------------------------------------------- /og3d_src/model/external_tools/pointnet2/_ext_src/include/ball_query.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cshizhe/vil3dref/HEAD/og3d_src/model/external_tools/pointnet2/_ext_src/include/ball_query.h -------------------------------------------------------------------------------- /og3d_src/model/external_tools/pointnet2/_ext_src/include/cuda_utils.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cshizhe/vil3dref/HEAD/og3d_src/model/external_tools/pointnet2/_ext_src/include/cuda_utils.h -------------------------------------------------------------------------------- /og3d_src/model/external_tools/pointnet2/_ext_src/include/group_points.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cshizhe/vil3dref/HEAD/og3d_src/model/external_tools/pointnet2/_ext_src/include/group_points.h -------------------------------------------------------------------------------- /og3d_src/model/external_tools/pointnet2/_ext_src/include/interpolate.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cshizhe/vil3dref/HEAD/og3d_src/model/external_tools/pointnet2/_ext_src/include/interpolate.h -------------------------------------------------------------------------------- /og3d_src/model/external_tools/pointnet2/_ext_src/include/sampling.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cshizhe/vil3dref/HEAD/og3d_src/model/external_tools/pointnet2/_ext_src/include/sampling.h -------------------------------------------------------------------------------- /og3d_src/model/external_tools/pointnet2/_ext_src/include/utils.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cshizhe/vil3dref/HEAD/og3d_src/model/external_tools/pointnet2/_ext_src/include/utils.h -------------------------------------------------------------------------------- /og3d_src/model/external_tools/pointnet2/_ext_src/src/ball_query.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cshizhe/vil3dref/HEAD/og3d_src/model/external_tools/pointnet2/_ext_src/src/ball_query.cpp -------------------------------------------------------------------------------- /og3d_src/model/external_tools/pointnet2/_ext_src/src/ball_query_gpu.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cshizhe/vil3dref/HEAD/og3d_src/model/external_tools/pointnet2/_ext_src/src/ball_query_gpu.cu -------------------------------------------------------------------------------- /og3d_src/model/external_tools/pointnet2/_ext_src/src/bindings.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cshizhe/vil3dref/HEAD/og3d_src/model/external_tools/pointnet2/_ext_src/src/bindings.cpp -------------------------------------------------------------------------------- /og3d_src/model/external_tools/pointnet2/_ext_src/src/group_points.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cshizhe/vil3dref/HEAD/og3d_src/model/external_tools/pointnet2/_ext_src/src/group_points.cpp -------------------------------------------------------------------------------- /og3d_src/model/external_tools/pointnet2/_ext_src/src/group_points_gpu.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cshizhe/vil3dref/HEAD/og3d_src/model/external_tools/pointnet2/_ext_src/src/group_points_gpu.cu -------------------------------------------------------------------------------- /og3d_src/model/external_tools/pointnet2/_ext_src/src/interpolate.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cshizhe/vil3dref/HEAD/og3d_src/model/external_tools/pointnet2/_ext_src/src/interpolate.cpp -------------------------------------------------------------------------------- /og3d_src/model/external_tools/pointnet2/_ext_src/src/interpolate_gpu.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cshizhe/vil3dref/HEAD/og3d_src/model/external_tools/pointnet2/_ext_src/src/interpolate_gpu.cu -------------------------------------------------------------------------------- /og3d_src/model/external_tools/pointnet2/_ext_src/src/sampling.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cshizhe/vil3dref/HEAD/og3d_src/model/external_tools/pointnet2/_ext_src/src/sampling.cpp -------------------------------------------------------------------------------- /og3d_src/model/external_tools/pointnet2/_ext_src/src/sampling_gpu.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cshizhe/vil3dref/HEAD/og3d_src/model/external_tools/pointnet2/_ext_src/src/sampling_gpu.cu -------------------------------------------------------------------------------- /og3d_src/model/external_tools/pointnet2/pointnet2.egg-info/PKG-INFO: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cshizhe/vil3dref/HEAD/og3d_src/model/external_tools/pointnet2/pointnet2.egg-info/PKG-INFO -------------------------------------------------------------------------------- /og3d_src/model/external_tools/pointnet2/pointnet2.egg-info/SOURCES.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cshizhe/vil3dref/HEAD/og3d_src/model/external_tools/pointnet2/pointnet2.egg-info/SOURCES.txt -------------------------------------------------------------------------------- /og3d_src/model/external_tools/pointnet2/pointnet2.egg-info/dependency_links.txt: -------------------------------------------------------------------------------- 1 | 2 | -------------------------------------------------------------------------------- /og3d_src/model/external_tools/pointnet2/pointnet2.egg-info/top_level.txt: -------------------------------------------------------------------------------- 1 | pointnet2 2 | -------------------------------------------------------------------------------- /og3d_src/model/external_tools/pointnet2/pointnet2_modules.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cshizhe/vil3dref/HEAD/og3d_src/model/external_tools/pointnet2/pointnet2_modules.py -------------------------------------------------------------------------------- /og3d_src/model/external_tools/pointnet2/pointnet2_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cshizhe/vil3dref/HEAD/og3d_src/model/external_tools/pointnet2/pointnet2_test.py -------------------------------------------------------------------------------- /og3d_src/model/external_tools/pointnet2/pointnet2_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cshizhe/vil3dref/HEAD/og3d_src/model/external_tools/pointnet2/pointnet2_utils.py -------------------------------------------------------------------------------- /og3d_src/model/external_tools/pointnet2/pytorch_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cshizhe/vil3dref/HEAD/og3d_src/model/external_tools/pointnet2/pytorch_utils.py -------------------------------------------------------------------------------- /og3d_src/model/external_tools/pointnet2/setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cshizhe/vil3dref/HEAD/og3d_src/model/external_tools/pointnet2/setup.py -------------------------------------------------------------------------------- /og3d_src/model/mmt_module.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cshizhe/vil3dref/HEAD/og3d_src/model/mmt_module.py -------------------------------------------------------------------------------- /og3d_src/model/obj_encoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cshizhe/vil3dref/HEAD/og3d_src/model/obj_encoder.py -------------------------------------------------------------------------------- /og3d_src/model/referit3d_net.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cshizhe/vil3dref/HEAD/og3d_src/model/referit3d_net.py -------------------------------------------------------------------------------- /og3d_src/model/referit3d_net_mix.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cshizhe/vil3dref/HEAD/og3d_src/model/referit3d_net_mix.py -------------------------------------------------------------------------------- /og3d_src/model/txt_encoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cshizhe/vil3dref/HEAD/og3d_src/model/txt_encoder.py -------------------------------------------------------------------------------- /og3d_src/optim/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cshizhe/vil3dref/HEAD/og3d_src/optim/__init__.py -------------------------------------------------------------------------------- /og3d_src/optim/adamw.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cshizhe/vil3dref/HEAD/og3d_src/optim/adamw.py -------------------------------------------------------------------------------- /og3d_src/optim/lookahead.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cshizhe/vil3dref/HEAD/og3d_src/optim/lookahead.py -------------------------------------------------------------------------------- /og3d_src/optim/misc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cshizhe/vil3dref/HEAD/og3d_src/optim/misc.py -------------------------------------------------------------------------------- /og3d_src/optim/radam.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cshizhe/vil3dref/HEAD/og3d_src/optim/radam.py -------------------------------------------------------------------------------- /og3d_src/optim/ralamb.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cshizhe/vil3dref/HEAD/og3d_src/optim/ralamb.py -------------------------------------------------------------------------------- /og3d_src/optim/rangerlars.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cshizhe/vil3dref/HEAD/og3d_src/optim/rangerlars.py -------------------------------------------------------------------------------- /og3d_src/optim/sched.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cshizhe/vil3dref/HEAD/og3d_src/optim/sched.py -------------------------------------------------------------------------------- /og3d_src/parser.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cshizhe/vil3dref/HEAD/og3d_src/parser.py -------------------------------------------------------------------------------- /og3d_src/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cshizhe/vil3dref/HEAD/og3d_src/train.py -------------------------------------------------------------------------------- /og3d_src/train_mix.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cshizhe/vil3dref/HEAD/og3d_src/train_mix.py -------------------------------------------------------------------------------- /og3d_src/train_pcd_backbone.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cshizhe/vil3dref/HEAD/og3d_src/train_pcd_backbone.py -------------------------------------------------------------------------------- /og3d_src/utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /og3d_src/utils/distributed.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cshizhe/vil3dref/HEAD/og3d_src/utils/distributed.py -------------------------------------------------------------------------------- /og3d_src/utils/logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cshizhe/vil3dref/HEAD/og3d_src/utils/logger.py -------------------------------------------------------------------------------- /og3d_src/utils/misc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cshizhe/vil3dref/HEAD/og3d_src/utils/misc.py -------------------------------------------------------------------------------- /og3d_src/utils/save.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cshizhe/vil3dref/HEAD/og3d_src/utils/save.py -------------------------------------------------------------------------------- /preprocess/prepare_nr3d_annos.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cshizhe/vil3dref/HEAD/preprocess/prepare_nr3d_annos.py -------------------------------------------------------------------------------- /preprocess/prepare_scanrefer_annos.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cshizhe/vil3dref/HEAD/preprocess/prepare_scanrefer_annos.py -------------------------------------------------------------------------------- /preprocess/scannetv2/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cshizhe/vil3dref/HEAD/preprocess/scannetv2/README.md -------------------------------------------------------------------------------- /preprocess/scannetv2/precompute_instance_bboxes.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cshizhe/vil3dref/HEAD/preprocess/scannetv2/precompute_instance_bboxes.py -------------------------------------------------------------------------------- /preprocess/scannetv2/precompute_instance_colors.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cshizhe/vil3dref/HEAD/preprocess/scannetv2/precompute_instance_colors.py -------------------------------------------------------------------------------- /preprocess/scannetv2/precompute_pred_instance_colors.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cshizhe/vil3dref/HEAD/preprocess/scannetv2/precompute_pred_instance_colors.py -------------------------------------------------------------------------------- /preprocess/scannetv2/precompute_scene_locs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cshizhe/vil3dref/HEAD/preprocess/scannetv2/precompute_scene_locs.py -------------------------------------------------------------------------------- /preprocess/scannetv2/prepare_scannet_data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cshizhe/vil3dref/HEAD/preprocess/scannetv2/prepare_scannet_data.py -------------------------------------------------------------------------------- /preprocess/scannetv2/scannetv2_label_glovefts.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cshizhe/vil3dref/HEAD/preprocess/scannetv2/scannetv2_label_glovefts.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cshizhe/vil3dref/HEAD/requirements.txt --------------------------------------------------------------------------------