资讯动态

CogAgent 部署教程:输入一张截图,让 9B 模型输出可执行的 GUI 操作

发布时间:2026/10/24 1:48:26 来源:尧图企业网站定制
CogAgent 部署教程输入一张截图让 9B 模型输出可执行的 GUI 操作【免费下载链接】CogAgentAn open-sourced end-to-end VLM-based GUI Agent项目地址: https://gitcode.com/GitHub_Trending/co/CogAgentCogAgent 是开源的端到端 VLM 基座 GUI Agent输入一张屏幕截图和一句中英文指令它返回该点击哪里、输入什么。按本文操作你可以在本地部署 CogAgent-9b-20241220 并跑通第一次 GUI 推理。先跑起来环境要求Python 3.10.16、NVIDIA GPUBF16 精度推理至少 29GB 显存。克隆仓库后在根目录执行pip install -r requirements.txt python inference/cli_demo.py --model_dir THUDM/cogagent-9b-20241220 --platform Mac --max_length 4096 --top_k 1 --output_image_path ./results --format_key status_action_op_sensitive--model_dir接受本地路径或模型仓库标识首次运行会自动下载 THUDM/cogagent-9b-20241220 权重也可从 HuggingFace、ModelScope 手动获取后填本地路径。启动后终端会依次提示输入任务描述和截图路径输入exit退出。模型返回带 bbox 的结果时会保存一张红框标注图到./results文件名为{你输入的图片名}_{对话轮次}.png。它是怎么工作的CogAgent 基于 GLM-4V-9B 构建定位是执行模型而非对话模型。它不维护连续对话每一轮推理都需要重新拼接结构化提示词顺序固定为 Task任务描述→ History steps历史操作→ Platform平台→ format 指令期望的返回格式。其中截图是必需输入纯文字无法完成 GUI Agent 任务。模型输出采用严格的 STR 格式而非 JSON按format_key要求返回 Status、Plan、Action 等字段末尾的 Grounded Operation 描述具体操作例如CLICK(box[[219,186,311,207]], element_info全部标为已读)box 坐标基于 0–1000 的归一化尺度示例代码除以 1000 后换算为像素坐标绘制 bbox。多步任务不是靠模型记住上下文而是把上几轮返回的 Grounded Operation 和 Action 作为 History steps 传回下一轮由你解析输出并驱动执行。常见用法命令行推理最短命令只给必填参数其余走默认值python inference/cli_demo.py --model_dir THUDM/cogagent-9b-20241220你会看到终端进入交互循环输入任务与截图路径后打印完整 query 和模型回复若回复含 box终端输出Annotated image saved at: ...对应的标注图出现在results目录。Web 界面演示python inference/web_demo.py --host 0.0.0.0 --port 7860 --model_dir THUDM/cogagent-9b-20241220 --format_key status_action_op_sensitive --platform Mac --output_dir ./results你会看到 7860 端口打开 Gradio 页面左侧上传截图、输入任务模型流式返回文本右侧同步展示红框标注图页面还提供回退上一轮、清空历史、中途停止按钮。远程服务加本地自动操作在 GPU 服务器拉 OpenAI 接口格式的服务端再在本地跑客户端python app/openai_demo.py --model_path THUDM/cogagent-9b-20241220 --host 0.0.0.0 --port 7870python app/client.py --api_key EMPTY --base_url http://127.0.0.1:7870/v1 --client_name 127.0.0.1 --client_port 7860 --model CogAgent你会看到客户端 APP 接管电脑它持续截图发给服务端按模型返回的坐标执行点击、输入等动作运行中可随时按 stop 强制停止。此示例仅供学术参考详见 app/README.md。常用参数对照参数作用建议值--model_dir模型本地路径或仓库标识THUDM/cogagent-9b-20241220--platform截图所属平台决定操作坐标空间Mac / WIN / Mobile--format_key模型返回的字段格式status_action_op_sensitive--max_length单轮最大生成长度4096--output_image_path标注图保存目录web_demo 用--output_dir./resultsformat_key可选值对应五种格式action_op_sensitive、status_plan_action_op、status_action_op_sensitive、status_action_op、action_op是否返回敏感操作标记、Status、Plan 字段由它决定。避坑清单现象模型加载阶段 OOM 或无法启动。 原因BF16 推理至少需要 29GB 显存官方明确不建议 INT4 量化性能损失大。 解法换用 30GB 显存的 GPU显存不足时在 inference/cli_demo.py 中取消注释 INT8 量化行约 15GB 显存。现象模型返回的 bbox 位置明显偏移。 原因截图系统与--platform参数不匹配例如把 Mac 截图配了 WIN。 解法截图与平台参数保持同源未列出的系统可用 Mobile/WIN/Mac 近似但官方不保证效果。现象多轮执行时模型忘记之前步骤或输出字段缺失。 原因模型不支持连续对话且要求 Task、History steps、Platform、format 严格按顺序拼接。 解法即使无历史也要保留History steps:字段输出是 STR 格式不支持 JSON 解析按官方字段名提取。现象只发文字任务得不到有效 GUI 结果。 原因该模型必须传入图片纯文字对话无法完成 GUI Agent 任务。 解法每个任务都配一张当前屏幕截图。现象客户端 APP 连不上服务端或无法执行操作。 原因两端未打通网络需内网穿透或运行端缺少截图、录屏、模拟键鼠的权限macOS 需在系统设置中授予终端或 IDE。 解法先确认base_url http://127.0.0.1:7870/v1可达再补齐权限客户端建议用 venv 创建环境避免 conda 导致 tkinter 异常。还能做什么仓库自带 finetune/ 微调入口SFT 需要 8 张 A100每张至少 60GB 显存LoRA 单卡即可但要求一张 A100 至少 70GB 显存配置示例在 finetune/configs/lora.yaml完整动作空间定义见 Action_space.md。模型下载地址、在线体验入口和技术文档汇总在 README_zh.md 的模型资源表中问题与需求可在仓库 Issues 中提交。CogAgent 把 GUI 自动化简化为一张截图加一句指令你只需解析 Grounded Operation 接入自己的执行链路。下一步跑通「先跑起来」一节中的启动命令用自己的一张桌面截图验证能否输出带 bbox 的结果。【免费下载链接】CogAgentAn open-sourced end-to-end VLM-based GUI Agent项目地址: https://gitcode.com/GitHub_Trending/co/CogAgent创作声明:本文部分内容由AI辅助生成(AIGC),仅供参考

读完文章,也想定制专属网站?

尧图设计师 24 小时内与您沟通定制方案

免费获取报价 →
↑