├── .gitignore ├── LICENSE.BSD ├── README.md ├── data ├── LICENSE └── Stanford3DSG.pkl ├── experiment ├── citation_increasing_data.py ├── log │ └── .gitignore ├── preprocess_citation_dataset_with_subsampling.py ├── scene_graph_increasing_data.py └── scene_graph_increasing_iters.py ├── log └── .gitignore ├── neural_tree ├── .gitignore ├── __init__.py ├── dataset_loader │ ├── __init__.py │ ├── load_Stanford3d_object_room_dataset.py │ └── utils.py ├── h_tree │ ├── __init__.py │ ├── generate_junction_tree_hierarchies.py │ ├── h_tree_utils.py │ └── subsampling.py ├── models │ ├── __init__.py │ ├── neural_tree_message_passing.py │ ├── original_graph_message_passing.py │ └── utils.py └── utils │ └── base_training_job.py ├── scene_graph_experiment.py └── setup.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MIT-SPARK/neural_tree/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE.BSD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MIT-SPARK/neural_tree/HEAD/LICENSE.BSD -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MIT-SPARK/neural_tree/HEAD/README.md -------------------------------------------------------------------------------- /data/LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MIT-SPARK/neural_tree/HEAD/data/LICENSE -------------------------------------------------------------------------------- /data/Stanford3DSG.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MIT-SPARK/neural_tree/HEAD/data/Stanford3DSG.pkl -------------------------------------------------------------------------------- /experiment/citation_increasing_data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MIT-SPARK/neural_tree/HEAD/experiment/citation_increasing_data.py -------------------------------------------------------------------------------- /experiment/log/.gitignore: -------------------------------------------------------------------------------- 1 | * 2 | */ 3 | !.gitignore -------------------------------------------------------------------------------- /experiment/preprocess_citation_dataset_with_subsampling.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MIT-SPARK/neural_tree/HEAD/experiment/preprocess_citation_dataset_with_subsampling.py -------------------------------------------------------------------------------- /experiment/scene_graph_increasing_data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MIT-SPARK/neural_tree/HEAD/experiment/scene_graph_increasing_data.py -------------------------------------------------------------------------------- /experiment/scene_graph_increasing_iters.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MIT-SPARK/neural_tree/HEAD/experiment/scene_graph_increasing_iters.py -------------------------------------------------------------------------------- /log/.gitignore: -------------------------------------------------------------------------------- 1 | * 2 | */ 3 | !.gitignore -------------------------------------------------------------------------------- /neural_tree/.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MIT-SPARK/neural_tree/HEAD/neural_tree/.gitignore -------------------------------------------------------------------------------- /neural_tree/__init__.py: -------------------------------------------------------------------------------- 1 | from .utils import * -------------------------------------------------------------------------------- /neural_tree/dataset_loader/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MIT-SPARK/neural_tree/HEAD/neural_tree/dataset_loader/__init__.py -------------------------------------------------------------------------------- /neural_tree/dataset_loader/load_Stanford3d_object_room_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MIT-SPARK/neural_tree/HEAD/neural_tree/dataset_loader/load_Stanford3d_object_room_dataset.py -------------------------------------------------------------------------------- /neural_tree/dataset_loader/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MIT-SPARK/neural_tree/HEAD/neural_tree/dataset_loader/utils.py -------------------------------------------------------------------------------- /neural_tree/h_tree/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MIT-SPARK/neural_tree/HEAD/neural_tree/h_tree/__init__.py -------------------------------------------------------------------------------- /neural_tree/h_tree/generate_junction_tree_hierarchies.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MIT-SPARK/neural_tree/HEAD/neural_tree/h_tree/generate_junction_tree_hierarchies.py -------------------------------------------------------------------------------- /neural_tree/h_tree/h_tree_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MIT-SPARK/neural_tree/HEAD/neural_tree/h_tree/h_tree_utils.py -------------------------------------------------------------------------------- /neural_tree/h_tree/subsampling.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MIT-SPARK/neural_tree/HEAD/neural_tree/h_tree/subsampling.py -------------------------------------------------------------------------------- /neural_tree/models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MIT-SPARK/neural_tree/HEAD/neural_tree/models/__init__.py -------------------------------------------------------------------------------- /neural_tree/models/neural_tree_message_passing.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MIT-SPARK/neural_tree/HEAD/neural_tree/models/neural_tree_message_passing.py -------------------------------------------------------------------------------- /neural_tree/models/original_graph_message_passing.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MIT-SPARK/neural_tree/HEAD/neural_tree/models/original_graph_message_passing.py -------------------------------------------------------------------------------- /neural_tree/models/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MIT-SPARK/neural_tree/HEAD/neural_tree/models/utils.py -------------------------------------------------------------------------------- /neural_tree/utils/base_training_job.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MIT-SPARK/neural_tree/HEAD/neural_tree/utils/base_training_job.py -------------------------------------------------------------------------------- /scene_graph_experiment.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MIT-SPARK/neural_tree/HEAD/scene_graph_experiment.py -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MIT-SPARK/neural_tree/HEAD/setup.py --------------------------------------------------------------------------------