Go2Py is a Pythonic interface and driver for low-level and high-level control of Unitree Go2 quadruped robots. The motivation of this project is to remove the burden of developing interface, safety systems, and basic components required for starting locomotion research using the Go2 quadruped robot. It provides a modular pipeline for real-time communication with the robot in both simulated and real-world environments with a unified interface.
This project is comprised of the following components:
- C++ Bridge: A dockerized ROS2 bridge built upon the unitree_ros2 that implements a remote-controlled emergency stop and publishes the robot states as standard ROS2 topics usable by upstream systems such as NAV2.
- Robot Interface: A simple Python class that represents the robot and communicates with the C++ bridge through either DDS (ROS independent) or ROS2 interfaces.
- Robot Management FSM: A finite state machine for controlling the behavior of the robot up to the point of handover to the user low-level controller (sitting down, standing up) with safety monitors (motor temperatures, emergency stops).
- Robot Model: A simple to use Pinocchio wrapper for computing the kinematics and dynamics parameters of the robot.
- Simulation Interface: Simulation environments based on Mujoco and Nvidia Orbit (To be added) with a Python interface identical to the real robot.
Communication with the robot will be as simple as importing a Python class:
from Go2Py.robot.interface.dds import GO2Real
from Go2Py.robot.model import Go2Model
robot = GO2Real(mode='lowlevel')
model = Go2Model()
while running:
joint_state = robot.getJointStates()
imu = robot.getIMU()
remote = robot.getRemoteState()
model.update(state['q'], state['dq'],T,vel) # T and vel from the EKF
info = model.getInfo()
#User control computations ...
robot.setCommands(q_des, dq_des, kp, kd, tau_ff)An identical workflow can be followed for simulation:
from Go2Py.sim.mujoco import Go2Sim
from Go2Py.robot.model import Go2Model
robot = Go2Sim()
model = Go2Model()
robot.standDownReset()
while running:
joint_state = robot.getJointStates()
imu = robot.getIMU()
remote = robot.getRemoteState()
model.update(state['q'], state['dq'],T,vel) # T and vel from the EKF
info = model.getInfo()
#User control computations ...
robot.setCommands(q_des, dq_des, kp, kd, tau_ff)
robot.step()Follow through the steps here to setup the robot and Go2Py.
A set of examples and tutorials are provided to get you up and running quickly: