RoboBrain 2.0 具身大脑模型复现指南：统一感知、推理与规划

执行下面命令，进行依赖库安装：

pip install -r requirements.txt

3、安装 torch

这里指定 CUDA12.1 和 torch==2.5.0 的版本。

执行下面命令，进行 torch 的安装：

pip install torch==2.5.0 torchvision==0.20.0 torchaudio==2.5.0 --index-url https://download.pytorch.org/whl/cu121

4、模型推理

主要包括：图文问答（支持思考模式）、目标检测（视觉基础能力）、可供性预测（具身认知）、操作轨迹预测与规划（具身决策）、具身导航任务。

在首次运行时，需要下载模型权重的，比如 RoboBrain2.0-7B：

备注：需要在终端输入'huggingface-cli login'命令，然后输入'Access Tokens'进行登陆。而且下载 RoboBrain2.0 的模型，需要先签名认证，才能下载的（对应下面图片的'verified'）。

例如：https://huggingface.co/BAAI/RoboBrain2.0-7B/tree/main

示例 1：图文问答，使用 RoboBrain2.0-7B 模型，不开思考模式

示例代码：

from inference import UnifiedInference
model = UnifiedInference("BAAI/RoboBrain2.0-7B")
prompt = "What is shown in this image?"
image = "http://images.cocodataset.org/val2017/000000039769.jpg"
pred = model.inference(prompt, image, task="general", enable_thinking=False, do_sample=True)
print(f"Prediction:\n{pred}")

输入图片：

模型推理结果：

Loading Checkpoint ... Loading checkpoint shards: 100%|██████████████████████████████| 4/4 [00:01<00:00, 3.67it/s] Some parameters are on the meta device because they were offloaded to the cpu. ...... Model thinking support: True

==================== INPUT ==================== What is shown in this image?

Thinking disabled (but supported). Running inference ... Prediction: {'answer': 'Two cats are shown sleeping in the image, lying on a couch.'}

示例 2：图文问答，使用 RoboBrain2.0-7B 模型，开启思考模式

示例代码：

from inference import UnifiedInference
model = UnifiedInference("BAAI/RoboBrain2.0-7B")
prompt = "What is shown in this image?"
image = "http://images.cocodataset.org/val2017/000000039769.jpg"
pred = model.inference(prompt, image, task="general", enable_thinking=True, do_sample=True)
print(f"Prediction:\n{pred}")

输入图片（和示例 1 一样的）

模型推理结果，描述得很详细，而且准确：

Loading Checkpoint ... Loading checkpoint shards: 100%|██████████████████████████████| 4/4 [00:01<00:00, 3.67it/s] Some parameters are on the meta device because they were offloaded to the cpu. ...... Model thinking support: True

==================== INPUT ==================== What is shown in this image?

Thinking disabled (but supported). Running inference ... Prediction:

{'answer': 'The image shows two cats sleeping on a couch covered with a pink blanket or sheet, and a couple of remote controls nearby.', 'thinking': 'Within the visual input, there are two cats, both of which appear to be sleeping, lying on a couch. The couch has a pink blanket or sheet that covers its surface. Additionally, a couple of remote controls can be seen lying nearby, suggesting that someone might have been watching television or using the devices before the cats decided to snooze.'}

中文意思：

{'answer': '图片中，两只猫正睡在一张铺着粉色毯子或床单的沙发上，旁边还放着几个遥控器。', 'thinking': '在视觉输入中，可以看到两只猫，它们似乎都在睡觉，躺在沙发上。沙发上铺着一条粉色的毯子或床单。此外，还能看见几个遥控器放在旁边，暗示可能有人在猫决定打盹之前正在看电视或使用这些设备。'}

示例 3：图文问答，使用 RoboBrain2.0-3B 模型

示例代码：（3B 模型没有思考模式的）

from inference import UnifiedInference
model = UnifiedInference("BAAI/RoboBrain2.0-3B")
prompt = "What is shown in this image?"
image = "http://images.cocodataset.org/val2017/000000039769.jpg"
pred = model.inference(prompt, image, task="general", do_sample=True)
print(f"Prediction:\n{pred}")

