├── .gitignore ├── Dockerfile ├── LICENSE ├── README.md ├── configs ├── idm.yaml ├── laom-labels.yaml ├── laom.yaml └── lapo.yaml ├── data-links.txt ├── data └── example-data.hdf5 ├── images ├── envs-vis.png ├── final_result_comb.jpg ├── lapo-pipeline.jpg └── logo.png ├── pyproject.toml ├── requirements.txt ├── scripts ├── data_collection │ ├── checkpoints │ │ ├── cheetah-run-expert │ │ │ ├── checkpoint.pt │ │ │ └── config.pt │ │ ├── hopper-hop-expert │ │ │ ├── checkpoint.pt │ │ │ └── config.pt │ │ ├── sac-humanoid-walk.zip │ │ └── walker-run-expert │ │ │ ├── checkpoint.pt │ │ │ └── config.pt │ ├── cleanrl_ppo.py │ ├── collect_data.py │ ├── collect_data_sb3.py │ └── sb3_sac.py ├── reproducing_figures.ipynb └── sample_labeled_data.py ├── src ├── augmentations.py ├── dcs │ ├── background.py │ ├── camera.py │ ├── color.py │ ├── suite.py │ └── suite_utils.py ├── nn.py ├── scheduler.py └── utils.py ├── train_idm.py ├── train_laom.py ├── train_laom_labels.py └── train_lapo.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dunnolab/laom/HEAD/.gitignore -------------------------------------------------------------------------------- /Dockerfile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dunnolab/laom/HEAD/Dockerfile -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dunnolab/laom/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dunnolab/laom/HEAD/README.md -------------------------------------------------------------------------------- /configs/idm.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dunnolab/laom/HEAD/configs/idm.yaml -------------------------------------------------------------------------------- /configs/laom-labels.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dunnolab/laom/HEAD/configs/laom-labels.yaml -------------------------------------------------------------------------------- /configs/laom.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dunnolab/laom/HEAD/configs/laom.yaml -------------------------------------------------------------------------------- /configs/lapo.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dunnolab/laom/HEAD/configs/lapo.yaml -------------------------------------------------------------------------------- /data-links.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dunnolab/laom/HEAD/data-links.txt -------------------------------------------------------------------------------- /data/example-data.hdf5: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dunnolab/laom/HEAD/data/example-data.hdf5 -------------------------------------------------------------------------------- /images/envs-vis.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dunnolab/laom/HEAD/images/envs-vis.png -------------------------------------------------------------------------------- /images/final_result_comb.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dunnolab/laom/HEAD/images/final_result_comb.jpg -------------------------------------------------------------------------------- /images/lapo-pipeline.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dunnolab/laom/HEAD/images/lapo-pipeline.jpg -------------------------------------------------------------------------------- /images/logo.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dunnolab/laom/HEAD/images/logo.png -------------------------------------------------------------------------------- /pyproject.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dunnolab/laom/HEAD/pyproject.toml -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dunnolab/laom/HEAD/requirements.txt -------------------------------------------------------------------------------- /scripts/data_collection/checkpoints/cheetah-run-expert/checkpoint.pt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dunnolab/laom/HEAD/scripts/data_collection/checkpoints/cheetah-run-expert/checkpoint.pt -------------------------------------------------------------------------------- /scripts/data_collection/checkpoints/cheetah-run-expert/config.pt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dunnolab/laom/HEAD/scripts/data_collection/checkpoints/cheetah-run-expert/config.pt -------------------------------------------------------------------------------- /scripts/data_collection/checkpoints/hopper-hop-expert/checkpoint.pt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dunnolab/laom/HEAD/scripts/data_collection/checkpoints/hopper-hop-expert/checkpoint.pt -------------------------------------------------------------------------------- /scripts/data_collection/checkpoints/hopper-hop-expert/config.pt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dunnolab/laom/HEAD/scripts/data_collection/checkpoints/hopper-hop-expert/config.pt -------------------------------------------------------------------------------- /scripts/data_collection/checkpoints/sac-humanoid-walk.zip: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dunnolab/laom/HEAD/scripts/data_collection/checkpoints/sac-humanoid-walk.zip -------------------------------------------------------------------------------- /scripts/data_collection/checkpoints/walker-run-expert/checkpoint.pt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dunnolab/laom/HEAD/scripts/data_collection/checkpoints/walker-run-expert/checkpoint.pt -------------------------------------------------------------------------------- /scripts/data_collection/checkpoints/walker-run-expert/config.pt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dunnolab/laom/HEAD/scripts/data_collection/checkpoints/walker-run-expert/config.pt -------------------------------------------------------------------------------- /scripts/data_collection/cleanrl_ppo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dunnolab/laom/HEAD/scripts/data_collection/cleanrl_ppo.py -------------------------------------------------------------------------------- /scripts/data_collection/collect_data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dunnolab/laom/HEAD/scripts/data_collection/collect_data.py -------------------------------------------------------------------------------- /scripts/data_collection/collect_data_sb3.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dunnolab/laom/HEAD/scripts/data_collection/collect_data_sb3.py -------------------------------------------------------------------------------- /scripts/data_collection/sb3_sac.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dunnolab/laom/HEAD/scripts/data_collection/sb3_sac.py -------------------------------------------------------------------------------- /scripts/reproducing_figures.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dunnolab/laom/HEAD/scripts/reproducing_figures.ipynb -------------------------------------------------------------------------------- /scripts/sample_labeled_data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dunnolab/laom/HEAD/scripts/sample_labeled_data.py -------------------------------------------------------------------------------- /src/augmentations.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dunnolab/laom/HEAD/src/augmentations.py -------------------------------------------------------------------------------- /src/dcs/background.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dunnolab/laom/HEAD/src/dcs/background.py -------------------------------------------------------------------------------- /src/dcs/camera.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dunnolab/laom/HEAD/src/dcs/camera.py -------------------------------------------------------------------------------- /src/dcs/color.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dunnolab/laom/HEAD/src/dcs/color.py -------------------------------------------------------------------------------- /src/dcs/suite.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dunnolab/laom/HEAD/src/dcs/suite.py -------------------------------------------------------------------------------- /src/dcs/suite_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dunnolab/laom/HEAD/src/dcs/suite_utils.py -------------------------------------------------------------------------------- /src/nn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dunnolab/laom/HEAD/src/nn.py -------------------------------------------------------------------------------- /src/scheduler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dunnolab/laom/HEAD/src/scheduler.py -------------------------------------------------------------------------------- /src/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dunnolab/laom/HEAD/src/utils.py -------------------------------------------------------------------------------- /train_idm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dunnolab/laom/HEAD/train_idm.py -------------------------------------------------------------------------------- /train_laom.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dunnolab/laom/HEAD/train_laom.py -------------------------------------------------------------------------------- /train_laom_labels.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dunnolab/laom/HEAD/train_laom_labels.py -------------------------------------------------------------------------------- /train_lapo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dunnolab/laom/HEAD/train_lapo.py --------------------------------------------------------------------------------