├── .gitignore ├── README.md ├── conda_env.yml └── src ├── agent ├── bc.py ├── drqv2.py ├── potil.py ├── resnet.py └── vit.py ├── cfgs ├── agent │ ├── bc.yaml │ ├── drqv2.yaml │ └── potil.yaml ├── config.yaml └── suite │ ├── kitchen.yaml │ ├── kitchen_task │ ├── open_door.yaml │ ├── open_micro.yaml │ ├── params.yaml │ ├── slide_door.yaml │ ├── turn_knob.yaml │ └── turn_light_on.yaml │ ├── metaworld.yaml │ ├── metaworld_task │ ├── bin_picking.yaml │ ├── button_press_topdown.yaml │ ├── coffee_pull.yaml │ ├── door_open.yaml │ ├── drawer_close.yaml │ ├── hammer.yaml │ ├── lever_pull.yaml │ ├── params.yaml │ └── window_close.yaml │ ├── robosuite.yaml │ └── robosuite_task │ ├── jaco_door.yaml │ ├── jaco_lift.yaml │ ├── jaco_twoarm_peginhole.yaml │ ├── panda_door.yaml │ ├── panda_lift.yaml │ ├── panda_nut_assembly_square.yaml │ ├── panda_pickplace_can.yaml │ ├── panda_twoarm_peginhole.yaml │ └── params.yaml ├── logger.py ├── replay_buffer.py ├── rewarder.py ├── suite ├── kitchen.py ├── metaworld.py └── robosuite.py ├── train_bc.py ├── train_rl.py ├── train_vrf.py ├── utils.py └── video.py /.gitignore: -------------------------------------------------------------------------------- 1 | .DS_Store 2 | .idea -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Yingdong-Hu/PVM-Robotics/HEAD/README.md -------------------------------------------------------------------------------- /conda_env.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Yingdong-Hu/PVM-Robotics/HEAD/conda_env.yml -------------------------------------------------------------------------------- /src/agent/bc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Yingdong-Hu/PVM-Robotics/HEAD/src/agent/bc.py -------------------------------------------------------------------------------- /src/agent/drqv2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Yingdong-Hu/PVM-Robotics/HEAD/src/agent/drqv2.py -------------------------------------------------------------------------------- /src/agent/potil.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Yingdong-Hu/PVM-Robotics/HEAD/src/agent/potil.py -------------------------------------------------------------------------------- /src/agent/resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Yingdong-Hu/PVM-Robotics/HEAD/src/agent/resnet.py -------------------------------------------------------------------------------- /src/agent/vit.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Yingdong-Hu/PVM-Robotics/HEAD/src/agent/vit.py -------------------------------------------------------------------------------- /src/cfgs/agent/bc.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Yingdong-Hu/PVM-Robotics/HEAD/src/cfgs/agent/bc.yaml -------------------------------------------------------------------------------- /src/cfgs/agent/drqv2.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Yingdong-Hu/PVM-Robotics/HEAD/src/cfgs/agent/drqv2.yaml -------------------------------------------------------------------------------- /src/cfgs/agent/potil.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Yingdong-Hu/PVM-Robotics/HEAD/src/cfgs/agent/potil.yaml -------------------------------------------------------------------------------- /src/cfgs/config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Yingdong-Hu/PVM-Robotics/HEAD/src/cfgs/config.yaml -------------------------------------------------------------------------------- /src/cfgs/suite/kitchen.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Yingdong-Hu/PVM-Robotics/HEAD/src/cfgs/suite/kitchen.yaml -------------------------------------------------------------------------------- /src/cfgs/suite/kitchen_task/open_door.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Yingdong-Hu/PVM-Robotics/HEAD/src/cfgs/suite/kitchen_task/open_door.yaml -------------------------------------------------------------------------------- /src/cfgs/suite/kitchen_task/open_micro.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Yingdong-Hu/PVM-Robotics/HEAD/src/cfgs/suite/kitchen_task/open_micro.yaml -------------------------------------------------------------------------------- /src/cfgs/suite/kitchen_task/params.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Yingdong-Hu/PVM-Robotics/HEAD/src/cfgs/suite/kitchen_task/params.yaml -------------------------------------------------------------------------------- /src/cfgs/suite/kitchen_task/slide_door.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Yingdong-Hu/PVM-Robotics/HEAD/src/cfgs/suite/kitchen_task/slide_door.yaml -------------------------------------------------------------------------------- /src/cfgs/suite/kitchen_task/turn_knob.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Yingdong-Hu/PVM-Robotics/HEAD/src/cfgs/suite/kitchen_task/turn_knob.yaml -------------------------------------------------------------------------------- /src/cfgs/suite/kitchen_task/turn_light_on.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Yingdong-Hu/PVM-Robotics/HEAD/src/cfgs/suite/kitchen_task/turn_light_on.yaml -------------------------------------------------------------------------------- /src/cfgs/suite/metaworld.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Yingdong-Hu/PVM-Robotics/HEAD/src/cfgs/suite/metaworld.yaml -------------------------------------------------------------------------------- /src/cfgs/suite/metaworld_task/bin_picking.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Yingdong-Hu/PVM-Robotics/HEAD/src/cfgs/suite/metaworld_task/bin_picking.yaml -------------------------------------------------------------------------------- /src/cfgs/suite/metaworld_task/button_press_topdown.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Yingdong-Hu/PVM-Robotics/HEAD/src/cfgs/suite/metaworld_task/button_press_topdown.yaml -------------------------------------------------------------------------------- /src/cfgs/suite/metaworld_task/coffee_pull.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Yingdong-Hu/PVM-Robotics/HEAD/src/cfgs/suite/metaworld_task/coffee_pull.yaml -------------------------------------------------------------------------------- /src/cfgs/suite/metaworld_task/door_open.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Yingdong-Hu/PVM-Robotics/HEAD/src/cfgs/suite/metaworld_task/door_open.yaml -------------------------------------------------------------------------------- /src/cfgs/suite/metaworld_task/drawer_close.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Yingdong-Hu/PVM-Robotics/HEAD/src/cfgs/suite/metaworld_task/drawer_close.yaml -------------------------------------------------------------------------------- /src/cfgs/suite/metaworld_task/hammer.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Yingdong-Hu/PVM-Robotics/HEAD/src/cfgs/suite/metaworld_task/hammer.yaml -------------------------------------------------------------------------------- /src/cfgs/suite/metaworld_task/lever_pull.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Yingdong-Hu/PVM-Robotics/HEAD/src/cfgs/suite/metaworld_task/lever_pull.yaml -------------------------------------------------------------------------------- /src/cfgs/suite/metaworld_task/params.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Yingdong-Hu/PVM-Robotics/HEAD/src/cfgs/suite/metaworld_task/params.yaml -------------------------------------------------------------------------------- /src/cfgs/suite/metaworld_task/window_close.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Yingdong-Hu/PVM-Robotics/HEAD/src/cfgs/suite/metaworld_task/window_close.yaml -------------------------------------------------------------------------------- /src/cfgs/suite/robosuite.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Yingdong-Hu/PVM-Robotics/HEAD/src/cfgs/suite/robosuite.yaml -------------------------------------------------------------------------------- /src/cfgs/suite/robosuite_task/jaco_door.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Yingdong-Hu/PVM-Robotics/HEAD/src/cfgs/suite/robosuite_task/jaco_door.yaml -------------------------------------------------------------------------------- /src/cfgs/suite/robosuite_task/jaco_lift.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Yingdong-Hu/PVM-Robotics/HEAD/src/cfgs/suite/robosuite_task/jaco_lift.yaml -------------------------------------------------------------------------------- /src/cfgs/suite/robosuite_task/jaco_twoarm_peginhole.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Yingdong-Hu/PVM-Robotics/HEAD/src/cfgs/suite/robosuite_task/jaco_twoarm_peginhole.yaml -------------------------------------------------------------------------------- /src/cfgs/suite/robosuite_task/panda_door.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Yingdong-Hu/PVM-Robotics/HEAD/src/cfgs/suite/robosuite_task/panda_door.yaml -------------------------------------------------------------------------------- /src/cfgs/suite/robosuite_task/panda_lift.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Yingdong-Hu/PVM-Robotics/HEAD/src/cfgs/suite/robosuite_task/panda_lift.yaml -------------------------------------------------------------------------------- /src/cfgs/suite/robosuite_task/panda_nut_assembly_square.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Yingdong-Hu/PVM-Robotics/HEAD/src/cfgs/suite/robosuite_task/panda_nut_assembly_square.yaml -------------------------------------------------------------------------------- /src/cfgs/suite/robosuite_task/panda_pickplace_can.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Yingdong-Hu/PVM-Robotics/HEAD/src/cfgs/suite/robosuite_task/panda_pickplace_can.yaml -------------------------------------------------------------------------------- /src/cfgs/suite/robosuite_task/panda_twoarm_peginhole.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Yingdong-Hu/PVM-Robotics/HEAD/src/cfgs/suite/robosuite_task/panda_twoarm_peginhole.yaml -------------------------------------------------------------------------------- /src/cfgs/suite/robosuite_task/params.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Yingdong-Hu/PVM-Robotics/HEAD/src/cfgs/suite/robosuite_task/params.yaml -------------------------------------------------------------------------------- /src/logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Yingdong-Hu/PVM-Robotics/HEAD/src/logger.py -------------------------------------------------------------------------------- /src/replay_buffer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Yingdong-Hu/PVM-Robotics/HEAD/src/replay_buffer.py -------------------------------------------------------------------------------- /src/rewarder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Yingdong-Hu/PVM-Robotics/HEAD/src/rewarder.py -------------------------------------------------------------------------------- /src/suite/kitchen.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Yingdong-Hu/PVM-Robotics/HEAD/src/suite/kitchen.py -------------------------------------------------------------------------------- /src/suite/metaworld.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Yingdong-Hu/PVM-Robotics/HEAD/src/suite/metaworld.py -------------------------------------------------------------------------------- /src/suite/robosuite.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Yingdong-Hu/PVM-Robotics/HEAD/src/suite/robosuite.py -------------------------------------------------------------------------------- /src/train_bc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Yingdong-Hu/PVM-Robotics/HEAD/src/train_bc.py -------------------------------------------------------------------------------- /src/train_rl.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Yingdong-Hu/PVM-Robotics/HEAD/src/train_rl.py -------------------------------------------------------------------------------- /src/train_vrf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Yingdong-Hu/PVM-Robotics/HEAD/src/train_vrf.py -------------------------------------------------------------------------------- /src/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Yingdong-Hu/PVM-Robotics/HEAD/src/utils.py -------------------------------------------------------------------------------- /src/video.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Yingdong-Hu/PVM-Robotics/HEAD/src/video.py --------------------------------------------------------------------------------