Skip to content

Robot Agent using VLLMs to make long horizon plans. Isaac Sim, BEHAVIOR, Robosuite

Notifications You must be signed in to change notification settings

AshisGhosh/roboai

Repository files navigation

RoboAI: Playground + Framework for applying LLM/VLMs to Robots in Sim

Update Videos:

  • May 27 2024 - VIDEO - Robot learning task relevant information and factoring that in the plan -- integrated with OmniGibson from Stanford/NVIDIA
  • May 8 2024 - VIDEO - Robot going from plain text to grasping attempt -- integrated with ROS2, MoveIt2, a grasping model and Isaac Sim.

Simulation Frameworks

MuJoCo & Robosuite

Mujoco is Google DeepMind's physics simulation.

Robosuite is a modular framework built on top of MuJoCo.

In the /robosim folder you'll find a Robosuite/MuJoCo sim environment:

  • Focused on Panda arm grasping objects in pick and place environment
  • Camera views to focus on objects
  • Markers to indicate robot goal and grasp targets
  • Simple API to control the robot

Isaac Sim

Isaac Sim is NVIDIA's robot simulation powered by GPUs.

Isaac Sim offers advanced tooling as well as close to real rendering. This was adopted to better test vision models.

Isaac Sim does not support external async frameworks as well - the development towards it in this project is still in progress and may need some re-architecting.

The simulation

  • Focuses on the Panda arm on a table with objects to grasp
  • Cameras for different views
  • Initial work on Markers - rendering/material support is still WIP

Models & LLM Framework

The high-level goal is to be able to command a robot to complete a long-horizon task with natural language.

An example would be to "clear the messy table".

LLMs

LLMs are used in planning layer. Once the scene is understood an LLM (either iteratively or with CoT/ToT) to generate a robot affordable plan.

Currently focused on free models hosted on openrouter.ai.

VLMs

VLMs are an extremely fast changing space. Current work is focused on:

About

Robot Agent using VLLMs to make long horizon plans. Isaac Sim, BEHAVIOR, Robosuite

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published