├── .gitignore ├── LICENSE ├── README.md ├── cfgs ├── config.yaml └── task │ ├── buggy.yaml │ └── medium.yaml ├── conda_env.yml ├── custom_gym ├── async_vector_env.py ├── buggy_ctrl.py ├── buggy_env.py ├── custom_torch_layer.py ├── multi_unity.py ├── subproc_env.py └── vector_env.py ├── dmc.py ├── dqn.py ├── drqv2.py ├── logger.py ├── method.png ├── models ├── __init__.py ├── base_model.py ├── blocks.py ├── networks.py ├── pix2seg_model.py ├── pix2segdepth_model.py ├── torchutils.py └── utils.py ├── options ├── __init__.py └── base_options_simple.py ├── overview.png ├── replay_buffer.py ├── sample_images ├── real_img72.png ├── real_img779.png ├── sim_img.png └── sim_seg.png ├── sim2seg_segmentation.ipynb ├── train.py ├── utils.py └── video.py /.gitignore: -------------------------------------------------------------------------------- 1 | __pycache__/ 2 | .ipynb_checkpoints/ 3 | exp_local 4 | wandb 5 | -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rll-research/sim2seg/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rll-research/sim2seg/HEAD/README.md -------------------------------------------------------------------------------- /cfgs/config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rll-research/sim2seg/HEAD/cfgs/config.yaml -------------------------------------------------------------------------------- /cfgs/task/buggy.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rll-research/sim2seg/HEAD/cfgs/task/buggy.yaml -------------------------------------------------------------------------------- /cfgs/task/medium.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rll-research/sim2seg/HEAD/cfgs/task/medium.yaml -------------------------------------------------------------------------------- /conda_env.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rll-research/sim2seg/HEAD/conda_env.yml -------------------------------------------------------------------------------- /custom_gym/async_vector_env.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rll-research/sim2seg/HEAD/custom_gym/async_vector_env.py -------------------------------------------------------------------------------- /custom_gym/buggy_ctrl.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rll-research/sim2seg/HEAD/custom_gym/buggy_ctrl.py -------------------------------------------------------------------------------- /custom_gym/buggy_env.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rll-research/sim2seg/HEAD/custom_gym/buggy_env.py -------------------------------------------------------------------------------- /custom_gym/custom_torch_layer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rll-research/sim2seg/HEAD/custom_gym/custom_torch_layer.py -------------------------------------------------------------------------------- /custom_gym/multi_unity.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rll-research/sim2seg/HEAD/custom_gym/multi_unity.py -------------------------------------------------------------------------------- /custom_gym/subproc_env.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rll-research/sim2seg/HEAD/custom_gym/subproc_env.py -------------------------------------------------------------------------------- /custom_gym/vector_env.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rll-research/sim2seg/HEAD/custom_gym/vector_env.py -------------------------------------------------------------------------------- /dmc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rll-research/sim2seg/HEAD/dmc.py -------------------------------------------------------------------------------- /dqn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rll-research/sim2seg/HEAD/dqn.py -------------------------------------------------------------------------------- /drqv2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rll-research/sim2seg/HEAD/drqv2.py -------------------------------------------------------------------------------- /logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rll-research/sim2seg/HEAD/logger.py -------------------------------------------------------------------------------- /method.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rll-research/sim2seg/HEAD/method.png -------------------------------------------------------------------------------- /models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rll-research/sim2seg/HEAD/models/__init__.py -------------------------------------------------------------------------------- /models/base_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rll-research/sim2seg/HEAD/models/base_model.py -------------------------------------------------------------------------------- /models/blocks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rll-research/sim2seg/HEAD/models/blocks.py -------------------------------------------------------------------------------- /models/networks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rll-research/sim2seg/HEAD/models/networks.py -------------------------------------------------------------------------------- /models/pix2seg_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rll-research/sim2seg/HEAD/models/pix2seg_model.py -------------------------------------------------------------------------------- /models/pix2segdepth_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rll-research/sim2seg/HEAD/models/pix2segdepth_model.py -------------------------------------------------------------------------------- /models/torchutils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rll-research/sim2seg/HEAD/models/torchutils.py -------------------------------------------------------------------------------- /models/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rll-research/sim2seg/HEAD/models/utils.py -------------------------------------------------------------------------------- /options/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /options/base_options_simple.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rll-research/sim2seg/HEAD/options/base_options_simple.py -------------------------------------------------------------------------------- /overview.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rll-research/sim2seg/HEAD/overview.png -------------------------------------------------------------------------------- /replay_buffer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rll-research/sim2seg/HEAD/replay_buffer.py -------------------------------------------------------------------------------- /sample_images/real_img72.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rll-research/sim2seg/HEAD/sample_images/real_img72.png -------------------------------------------------------------------------------- /sample_images/real_img779.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rll-research/sim2seg/HEAD/sample_images/real_img779.png -------------------------------------------------------------------------------- /sample_images/sim_img.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rll-research/sim2seg/HEAD/sample_images/sim_img.png -------------------------------------------------------------------------------- /sample_images/sim_seg.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rll-research/sim2seg/HEAD/sample_images/sim_seg.png -------------------------------------------------------------------------------- /sim2seg_segmentation.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rll-research/sim2seg/HEAD/sim2seg_segmentation.ipynb -------------------------------------------------------------------------------- /train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rll-research/sim2seg/HEAD/train.py -------------------------------------------------------------------------------- /utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rll-research/sim2seg/HEAD/utils.py -------------------------------------------------------------------------------- /video.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rll-research/sim2seg/HEAD/video.py --------------------------------------------------------------------------------