├── README.md ├── controller.py ├── envs ├── __init__.py ├── __pycache__ │ ├── __init__.cpython-36.pyc │ ├── fetch_pick_and_place.cpython-36.pyc │ ├── fetch_push.cpython-36.pyc │ └── fetch_reach.cpython-36.pyc ├── fetch_pick_and_place.py ├── fetch_push.py ├── fetch_reach.py ├── four_rooms │ ├── __pycache__ │ │ ├── ant.cpython-36.pyc │ │ ├── ant_maze_env.cpython-36.pyc │ │ ├── create_maze_env.cpython-36.pyc │ │ ├── maze_env.cpython-36.pyc │ │ ├── maze_env_utils.cpython-36.pyc │ │ ├── point.cpython-36.pyc │ │ └── return_env.cpython-36.pyc │ ├── ant.py │ ├── ant_maze_env.py │ ├── assets │ │ ├── ant.xml │ │ ├── point.xml │ │ ├── point2.xml │ │ ├── point_pos.xml │ │ └── point_pos_block.xml │ ├── create_maze_env.py │ ├── maze_env.py │ ├── maze_env_utils.py │ ├── point.py │ └── return_env.py └── reacher │ ├── __pycache__ │ └── reacher_three.cpython-36.pyc │ ├── reacher_three.py │ └── reacher_three.xml ├── evaluate.py ├── experiments ├── FetchPickAndPlace │ ├── experiment_args.json │ └── parameters_final.pkl └── FetchPush │ ├── experiment_args.json │ ├── logs │ ├── events.out.tfevents.1589461389.henry-PC │ ├── events.out.tfevents.1589461826.henry-PC │ ├── events.out.tfevents.1589461885.henry-PC │ ├── events.out.tfevents.1589461902.henry-PC │ ├── events.out.tfevents.1589462467.henry-PC │ ├── events.out.tfevents.1589462645.henry-PC │ └── events.out.tfevents.1589462943.henry-PC │ └── parameters_final.pkl ├── imagination.py ├── networks.py ├── one_step_model.py ├── planner.py ├── plot_utils.py ├── replay_buffer.py ├── spectral_norm.py ├── train.py └── visualise_trajectories.ipynb /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/henrycharlesworth/PlanGAN/HEAD/README.md -------------------------------------------------------------------------------- /controller.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/henrycharlesworth/PlanGAN/HEAD/controller.py -------------------------------------------------------------------------------- /envs/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/henrycharlesworth/PlanGAN/HEAD/envs/__init__.py -------------------------------------------------------------------------------- /envs/__pycache__/__init__.cpython-36.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/henrycharlesworth/PlanGAN/HEAD/envs/__pycache__/__init__.cpython-36.pyc -------------------------------------------------------------------------------- /envs/__pycache__/fetch_pick_and_place.cpython-36.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/henrycharlesworth/PlanGAN/HEAD/envs/__pycache__/fetch_pick_and_place.cpython-36.pyc -------------------------------------------------------------------------------- /envs/__pycache__/fetch_push.cpython-36.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/henrycharlesworth/PlanGAN/HEAD/envs/__pycache__/fetch_push.cpython-36.pyc -------------------------------------------------------------------------------- /envs/__pycache__/fetch_reach.cpython-36.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/henrycharlesworth/PlanGAN/HEAD/envs/__pycache__/fetch_reach.cpython-36.pyc -------------------------------------------------------------------------------- /envs/fetch_pick_and_place.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/henrycharlesworth/PlanGAN/HEAD/envs/fetch_pick_and_place.py -------------------------------------------------------------------------------- /envs/fetch_push.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/henrycharlesworth/PlanGAN/HEAD/envs/fetch_push.py -------------------------------------------------------------------------------- /envs/fetch_reach.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/henrycharlesworth/PlanGAN/HEAD/envs/fetch_reach.py -------------------------------------------------------------------------------- /envs/four_rooms/__pycache__/ant.cpython-36.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/henrycharlesworth/PlanGAN/HEAD/envs/four_rooms/__pycache__/ant.cpython-36.pyc -------------------------------------------------------------------------------- /envs/four_rooms/__pycache__/ant_maze_env.cpython-36.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/henrycharlesworth/PlanGAN/HEAD/envs/four_rooms/__pycache__/ant_maze_env.cpython-36.pyc -------------------------------------------------------------------------------- /envs/four_rooms/__pycache__/create_maze_env.cpython-36.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/henrycharlesworth/PlanGAN/HEAD/envs/four_rooms/__pycache__/create_maze_env.cpython-36.pyc -------------------------------------------------------------------------------- /envs/four_rooms/__pycache__/maze_env.cpython-36.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/henrycharlesworth/PlanGAN/HEAD/envs/four_rooms/__pycache__/maze_env.cpython-36.pyc -------------------------------------------------------------------------------- /envs/four_rooms/__pycache__/maze_env_utils.cpython-36.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/henrycharlesworth/PlanGAN/HEAD/envs/four_rooms/__pycache__/maze_env_utils.cpython-36.pyc -------------------------------------------------------------------------------- /envs/four_rooms/__pycache__/point.cpython-36.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/henrycharlesworth/PlanGAN/HEAD/envs/four_rooms/__pycache__/point.cpython-36.pyc -------------------------------------------------------------------------------- /envs/four_rooms/__pycache__/return_env.cpython-36.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/henrycharlesworth/PlanGAN/HEAD/envs/four_rooms/__pycache__/return_env.cpython-36.pyc -------------------------------------------------------------------------------- /envs/four_rooms/ant.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/henrycharlesworth/PlanGAN/HEAD/envs/four_rooms/ant.py -------------------------------------------------------------------------------- /envs/four_rooms/ant_maze_env.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/henrycharlesworth/PlanGAN/HEAD/envs/four_rooms/ant_maze_env.py -------------------------------------------------------------------------------- /envs/four_rooms/assets/ant.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/henrycharlesworth/PlanGAN/HEAD/envs/four_rooms/assets/ant.xml -------------------------------------------------------------------------------- /envs/four_rooms/assets/point.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/henrycharlesworth/PlanGAN/HEAD/envs/four_rooms/assets/point.xml -------------------------------------------------------------------------------- /envs/four_rooms/assets/point2.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/henrycharlesworth/PlanGAN/HEAD/envs/four_rooms/assets/point2.xml -------------------------------------------------------------------------------- /envs/four_rooms/assets/point_pos.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/henrycharlesworth/PlanGAN/HEAD/envs/four_rooms/assets/point_pos.xml -------------------------------------------------------------------------------- /envs/four_rooms/assets/point_pos_block.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/henrycharlesworth/PlanGAN/HEAD/envs/four_rooms/assets/point_pos_block.xml -------------------------------------------------------------------------------- /envs/four_rooms/create_maze_env.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/henrycharlesworth/PlanGAN/HEAD/envs/four_rooms/create_maze_env.py -------------------------------------------------------------------------------- /envs/four_rooms/maze_env.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/henrycharlesworth/PlanGAN/HEAD/envs/four_rooms/maze_env.py -------------------------------------------------------------------------------- /envs/four_rooms/maze_env_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/henrycharlesworth/PlanGAN/HEAD/envs/four_rooms/maze_env_utils.py -------------------------------------------------------------------------------- /envs/four_rooms/point.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/henrycharlesworth/PlanGAN/HEAD/envs/four_rooms/point.py -------------------------------------------------------------------------------- /envs/four_rooms/return_env.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/henrycharlesworth/PlanGAN/HEAD/envs/four_rooms/return_env.py -------------------------------------------------------------------------------- /envs/reacher/__pycache__/reacher_three.cpython-36.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/henrycharlesworth/PlanGAN/HEAD/envs/reacher/__pycache__/reacher_three.cpython-36.pyc -------------------------------------------------------------------------------- /envs/reacher/reacher_three.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/henrycharlesworth/PlanGAN/HEAD/envs/reacher/reacher_three.py -------------------------------------------------------------------------------- /envs/reacher/reacher_three.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/henrycharlesworth/PlanGAN/HEAD/envs/reacher/reacher_three.xml -------------------------------------------------------------------------------- /evaluate.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/henrycharlesworth/PlanGAN/HEAD/evaluate.py -------------------------------------------------------------------------------- /experiments/FetchPickAndPlace/experiment_args.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/henrycharlesworth/PlanGAN/HEAD/experiments/FetchPickAndPlace/experiment_args.json -------------------------------------------------------------------------------- /experiments/FetchPickAndPlace/parameters_final.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/henrycharlesworth/PlanGAN/HEAD/experiments/FetchPickAndPlace/parameters_final.pkl -------------------------------------------------------------------------------- /experiments/FetchPush/experiment_args.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/henrycharlesworth/PlanGAN/HEAD/experiments/FetchPush/experiment_args.json -------------------------------------------------------------------------------- /experiments/FetchPush/logs/events.out.tfevents.1589461389.henry-PC: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /experiments/FetchPush/logs/events.out.tfevents.1589461826.henry-PC: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /experiments/FetchPush/logs/events.out.tfevents.1589461885.henry-PC: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /experiments/FetchPush/logs/events.out.tfevents.1589461902.henry-PC: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /experiments/FetchPush/logs/events.out.tfevents.1589462467.henry-PC: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /experiments/FetchPush/logs/events.out.tfevents.1589462645.henry-PC: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /experiments/FetchPush/logs/events.out.tfevents.1589462943.henry-PC: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /experiments/FetchPush/parameters_final.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/henrycharlesworth/PlanGAN/HEAD/experiments/FetchPush/parameters_final.pkl -------------------------------------------------------------------------------- /imagination.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/henrycharlesworth/PlanGAN/HEAD/imagination.py -------------------------------------------------------------------------------- /networks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/henrycharlesworth/PlanGAN/HEAD/networks.py -------------------------------------------------------------------------------- /one_step_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/henrycharlesworth/PlanGAN/HEAD/one_step_model.py -------------------------------------------------------------------------------- /planner.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/henrycharlesworth/PlanGAN/HEAD/planner.py -------------------------------------------------------------------------------- /plot_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/henrycharlesworth/PlanGAN/HEAD/plot_utils.py -------------------------------------------------------------------------------- /replay_buffer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/henrycharlesworth/PlanGAN/HEAD/replay_buffer.py -------------------------------------------------------------------------------- /spectral_norm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/henrycharlesworth/PlanGAN/HEAD/spectral_norm.py -------------------------------------------------------------------------------- /train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/henrycharlesworth/PlanGAN/HEAD/train.py -------------------------------------------------------------------------------- /visualise_trajectories.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/henrycharlesworth/PlanGAN/HEAD/visualise_trajectories.ipynb --------------------------------------------------------------------------------