├── .dockerignore ├── .gitignore ├── Dockerfile ├── LICENSE ├── README.md ├── baseline_configs ├── __init__.py ├── one_phase │ ├── __init__.py │ ├── one_phase_rgb_base.py │ ├── one_phase_rgb_clipresnet50_dagger.py │ ├── one_phase_rgb_dagger.py │ ├── one_phase_rgb_il_base.py │ ├── one_phase_rgb_ppo.py │ ├── one_phase_rgb_resnet50_dagger.py │ ├── one_phase_rgb_resnet_dagger.py │ ├── one_phase_rgb_resnet_frozen_map_dagger.py │ └── one_phase_rgb_resnet_ppo.py ├── rearrange_base.py ├── two_phase │ ├── __init__.py │ ├── two_phase_rgb_base.py │ ├── two_phase_rgb_ppowalkthrough_ilunshuffle.py │ ├── two_phase_rgb_resnet_frozen_map_ppowalkthrough_ilunshuffle.py │ └── two_phase_rgb_resnet_ppowalkthrough_ilunshuffle.py └── walkthrough │ ├── __init__.py │ ├── walkthrough_rgb_base.py │ ├── walkthrough_rgb_mapping_ppo.py │ ├── walkthrough_rgb_ppo.py │ └── walkthrough_rgb_resnet_ppo.py ├── data ├── 2021 │ ├── combined.pkl.gz │ ├── test.pkl.gz │ ├── train.pkl.gz │ ├── train_unseen.pkl.gz │ └── val.pkl.gz ├── 2022 │ ├── combined.pkl.gz │ ├── test.pkl.gz │ ├── train.pkl.gz │ └── val.pkl.gz └── 2023 │ ├── combined.pkl.gz │ ├── test.pkl.gz │ ├── train.pkl.gz │ └── val.pkl.gz ├── datagen ├── __init__.py ├── create_combined_dataset.py ├── datagen_constants.py ├── datagen_runner.py └── datagen_utils.py ├── environment.yml ├── example.py ├── pretrained_model_ckpts └── .gitignore ├── rearrange ├── __init__.py ├── baseline_models.py ├── constants.py ├── environment.py ├── expert.py ├── losses.py ├── sensors.py ├── tasks.py └── utils.py ├── rearrange_constants.py └── requirements.txt /.dockerignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/ai2thor-rearrangement/HEAD/.dockerignore -------------------------------------------------------------------------------- /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/ai2thor-rearrangement/HEAD/.gitignore -------------------------------------------------------------------------------- /Dockerfile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/ai2thor-rearrangement/HEAD/Dockerfile -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/ai2thor-rearrangement/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/ai2thor-rearrangement/HEAD/README.md -------------------------------------------------------------------------------- /baseline_configs/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /baseline_configs/one_phase/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /baseline_configs/one_phase/one_phase_rgb_base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/ai2thor-rearrangement/HEAD/baseline_configs/one_phase/one_phase_rgb_base.py -------------------------------------------------------------------------------- /baseline_configs/one_phase/one_phase_rgb_clipresnet50_dagger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/ai2thor-rearrangement/HEAD/baseline_configs/one_phase/one_phase_rgb_clipresnet50_dagger.py -------------------------------------------------------------------------------- /baseline_configs/one_phase/one_phase_rgb_dagger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/ai2thor-rearrangement/HEAD/baseline_configs/one_phase/one_phase_rgb_dagger.py -------------------------------------------------------------------------------- /baseline_configs/one_phase/one_phase_rgb_il_base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/ai2thor-rearrangement/HEAD/baseline_configs/one_phase/one_phase_rgb_il_base.py -------------------------------------------------------------------------------- /baseline_configs/one_phase/one_phase_rgb_ppo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/ai2thor-rearrangement/HEAD/baseline_configs/one_phase/one_phase_rgb_ppo.py -------------------------------------------------------------------------------- /baseline_configs/one_phase/one_phase_rgb_resnet50_dagger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/ai2thor-rearrangement/HEAD/baseline_configs/one_phase/one_phase_rgb_resnet50_dagger.py -------------------------------------------------------------------------------- /baseline_configs/one_phase/one_phase_rgb_resnet_dagger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/ai2thor-rearrangement/HEAD/baseline_configs/one_phase/one_phase_rgb_resnet_dagger.py -------------------------------------------------------------------------------- /baseline_configs/one_phase/one_phase_rgb_resnet_frozen_map_dagger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/ai2thor-rearrangement/HEAD/baseline_configs/one_phase/one_phase_rgb_resnet_frozen_map_dagger.py -------------------------------------------------------------------------------- /baseline_configs/one_phase/one_phase_rgb_resnet_ppo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/ai2thor-rearrangement/HEAD/baseline_configs/one_phase/one_phase_rgb_resnet_ppo.py -------------------------------------------------------------------------------- /baseline_configs/rearrange_base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/ai2thor-rearrangement/HEAD/baseline_configs/rearrange_base.py -------------------------------------------------------------------------------- /baseline_configs/two_phase/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /baseline_configs/two_phase/two_phase_rgb_base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/ai2thor-rearrangement/HEAD/baseline_configs/two_phase/two_phase_rgb_base.py -------------------------------------------------------------------------------- /baseline_configs/two_phase/two_phase_rgb_ppowalkthrough_ilunshuffle.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/ai2thor-rearrangement/HEAD/baseline_configs/two_phase/two_phase_rgb_ppowalkthrough_ilunshuffle.py -------------------------------------------------------------------------------- /baseline_configs/two_phase/two_phase_rgb_resnet_frozen_map_ppowalkthrough_ilunshuffle.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/ai2thor-rearrangement/HEAD/baseline_configs/two_phase/two_phase_rgb_resnet_frozen_map_ppowalkthrough_ilunshuffle.py -------------------------------------------------------------------------------- /baseline_configs/two_phase/two_phase_rgb_resnet_ppowalkthrough_ilunshuffle.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/ai2thor-rearrangement/HEAD/baseline_configs/two_phase/two_phase_rgb_resnet_ppowalkthrough_ilunshuffle.py -------------------------------------------------------------------------------- /baseline_configs/walkthrough/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /baseline_configs/walkthrough/walkthrough_rgb_base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/ai2thor-rearrangement/HEAD/baseline_configs/walkthrough/walkthrough_rgb_base.py -------------------------------------------------------------------------------- /baseline_configs/walkthrough/walkthrough_rgb_mapping_ppo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/ai2thor-rearrangement/HEAD/baseline_configs/walkthrough/walkthrough_rgb_mapping_ppo.py -------------------------------------------------------------------------------- /baseline_configs/walkthrough/walkthrough_rgb_ppo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/ai2thor-rearrangement/HEAD/baseline_configs/walkthrough/walkthrough_rgb_ppo.py -------------------------------------------------------------------------------- /baseline_configs/walkthrough/walkthrough_rgb_resnet_ppo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/ai2thor-rearrangement/HEAD/baseline_configs/walkthrough/walkthrough_rgb_resnet_ppo.py -------------------------------------------------------------------------------- /data/2021/combined.pkl.gz: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/ai2thor-rearrangement/HEAD/data/2021/combined.pkl.gz -------------------------------------------------------------------------------- /data/2021/test.pkl.gz: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/ai2thor-rearrangement/HEAD/data/2021/test.pkl.gz -------------------------------------------------------------------------------- /data/2021/train.pkl.gz: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/ai2thor-rearrangement/HEAD/data/2021/train.pkl.gz -------------------------------------------------------------------------------- /data/2021/train_unseen.pkl.gz: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/ai2thor-rearrangement/HEAD/data/2021/train_unseen.pkl.gz -------------------------------------------------------------------------------- /data/2021/val.pkl.gz: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/ai2thor-rearrangement/HEAD/data/2021/val.pkl.gz -------------------------------------------------------------------------------- /data/2022/combined.pkl.gz: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/ai2thor-rearrangement/HEAD/data/2022/combined.pkl.gz -------------------------------------------------------------------------------- /data/2022/test.pkl.gz: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/ai2thor-rearrangement/HEAD/data/2022/test.pkl.gz -------------------------------------------------------------------------------- /data/2022/train.pkl.gz: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/ai2thor-rearrangement/HEAD/data/2022/train.pkl.gz -------------------------------------------------------------------------------- /data/2022/val.pkl.gz: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/ai2thor-rearrangement/HEAD/data/2022/val.pkl.gz -------------------------------------------------------------------------------- /data/2023/combined.pkl.gz: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/ai2thor-rearrangement/HEAD/data/2023/combined.pkl.gz -------------------------------------------------------------------------------- /data/2023/test.pkl.gz: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/ai2thor-rearrangement/HEAD/data/2023/test.pkl.gz -------------------------------------------------------------------------------- /data/2023/train.pkl.gz: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/ai2thor-rearrangement/HEAD/data/2023/train.pkl.gz -------------------------------------------------------------------------------- /data/2023/val.pkl.gz: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/ai2thor-rearrangement/HEAD/data/2023/val.pkl.gz -------------------------------------------------------------------------------- /datagen/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /datagen/create_combined_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/ai2thor-rearrangement/HEAD/datagen/create_combined_dataset.py -------------------------------------------------------------------------------- /datagen/datagen_constants.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/ai2thor-rearrangement/HEAD/datagen/datagen_constants.py -------------------------------------------------------------------------------- /datagen/datagen_runner.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/ai2thor-rearrangement/HEAD/datagen/datagen_runner.py -------------------------------------------------------------------------------- /datagen/datagen_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/ai2thor-rearrangement/HEAD/datagen/datagen_utils.py -------------------------------------------------------------------------------- /environment.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/ai2thor-rearrangement/HEAD/environment.yml -------------------------------------------------------------------------------- /example.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/ai2thor-rearrangement/HEAD/example.py -------------------------------------------------------------------------------- /pretrained_model_ckpts/.gitignore: -------------------------------------------------------------------------------- 1 | * 2 | !.gitignore -------------------------------------------------------------------------------- /rearrange/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /rearrange/baseline_models.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/ai2thor-rearrangement/HEAD/rearrange/baseline_models.py -------------------------------------------------------------------------------- /rearrange/constants.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/ai2thor-rearrangement/HEAD/rearrange/constants.py -------------------------------------------------------------------------------- /rearrange/environment.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/ai2thor-rearrangement/HEAD/rearrange/environment.py -------------------------------------------------------------------------------- /rearrange/expert.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/ai2thor-rearrangement/HEAD/rearrange/expert.py -------------------------------------------------------------------------------- /rearrange/losses.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/ai2thor-rearrangement/HEAD/rearrange/losses.py -------------------------------------------------------------------------------- /rearrange/sensors.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/ai2thor-rearrangement/HEAD/rearrange/sensors.py -------------------------------------------------------------------------------- /rearrange/tasks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/ai2thor-rearrangement/HEAD/rearrange/tasks.py -------------------------------------------------------------------------------- /rearrange/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/ai2thor-rearrangement/HEAD/rearrange/utils.py -------------------------------------------------------------------------------- /rearrange_constants.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/ai2thor-rearrangement/HEAD/rearrange_constants.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/ai2thor-rearrangement/HEAD/requirements.txt --------------------------------------------------------------------------------