使用场景
在 UI 导航任务中,Magma-8B 能够根据图像输入生成正确的操作指令,如点击特定按钮。
在机器人操作任务中,模型可以根据视频输入生成机械臂的操作路径。
在多模态问答任务中,Magma-8B 能够结合图像和文本生成准确的答案。
产品特色
支持图像和视频条件下的文本生成,如描述生成和问答。
具备视觉规划能力,可生成完成任务的视觉轨迹。
能够进行 UI 接地(如点击按钮)和机器人操作(如机械臂控制)。
通过无标签视频数据学习时空关系,提升泛化能力。
在多模态任务中表现出色,尤其在空间和时间理解方面。
使用教程
1. 安装必要的依赖包,如 transformers、torch、torchvision、Pillow 和 open_clip_torch。
2. 使用 transformers 库加载 Magma-8B 模型和处理器。
3. 准备输入数据,包括图像和文本提示。
4. 使用处理器对输入数据进行预处理,并将数据传递给模型。
5. 调用模型的生成函数,获取模型输出的文本结果。
6. 对生成的文本结果进行解码和后处理,以获取最终的输出内容。