模型推理结果：

......Model thinking support: False

==================== INPUT ==================== What is shown in this image?

Thinking disabled (but supported). Running inference ... Prediction:

{'answer': 'The image shows two cats laying next to each other on a pink blanket and a couch, with their tails touching. They appear to be sleeping or relaxing in close proximity.'}

示例 4：视觉基础能力，目标检测

示例代码：

from inference import UnifiedInference
import cv2, os, datetime
model = UnifiedInference("BAAI/RoboBrain2.0-7B")
prompt = "the person wearing a red hat"
image_path = "./assets/demo/grounding.jpg"
# 1. 运行推理，plot=True 会自动在 result/ 目录生成带框图
pred = model.inference(
 prompt, image_path, task="grounding", plot=True, enable_thinking=True, do_sample=True
)
print(f"Prediction:\n{pred}")
# 2. 只保存可视化结果（自己定目录 + 文件名）
save_dir = "vis_results"
os.makedirs(save_dir, exist_ok=True)
vis_img = cv2.imread("result/grounding_with_grounding_annotated.jpg")
out_path = os.path.join(save_dir, f"redhat_det_{datetime.datetime.now().strftime('%H%M%S')}.jpg")
cv2.imwrite(out_path, vis_img)
print("可视化结果已保存至:", os.path.abspath(out_path))

可视化结果：

模型推理：

==================== INPUT ==================== Please provide the bounding box coordinate of the region this sentence describes: the person wearing a red hat.

Thinking enabled. Running inference ... Plotting enabled. Drawing results on the image ... Extracted bounding boxes: [[0, 192, 186, 614]] Annotated image saved to: result/grounding_with_grounding_annotated.jpg Prediction: {'answer': '[0, 192, 186, 614]', 'thinking': "After examining the visual input, I notice two main subjects involved in this scene — an adult male and a child. The adult is wearing a blue garment, indicating a larger individual, while the child is clad in a red cap and a striped garment, suggesting a smaller presence. The child's hand is positioned near a banana peel held by both individuals.\n\nIn terms of spatial analysis, the child, identifiable by his distinct red cap, is visually closer to the right side of the image, while the adult male is towards the left side. The banana peel, grasped by both, is centrally located, bridging their interaction visually.\n\nAnalyzing the elements around them, they appear seated on a stone surface against a wall, which serves as the backdrop. The environment suggests an outdoor setting, reinforcing the casual nature of their interaction.\n\nThus, based on my visual-spatial reasoning capabilities, the child, wearing the red cap and striped outfit, is the one interacting with the banana peel, evidenced by his proximity to it and the distinct red accessory he wears."} 可视化结果已保存至：/home/lgp/2025_project/RoboBrain2.0/vis_results/redhat_det_161556.jpg

输出的提示词，也可以是中文的，比如：prompt = "找到一个香蕉"

可视化结果：

模型推理：

==================== INPUT ==================== Please provide the bounding box coordinate of the region this sentence describes: 找到一个香蕉。

