├── .gitignore ├── LICENSE ├── README.md ├── docs ├── pipeline.png └── splash_fig.png ├── install.sh ├── installation_requirements ├── README.md └── requirements-zeronvs.txt ├── pipeline_conf.yaml ├── pyproject.toml └── rovi_aug ├── __init__.py ├── data_processing ├── __init__.py ├── augment_dataset.py └── multithreaded_adhoc_tfds_builder.py ├── mods ├── __init__.py ├── base_mod.py ├── merge_mod.py ├── robot_mask_mod.py ├── robot_to_robot_mod.py ├── video_inpaint_mod.py └── view_augmentation_mod.py └── view_augmentation ├── __init__.py ├── sampler ├── __init__.py ├── trajectory_consistent_sampler.py ├── uniform_view_sampler.py └── view_sampler.py └── view_augmentation.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BerkeleyAutomation/rovi-aug/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BerkeleyAutomation/rovi-aug/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BerkeleyAutomation/rovi-aug/HEAD/README.md -------------------------------------------------------------------------------- /docs/pipeline.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BerkeleyAutomation/rovi-aug/HEAD/docs/pipeline.png -------------------------------------------------------------------------------- /docs/splash_fig.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BerkeleyAutomation/rovi-aug/HEAD/docs/splash_fig.png -------------------------------------------------------------------------------- /install.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BerkeleyAutomation/rovi-aug/HEAD/install.sh -------------------------------------------------------------------------------- /installation_requirements/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BerkeleyAutomation/rovi-aug/HEAD/installation_requirements/README.md -------------------------------------------------------------------------------- /installation_requirements/requirements-zeronvs.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BerkeleyAutomation/rovi-aug/HEAD/installation_requirements/requirements-zeronvs.txt -------------------------------------------------------------------------------- /pipeline_conf.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BerkeleyAutomation/rovi-aug/HEAD/pipeline_conf.yaml -------------------------------------------------------------------------------- /pyproject.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BerkeleyAutomation/rovi-aug/HEAD/pyproject.toml -------------------------------------------------------------------------------- /rovi_aug/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /rovi_aug/data_processing/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /rovi_aug/data_processing/augment_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BerkeleyAutomation/rovi-aug/HEAD/rovi_aug/data_processing/augment_dataset.py -------------------------------------------------------------------------------- /rovi_aug/data_processing/multithreaded_adhoc_tfds_builder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BerkeleyAutomation/rovi-aug/HEAD/rovi_aug/data_processing/multithreaded_adhoc_tfds_builder.py -------------------------------------------------------------------------------- /rovi_aug/mods/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BerkeleyAutomation/rovi-aug/HEAD/rovi_aug/mods/__init__.py -------------------------------------------------------------------------------- /rovi_aug/mods/base_mod.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BerkeleyAutomation/rovi-aug/HEAD/rovi_aug/mods/base_mod.py -------------------------------------------------------------------------------- /rovi_aug/mods/merge_mod.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BerkeleyAutomation/rovi-aug/HEAD/rovi_aug/mods/merge_mod.py -------------------------------------------------------------------------------- /rovi_aug/mods/robot_mask_mod.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BerkeleyAutomation/rovi-aug/HEAD/rovi_aug/mods/robot_mask_mod.py -------------------------------------------------------------------------------- /rovi_aug/mods/robot_to_robot_mod.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BerkeleyAutomation/rovi-aug/HEAD/rovi_aug/mods/robot_to_robot_mod.py -------------------------------------------------------------------------------- /rovi_aug/mods/video_inpaint_mod.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BerkeleyAutomation/rovi-aug/HEAD/rovi_aug/mods/video_inpaint_mod.py -------------------------------------------------------------------------------- /rovi_aug/mods/view_augmentation_mod.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BerkeleyAutomation/rovi-aug/HEAD/rovi_aug/mods/view_augmentation_mod.py -------------------------------------------------------------------------------- /rovi_aug/view_augmentation/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /rovi_aug/view_augmentation/sampler/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /rovi_aug/view_augmentation/sampler/trajectory_consistent_sampler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BerkeleyAutomation/rovi-aug/HEAD/rovi_aug/view_augmentation/sampler/trajectory_consistent_sampler.py -------------------------------------------------------------------------------- /rovi_aug/view_augmentation/sampler/uniform_view_sampler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BerkeleyAutomation/rovi-aug/HEAD/rovi_aug/view_augmentation/sampler/uniform_view_sampler.py -------------------------------------------------------------------------------- /rovi_aug/view_augmentation/sampler/view_sampler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BerkeleyAutomation/rovi-aug/HEAD/rovi_aug/view_augmentation/sampler/view_sampler.py -------------------------------------------------------------------------------- /rovi_aug/view_augmentation/view_augmentation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BerkeleyAutomation/rovi-aug/HEAD/rovi_aug/view_augmentation/view_augmentation.py --------------------------------------------------------------------------------