Skip to content

大模型具身智能比赛-机器人控制端

License

Notifications You must be signed in to change notification settings

zainhub/RoboWaiter

 
 

Repository files navigation

RoboWaiter

本项目为参加达闼杯“机器人大模型与具身智能挑战赛”的参赛作品。我们的目标是结合前沿的大模型技术和具身智能技术,开发能在模拟的咖啡厅场景中承担服务员角色并自主完成各种具身任务的智能机器人。这里是我们的参赛作品《基于大模型和行为树和生成式具身智能体》的机器人控制端代码。

1. 技术简介

我们提出基于大模型和行为树的生成式具身智能体系统框架

1 行为树是系统的中枢,作为大模型和具身智能之间的桥梁,解决两者结合的挑战

2 大语言模型是系统的大脑。一方面,我们设计了向量数据库和工具调用,另一方面,在实现智能体规划上,我们不再需要大语言模型输出完整的动作序列,而仅仅给出一个任务目标,这大大缓解了大模型的具身幻觉现象。

3 而具身机器人是系统的躯体,在条件节点感知和动作节点控制的函数中,我们优化了接口调用和算法设计,提高感知高效性和控制准确性

image

2. 项目安装(必看)

2.1 环境要求

Python=3.10

2.2 安装步骤

git clone https://github.com/HPCL-EI/RoboWaiter.git
cd RoboWaiter
pip install -e .

以上步骤将完成robowaiter项目以及相关依赖库的安装

2.3 安装UI

  1. 安装 graphviz-9.0.0 (详见官网)

  2. 将软件安装目录的bin文件添加到系统环境中。如电脑是 Windows 系统,Graphviz 安装在 D:\Program Files (x86)\Graphviz2.38,该目录下有bin文件,将该路径添加到电脑系统环境变量 path 中,即 D:\Program Files (x86)\Graphviz2.38\bin。如果不行,则需要重启。

  3. 安装向量数据库 conda install -c conda-forge faiss

  4. 安装自然语言处理和翻译工具,用于计算相似性

    pip install translate
    pip install spacy 
    python -m spacy download zh_core_web_lg
    

    zh_core_web_lg 如果下载较慢,可以直接通过分享的网盘链接下载

    链接:https://pan.baidu.com/s/1vr7dqHsgnh6UChymQc26VA 提取码:1201 --来自百度网盘超级会员V7的分享

    pip install zh_core_web_lg-3.7.0-py3-none-any.whl
    

2.4 部署大模型

可采用本地部署大模型或调用在线大模型API的方式。

  1. 本地部署 比赛中使用的是ChatGLM3模型,并部署在本地集群上。 可参考 https://github.com/THUDM/ChatGLM3 官方链接进行部署,并根据该小节来开放API。

  2. 使用在线大模型API 若不方便本地部署大模型,建议直接调用在线大模型的API,如GLM4,文心一言, GPT4 等。

无论是本地部署还是直接使用在线大模型API,最终都要在 robowaiter/llm_client/single_round.py 文件中修改相应的API地址,以便机器人与大模型通信。

2.5 快速入门

  1. 下载仿真器,解压并运行CafeSimulator.exe,即可打开仿真器,此时显示空场景,等待代码端生成场景并完成机器人交互。

  2. 不使用 UI 界面 :运行 tasks_no_ui 文件夹下的任意场景即可实现机器人控制端与仿真器的交互

  3. 使用 UI 界面:运行 run_ui.py ,显示下面的界面。点击左侧的按钮,机器人就会执行相应的任务。也可以在右上方直接输出目标状态或者对话和机器人直接交互。 1701353768755

3. 代码框架介绍

代码库被组织成几个模块,每个模块负责系统功能的一部分:

  • behavior_lib: behavior_lib 是行为树节点库类,包括行为树的动作节点和条件节点。它们分别存放在 actcond 文件夹下。
  • behavior_tree: behavior_tree 是行为树算法类,包括 ptml 编译器、最优行为树逆向扩展算法等。
  • robot: robot 是机器人类,包括从 ptml加载行为树的方法,以及执行行为树的方法等。
  • llm_client: llm_client 是大模型类,主要实现了大模型的数据集构建、数据处理工具、大模型调用接口、大模型评测、工具调用、工具注册、向量数据库、单论对话、对轮对话等方法或接口。

​ 调用大模型接口。运行llm_client.py文件调用大模型进行多轮对话。输入字符即可等待回答/

cd robowaiter/llm_client
python multi_rounds.py
  • scene: scene 是场景基类,该类实现了一些通用的场景操作接口,实现了与 UE 和咖啡厅仿真场景的通信。其中,包括了官方已经封装好的各种接口,如场景初始化、行人控制、操作动画设置、物品设置、机器人 IK 接口等。task_map 返回的任务场景都继承于 Scene。此外,在 scene/ui 中,我们实现了 UI 的界面设计和接口封装。
  • utils: utils为其它工具类,比如绘制行为树并输出为图片文件。
  • algos: algos 是其它算法类,包括MemGPT、导航算法 (navigator)、边界探索 (explore)、视觉算法 (vision)、向量数据库 (retrieval) 等。
  • tasks: tasks 文件夹中存放的场景定义及运行代码。
缩写 任务
AEM 主动探索和记忆
GQA 具身多轮对话
VLN 视觉语言导航
VLM 视觉语言操作
OT 复杂开放任务
AT 自主任务
CafeDailyOperations 整体展示:咖啡厅的一天
Interact 命令行自由交互

4. 花絮

机器人根据顾客的点单,完成订单并送餐

image

顾客询问物品位置,并要求机器人送回

image

版权所有 (c) [2023] [NUDT-HPCL-EI]

About

大模型具身智能比赛-机器人控制端

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • Python 99.7%
  • Other 0.3%