Thinking enabled. Running inference ... Plotting enabled. Drawing results on the image ... Extracted bounding boxes: [[205, 368, 290, 497]] Annotated image saved to: result/grounding_with_grounding_annotated.jpg Prediction: {'answer': '[205, 368, 290, 497]', 'thinking': 'From the visual input, I perceive a scene involving an adult and a child. The child is holding a banana, which is a key object in the scene. The adult appears to be sitting next to or behind the child, engaged with them. The child's position is slightly turned away from the observer, creating an interactive moment.\n\nAnalyzing the spatial arrangement, the banana is held in such a way that its curved shape and peel are prominent features in the visual field. This positioning allows for a clear identification of the fruit's location within the frame.\n\nThe task requires identifying the banana's location relative to certain elements in the visual input. Given the child's direct interaction with the banana, it can be determined as being held by the child. From the visual context, the banana is indeed found within the child's grasp. Therefore, based on the observed details, the answer to the question, "What is the banana?" is that it is positioned where the child is holding it.'} 可视化结果已保存至：/home/lgp/2025_project/RoboBrain2.0/vis_results/redhat_det_162204.jpg

示例 5：具身认知，用于可供性预测

示例代码：

from inference import UnifiedInference
import cv2, os, datetime
model = UnifiedInference("BAAI/RoboBrain2.0-7B")
prompt = "如何抓住杯子"
image = "./assets/demo/affordance.jpg"
# 运行推理，plot=True 会自动保存可视化图到 result/affordance_with_affordance_annotated.jpg
pred = model.inference(
 prompt, image, task="affordance", plot=True, enable_thinking=True, do_sample=True
)
print(f"Prediction:\n{pred}")
# ----------- 仅保存可视化结果 -----------
save_dir = "vis_results"
os.makedirs(save_dir, exist_ok=True)
vis_img = cv2.imread("result/affordance_with_affordance_annotated.jpg")
out_path = os.path.join(save_dir, f"affordance_{datetime.datetime.now().strftime('%H%M%S')}.jpg")
cv2.imwrite(out_path, vis_img)
print("可视化结果已保存至:", os.path.abspath(out_path))

可视化效果：

模型推理结果：

==================== INPUT ==================== You are a robot using the joint control. The task is "如何抓住杯子. Please predict a possible affordance area of the end effector.

Thinking enabled. Running inference ... Plotting enabled. Drawing results on the image ... Extracted bounding boxes: [[561, 224, 623, 303]] Annotated image saved to: result/affordance_with_affordance_annotated.jpg Prediction: {'answer': '[561, 224, 623, 303]', 'thinking': 'From the visual input, the object is recognized as a white mug, with a cylindrical body and a round handle on its right side. The approximate size suggests that it's about 10cm tall and has an opening wide enough to hold a standard coffee cup's capacity, making it around 7-8 cm in diameter. The mug appears to be made from ceramic or similar material, inferred by its smooth surface and uniform color. It rests stably on what seems like a wooden table surface, with no immediate obstructions nearby.\n\nMy end-effector, capable of joint control manipulation, is designed to grasp objects securely. Its design allows for precision in grasping curved surfaces and handles up to a certain height and thickness. The handle on the mug presents a suitable engagement point due to its shape and size, which my gripper can wrap around effectively without slipping.\n\nThe current task is to "如何抓住杯子，" which involves identifying the most secure area to grip the mug. Given its position on the table, the handle stands out as the best affordance area because it provides a stable, protruding edge that is easy to approach and firmly grasp. \n\nTherefore, the mug's handle affords an optimal interaction point. This choice is supported by both the visually confirmed handle dimensions compatible with my gripper's capabilities and the requirement to accomplish the task efficiently.'} 可视化结果已保存至：/home/lgp/2025_project/RoboBrain2.0/vis_results/affordance_162938.jpg

示例 6：用于具身的轨迹预测

示例代码：

from inference import UnifiedInference
import cv2, os, datetime
model = UnifiedInference("BAAI/RoboBrain2.0-7B")
prompt = "伸手去拿红色的瓶子"
image = "./assets/demo/trajectory.jpg"
# 运行推理，plot=True 会自动保存可视化图到 result/trajectory_with_trajectory_annotated.jpg
pred = model.inference(
 prompt, image, task="trajectory", plot=True, enable_thinking=True, do_sample=True
)
print(f"Prediction:\n{pred}")

可视化效果：

推理结果：

==================== INPUT ==================== You are a robot using the joint control. The task is "伸手去拿红色的瓶子. Please predict up to 10 key trajectory points to complete the task. Your answer should be formatted as a list of tuples, i.e. [[x1, y1], [x2, y2], ...], where each tuple contains the x and y coordinates of a point.

Thinking enabled. Running inference ... Plotting enabled. Drawing results on the image ... Extracted trajectory points: [[(145, 123), (259, 168), (327, 136)]] Annotated image saved to: result/trajectory_with_trajectory_annotated.jpg Prediction: {'answer': '[(145, 123), (259, 168), (327, 136)]', 'thinking': "From the visual input, the target object, a red bottle, is clearly identified resting upright on a black tray positioned towards the right side of the scene. My current end-effector position is to the left of the scene, near several other objects like cups and plates that do not obstruct the path directly. Notably, there's ample space between my initial position and the red bottle, allowing for an unobstructed path.\n\nMy joint control system enables me to generate smooth trajectories. I will plan a sequence of movements starting from my current location, moving towards the red bottle. Given the open field, I can calculate intermediate points ensuring a direct approach without collision with nearby objects. Up to 10 key points are suggested but fewer may suffice given the straightforward nature of this task.\n\nThe task involves reaching and grasping the red bottle. The trajectory must be efficient and direct, originating from the initial position on the left and terminating precisely at the bottle. Each segment of the motion should ensure clearance from surrounding objects, minimizing potential interaction. This requires careful placement of key waypoints to guide the motion smoothly.\n\nVerification confirms the path's logical progression. The sequence of points is checked to ensure they maintain a safe distance from obstacles while advancing toward the red bottle. The final point must accurately target the bottle's handle or body for a successful grasp.\n\nTherefore, based on this comprehensive visual analysis and planning, the key trajectory points to reach the red bottle are determined as [(145, 123), (259, 168), (327, 136)]. These points form a viable path from the initial position to the target while respecting the environment."}

示例 7：用于指向预测（具身认知）

示例代码：

from inference import UnifiedInference
model = UnifiedInference("BAAI/RoboBrain2.0-7B")
prompt = "在两个杯子之间的空隙中，找出几个可以放置杯子的位置"
image = "./assets/demo/pointing.jpg"
pred = model.inference(
 prompt, image, task="pointing", plot=True, enable_thinking=True, do_sample=True
)
print(f"Prediction:\n{pred}")

可视化效果：

推理结果：

==================== INPUT ==================== 在两个杯子之间的空隙中，找出几个可以放置杯子的位置. Your answer should be formatted as a list of tuples, i.e. [(x1, y1), (x2, y2), ...], where each tuple contains the x and y coordinates of a point satisfying the conditions above. The coordinates should indicate the normalized pixel locations of the points in the image.

Thinking enabled. Running inference ... Plotting enabled. Drawing results on the image ... Extracted points:[(334, 320), (305, 316), (369, 315), (389, 317), (314, 313), (350, 314), (378, 313), (325, 317)] Annotated image saved to: result/pointing_with_pointing_annotated.jpg Prediction: {'answer': '[(334, 320), (305, 316), (369, 315), (389, 317), (314, 313), (350, 314), (378, 313), (325, 317)]', 'thinking': "From the visual input, two cups are clearly placed on a horizontal surface, with one blue cup on the left and another green cup on the right. The empty space between them is crucial for identifying potential placement points. This area lies flat on the surface and offers an unobstructed path, making it an ideal candidate for placing additional cups.\n\nMy advanced visual processing capability allows me to segment this gap visually, ensuring that any identified points will lie entirely within this region, avoiding edges or other objects nearby. It is essential to confirm that there are no physical obstructions like shadows or reflections, which might affect perceived space availability.\n\nThe task requires determining several spots within this vacant area. I start by verifying its dimensions, confirming the gap's width and height. Next, I select multiple points, spread evenly across the free space, to ensure diversity and coverage across the entire visible region, simulating different spots to place an additional cup effectively.\n\nEach identified point undergoes verification to ensure it falls strictly within the boundaries of the open space between the cups. Distinctness among points is also confirmed, avoiding overlap and maintaining separation.\n\nTherefore, through direct visual analysis, combined with my cognitive capabilities, the identified points within the vacant space are [(334, 320), (305, 316), (369, 315), (389, 317), (314, 313), (350, 314), (378, 313), (325, 317)]. These points fulfill all requirements as they are located centrally within the available space, ensuring effective placement."}

示例 8：用于具身导航任务

示例代码：

from inference import UnifiedInference
model = UnifiedInference("BAAI/RoboBrain2.0-7B")
prompt = "来到沙发这里，我需要休息了"
image = "./assets/demo/navigation.jpg"
pred = model.inference(
 prompt, image, task="pointing", plot=True, enable_thinking=True, do_sample=True
)
print(f"Prediction:\n{pred}")

可视化效果：