数据与分析 ✓ NVIDIA · 官方 weakly-supervised-segmentationmask-predictionviT-maetao-toolkitautomlcoco-annotationsmulti-gpu

TAO MAL 弱监督分割

用点或框标注训练模型,生成弱监督语义分割掩码。

FollowSkills 评估 · FSRS-2.0
不推荐
50/ 100 五分制 2.5 / 5
信任安全14 / 25 · 2.8/5

文档限定为 Read/Bash,并明确需要 Docker、NVIDIA Container Toolkit、GPU、数据源和检查点映射;但 Bash、容器、数据上传和默认启用 Weights & Biases 可能产生外部影响,未说明用户确认、敏感数据处理、遥测披露或回滚策略,因此扣分。NVIDIA 官方组织来源可支持归属,但不能替代安全证据。

可靠稳定7 / 20 · 1.8/5

SKILL.md、skill_info.yaml、模板和 schema 对 train/evaluate/inference、数据路径、检查点和 OOM 有较一致说明,并声明确定性 CUDA 设置;但没有实际执行日志、覆盖关键路径的测试或详细异常输入反馈,且模板/schema 内容较冗余,故按静态上限保守扣分。

适用触发11 / 15 · 3.7/5

名称、触发短语、训练/评估/推理场景、COCO 数据要求、GPU/显存要求和非兼容 CSV 说明较清楚;但未明确更多不适用范围、中文交互支持或中国大陆网络可达性,且依赖 Docker、NVIDIA GPU 和可能的 W&B,故扣分。

规范维护8 / 15 · 2.7/5

包含 Apache-2.0 许可证、版本 0.1.0、作者、标签、schema/template 组织和错误模式;但缺少 Instructions、Examples、FAQ、变更记录、明确维护责任/更新流程,作者格式也不完整,benchmark 记录了这些结构性问题,因此扣分。

有效结果6 / 15 · 2.0/5

核心目标、输入格式、spec_overrides、训练/评估/推理输出和检查点选择规则基本可直接指导工作;但没有可验证的代表性产物,效果依赖正确数据、预训练权重和高端 GPU,benchmark 仅含一个任务且静态审查不能确认实际结果,因此保守给分。

证据核验4 / 10 · 2.0/5

存在 schema 元数据、评估任务、benchmark 报告和静态验证发现,提供了一定可追溯性;但只有一个正向任务、无负向任务、无提交测试套件或 CI 覆盖,benchmark 结果也不能在本次静态审查中独立复现,因此扣分。

证据充分度: 评估于 2026年7月20日 审查版本 55f18499943e
上游仓库在本次评估后已有新提交;当前评分仍对应所示审查版本,可能尚未覆盖最新改动。
使用前请注意
  • 默认 W&B enable=true,文档未说明上传哪些数据或如何关闭遥测;处理敏感数据前应确认网络和遥测策略。
  • 需要 COCO 风格 annotations.json、匹配的 image file_name、预训练权重和至少 24GB GPU;CSV/普通归档数据不能直接使用。
  • AutoML 路由依赖打包 schema/template 和外部 tao-skill-bank 能力,缺失时只能回退直接训练;本次未执行验证。
  • 文档缺少明确的失败诊断、示例、FAQ、变更记录和中国大陆网络/中文支持说明。
查看完整评分方法 →

这个 Skill 能做什么,适合哪些场景?

该技能面向 NVIDIA TAO Toolkit 的 MAL(Mask Auto-Label)模型,用最少的点或框标注生成分割掩码。模型使用 ViT-MAE 主干,支持训练、评估和推理,并以 mIoU 作为监测指标。训练默认启用 AutoML;也可以针对单次运行关闭 AutoML。运行需要 Docker、NVIDIA Container Toolkit,以及至少一块 GPU。

读取 TAO 的模型配置、打包的训练 schema 和模板,构造训练、评估或推理所需的数据源路径,并运行相应的 TAO 工作流。它要求使用包含 images.tar.gz 和 annotations.json 的数据源,且 COCO 标注中的 file_name 必须与准备后的图像路径匹配。训练可通过 TAO AutoML 路由执行,也可进行直接训练;评估和推理使用指定或解析出的检查点,并生成模型结果或 MAL 推理 JSON。

  1. 需要用点标注或框标注训练弱监督分割模型的视觉团队。
  2. 希望使用 ViT-MAE 主干微调 MAL 模型的 TAO 用户。
  3. 需要在 COCO 格式数据集上训练并以 mIoU 监控效果的工程师。
  4. 需要对训练或 AutoML 检查点执行评估和推理的团队。
  5. 需要在多 GPU 或多节点环境中运行 Lightning 管理训练的用户。

这个 Skill 有哪些优点和局限?

优点
  • 覆盖 MAL 模型的训练、评估和推理流程。
  • 明确规定了 COCO 标注、数据路径和检查点解析要求。
  • 支持 AutoML、单 GPU、多 GPU 和多节点训练。
  • 提供常见显存不足和 crop_size 配置错误的处理建议。
局限
  • 需要 Docker、NVIDIA Container Toolkit 和至少一块 NVIDIA GPU;推荐每块 GPU 具备 24GB 以上显存。
  • 文档要求训练数据采用特定的压缩图像与 COCO JSON 结构,未转换的数据不能直接使用。
  • 源材料没有提供独立测试套件、示例数据集或实际性能结果。
  • README 的仓库级元数据标为 NOASSERTION;技能自身的 SKILL.md 标注为 Apache-2.0,仓库 README 声明整体采用 Apache 2.0 与 CC BY 4.0 双许可证。

