├── .gitignore ├── README.md ├── __init__.py ├── base ├── __init__.py └── base_model.py ├── data_cleaning ├── README.md ├── __init__.py ├── labels_tester.py ├── mask.py ├── script_colmap.py ├── split_video.py └── write_translations.py ├── dataloader ├── PushDataset.py ├── StackDataset.py └── __init__.py ├── download_data.py ├── environment.yml ├── grip_model └── grip_model.pth ├── images └── teaser.gif ├── models ├── __init__.py └── forward_model.py ├── push_test.py ├── scripts ├── __init__.py ├── push_trainer.py └── stack_trainer.py ├── stack_test.py ├── train.py └── utils ├── __init__.py ├── earlystop.py ├── get_result_vids.py ├── jsonreader.py ├── logger.py ├── plot.py ├── rotation_tools.py └── utils.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sarahisyoung/Visual-Imitation-Made-Easy/HEAD/.gitignore -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sarahisyoung/Visual-Imitation-Made-Easy/HEAD/README.md -------------------------------------------------------------------------------- /__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /base/__init__.py: -------------------------------------------------------------------------------- 1 | from .base_model import * 2 | -------------------------------------------------------------------------------- /base/base_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sarahisyoung/Visual-Imitation-Made-Easy/HEAD/base/base_model.py -------------------------------------------------------------------------------- /data_cleaning/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sarahisyoung/Visual-Imitation-Made-Easy/HEAD/data_cleaning/README.md -------------------------------------------------------------------------------- /data_cleaning/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /data_cleaning/labels_tester.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sarahisyoung/Visual-Imitation-Made-Easy/HEAD/data_cleaning/labels_tester.py -------------------------------------------------------------------------------- /data_cleaning/mask.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sarahisyoung/Visual-Imitation-Made-Easy/HEAD/data_cleaning/mask.py -------------------------------------------------------------------------------- /data_cleaning/script_colmap.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sarahisyoung/Visual-Imitation-Made-Easy/HEAD/data_cleaning/script_colmap.py -------------------------------------------------------------------------------- /data_cleaning/split_video.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sarahisyoung/Visual-Imitation-Made-Easy/HEAD/data_cleaning/split_video.py -------------------------------------------------------------------------------- /data_cleaning/write_translations.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sarahisyoung/Visual-Imitation-Made-Easy/HEAD/data_cleaning/write_translations.py -------------------------------------------------------------------------------- /dataloader/PushDataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sarahisyoung/Visual-Imitation-Made-Easy/HEAD/dataloader/PushDataset.py -------------------------------------------------------------------------------- /dataloader/StackDataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sarahisyoung/Visual-Imitation-Made-Easy/HEAD/dataloader/StackDataset.py -------------------------------------------------------------------------------- /dataloader/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /download_data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sarahisyoung/Visual-Imitation-Made-Easy/HEAD/download_data.py -------------------------------------------------------------------------------- /environment.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sarahisyoung/Visual-Imitation-Made-Easy/HEAD/environment.yml -------------------------------------------------------------------------------- /grip_model/grip_model.pth: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sarahisyoung/Visual-Imitation-Made-Easy/HEAD/grip_model/grip_model.pth -------------------------------------------------------------------------------- /images/teaser.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sarahisyoung/Visual-Imitation-Made-Easy/HEAD/images/teaser.gif -------------------------------------------------------------------------------- /models/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /models/forward_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sarahisyoung/Visual-Imitation-Made-Easy/HEAD/models/forward_model.py -------------------------------------------------------------------------------- /push_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sarahisyoung/Visual-Imitation-Made-Easy/HEAD/push_test.py -------------------------------------------------------------------------------- /scripts/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /scripts/push_trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sarahisyoung/Visual-Imitation-Made-Easy/HEAD/scripts/push_trainer.py -------------------------------------------------------------------------------- /scripts/stack_trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sarahisyoung/Visual-Imitation-Made-Easy/HEAD/scripts/stack_trainer.py -------------------------------------------------------------------------------- /stack_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sarahisyoung/Visual-Imitation-Made-Easy/HEAD/stack_test.py -------------------------------------------------------------------------------- /train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sarahisyoung/Visual-Imitation-Made-Easy/HEAD/train.py -------------------------------------------------------------------------------- /utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /utils/earlystop.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sarahisyoung/Visual-Imitation-Made-Easy/HEAD/utils/earlystop.py -------------------------------------------------------------------------------- /utils/get_result_vids.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sarahisyoung/Visual-Imitation-Made-Easy/HEAD/utils/get_result_vids.py -------------------------------------------------------------------------------- /utils/jsonreader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sarahisyoung/Visual-Imitation-Made-Easy/HEAD/utils/jsonreader.py -------------------------------------------------------------------------------- /utils/logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sarahisyoung/Visual-Imitation-Made-Easy/HEAD/utils/logger.py -------------------------------------------------------------------------------- /utils/plot.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sarahisyoung/Visual-Imitation-Made-Easy/HEAD/utils/plot.py -------------------------------------------------------------------------------- /utils/rotation_tools.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sarahisyoung/Visual-Imitation-Made-Easy/HEAD/utils/rotation_tools.py -------------------------------------------------------------------------------- /utils/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sarahisyoung/Visual-Imitation-Made-Easy/HEAD/utils/utils.py --------------------------------------------------------------------------------