AndrejOrsula / drl_grasping

Deep Reinforcement Learning for Robotic Grasping from Octrees
https://arxiv.org/pdf/2208.00818
BSD 3-Clause "New" or "Revised" License
404 stars 54 forks source link

Octree with proprioceptive observations #71

Closed AndrejOrsula closed 3 years ago

AndrejOrsula commented 3 years ago

Implements https://github.com/AndrejOrsula/drl_grasping/issues/38

Adds end-effector position, orientation and gripper state to observations space alongside octree. These proprioceptive observations are stacked together with octrees (if octree stacking is enabled).

Due to the current lack of Dict obs support in stable-baselines3 (track PR https://github.com/DLR-RM/stable-baselines3/pull/243), this implementation is kinda hacky, but works...