如何安装这个 Skill?

使用 NVIDIA 技能仓库提供的 skills CLI 安装:npx skills add nvidia/skills --skill tao-train-mask-auto-label --yes。README 说明 CLI 会处理安装目标;未提供更具体的本地目录要求。

如何使用这个 Skill?

在已加载技能的 Agent 中提出“train MAL”“Mask Auto-Label”或“weakly-supervised segmentation”等请求。准备 train、eval 或 inference 数据源时,提供 images.tar.gz 和 annotations.json,并在每个动作的 spec_overrides 中填写对应路径。训练前设置 train.pretrained_model_path 指向 ViT-MAE 权重;如需关闭 AutoML,明确说明“disable AutoML”或“plain training”。

常见问题

必须使用完整人工分割标注吗?
不需要。技能面向点或框等最少标注的弱监督分割,但数据仍需提供符合要求的 COCO-style annotations.json。
能否不使用 AutoML?
可以。对单次运行设置 automl_policy: off,或明确提出关闭 AutoML;此时使用直接模型训练。
显存不足时怎么办?
先将 dataset.crop_size 从 512 降到 384 或 256,也可以使用更小的 ViT-MAE 变体。不要把 crop_size 放在 model.crop_size 下。
是否支持多 GPU?
支持。默认使用 1 个 GPU,可通过 train.num_gpus、train.gpu_ids 和 train.num_nodes 配置;技能说明采用 Lightning 管理的分布式训练。

同仓库的其他 Skills

均来自 NVIDIA/skills

数据与分析 ✓ NVIDIA · 官方

Mask Grounding DINO 训练与部署技能

用文本提示训练、评估并部署开放词汇实例分割模型。

数据与分析 ✓ NVIDIA · 官方

TAO 标准训练流程

为 TAO 模型执行标准的训练、评估与导出流程。

数据与分析 ✓ NVIDIA · 官方

TAO NVPanoptix3D 三维全景重建

训练和运行基于 RGB 图像的三维全景分割与占用补全模型。

数据与分析 ✓ NVIDIA · 官方

TAO 视频动作识别训练

用 NVIDIA TAO 训练、评估、导出并推理 RGB、光流或多流视频动作识别模型。

数据与分析 ✓ NVIDIA · 官方

TAO Deformable DETR 目标检测

帮助 AI 代理训练、评估、导出、量化并部署轻量级 Deformable DETR 目标检测模型。

数据与分析 ✓ NVIDIA · 官方

TAO DINO 目标检测

指导使用 NVIDIA TAO 训练、评估和部署 DINO 目标检测器。

数据与分析 ✓ NVIDIA · 官方

TAO CenterPose 姿态估计技能

帮助代理训练、评估、导出和部署用于6自由度物体姿态估计的CenterPose模型。

数据与分析 ✓ NVIDIA · 官方

TAO 人员重识别训练技能

训练和部署用于跨摄像头人员匹配的 TAO ReID 模型。

开发与工程 ✓ NVIDIA · 官方

TAO 技能库能力发现

通过打包的清单快速查询 TAO Skill Bank 支持的工作流、模型、平台和 AutoML 能力。

数据与分析 ✓ NVIDIA · 官方

TAO 光学缺陷检测训练

用孪生网络训练和部署制造业缺陷检测模型。

数据与分析 ✓ NVIDIA · 官方

TAO RT-DETR 目标检测

指导使用 NVIDIA TAO 训练并部署实时 RT-DETR 检测模型。

开发与工程 ✓ NVIDIA · 官方

TAO Sparse4D 训练技能

帮助代理训练、评估、导出、量化和推理多摄像头时序3D检测跟踪模型。

数据与分析 ✓ NVIDIA · 官方

TAO BEVFusion 训练技能

用 TAO 训练、评估和推理融合激光雷达与摄像头数据的 BEV 三维检测模型。

自动化与运维 ✓ NVIDIA · 官方

TAO SLURM 集群运行

通过 SSH、SLURM 和 Pyxis/Enroot 在远程 GPU 集群上运行 TAO 作业。

自动化与运维 ✓ NVIDIA · 官方

TAO Kubernetes 作业运行

通过 Kubernetes 提交和监控带 NVIDIA GPU 的 TAO 训练作业。

数据与分析 ✓ NVIDIA · 官方

TAO OneFormer 通用分割技能

用单一任务条件化模型训练并部署全景、实例和语义图像分割。

数据与分析 ✓ NVIDIA · 官方

TAO SegFormer 语义分割技能

为 TAO SegFormer 提供语义分割训练、评估与部署指导。

自动化与运维 ✓ NVIDIA · 官方

TAO Brev GPU 运行技能

在 Brev GPU 实例上运行和管理 TAO 训练、评估与推理任务。

自动化与运维 ✓ NVIDIA · 官方

TAO 本地 Docker 运行

帮助 AI 代理在本机或远程单节点 Docker GPU 主机上运行 TAO 作业。

数据与分析 ✓ NVIDIA · 官方

TAO MAE 视觉预训练技能

指导 NVIDIA TAO 使用掩码自编码器进行视觉预训练与微调。

相关 Skills