vikashplus / robohive

A unified framework for robot learning
https://sites.google.com/view/robohive
Apache License 2.0
489 stars 82 forks source link

Added support for Xbox 360 controller based tele-operation in tutorials. #130

Closed dosssman closed 5 months ago

dosssman commented 7 months ago

Hello @vikashplus

Here is the PR that adds tele-operation based on the Xbox 360 controller added to vtils recently (thanks for the merge).

The tele-operation assumes the camera is facing the robot and the table when tele-operating to collect data for example. I have followed the poll_keybaord function while adaption to sensor data from the Gamepad class. One original addition is the use of the controller's X button to slow down the movement of the gripper arm for finer grained tele-operation.

I tested it with:

python robohive/tutorials/ee_teleop.py --input_device gamepad --horizon 5000

Hope this helps. Best regards.