跳到主要内容

MiniCPM-RobotManip

MiniCPM-Robot 具身智能系列

MiniCPM-RobotManip 是一个 1.5B 通用视觉-语言-动作模型,面向机器人操作任务。它以一套权重覆盖多个下游任务,并通过流式推理保留历史视觉上下文。

模型下载​

核心能力​

  • 通用操作:一套 1.5B 权重覆盖多种下游机器人操作任务。
  • 流式上下文:将历史观测持续纳入上下文,最高支持约 1 分钟视觉记忆。
  • 高效视觉编码:继承 MiniCPM-V 4.6 的视觉 Token 压缩,每帧由 256 个视觉 Token 压缩至 64 个。
  • 多视角推理:支持通过多个图像输入提供前视、腕部等不同相机视角。

输入与输出​

项目说明
输入一张或多张图像、自然语言指令、可选机器人状态与 embodiment ID
输出动作块,官方单样本脚本默认输出形状为 (30, 80)

快速上手​

跟随本教程可以完成 MiniCPM-RobotManip 的环境安装与单样本推理。 官方测试环境为 Python 3.10、PyTorch 2.6.0 和 CUDA 12.4。

获取代码​

Bash
git clone https://github.com/OpenBMB/MiniCPM-Robot.git
cd MiniCPM-Robot/MiniCPM-RobotManip

创建环境​

Bash
conda env create -f environment.yml
conda activate MiniCPM-RobotManip

准备模型权重​

从 Hugging Face 下载完整模型快照,并将路径作为推理命令的 --checkpoint 参数。

单视角推理​

Bash
python vla_infer.py \
--image frame.jpg \
--text "Pick up the red block." \
--checkpoint ./checkpoint \
--state-file state.npy \
--embodiment-id 0 \
--output action.npy

机器人状态可以省略,省略时脚本默认使用零状态;不传 --output 时,动作预测将以 JSON 输出。

多视角推理​

Bash
python vla_infer.py \
--image cam_front.jpg \
--image cam_wrist.jpg \
--text "Pick up the red block." \
--checkpoint ./checkpoint

输出说明​

官方单样本推理脚本返回一个动作块,默认形状为 (30, 80)。接入具体机器人前,需要根据 embodiment、状态维度和控制接口完成动作映射与安全约束。

相关资源​