SuperAPI AI 工具集
合作ENAI 行业资讯收藏
Skill 广场tao-generate-od-defects

tao-generate-od-defects

数据与分析

查看项目

更新于 10-21 12:49

技能介绍

用途

该技能在固定公共容器中调用上游生成器与伪标注器，执行 AnomalyGenNext 1.1 推理，从原生测试用例或已准备好的生成计划生成缺陷数据，并发布经过校验的细粒度与二值 COCO 标签。它不负责准备检测器缺口、运行 AMP、训练 AnomalyGenNext，也不将输出追加到检测器训练集。

主要能力

  • 支持两种输入方式：使用 --inputs-dir 指向 tao-prepare-anomalygennext-inputs 的完成输出；或使用 --input-data-path、--checkpoint 与匹配的 --recipe 提供原生 AnomalyGenNext 测试用例。
  • 需要提供 Cosmos3-Nano 基础检查点、完整的 AnomalyGenNext 检查点根目录以及一个或多个 GPU。
  • 要求每个测试用例图像与对齐掩码均存在，每个异常类型都出现在配方中，且请求的行数与测试用例匹配。
  • 当存在准备好的完整性清单时，生成前会校验每条记录的哈希。
  • 默认启用护栏，保留原生调用且不带护栏标志；仅在显式禁用时，包装器才追加 --no-guardrail。--no-guardrail 会同时关闭文本筛查、图像内容安全筛查与人脸模糊。
  • 检查点根目录会挂载到镜像的规范路径 /workspace/paidf-anomalygen/checkpoints 上，并在 GPU 工作前完成校验。
  • Nano 分词器、Edge 处理器与 DINOv2 资产始终必需；启用护栏的运行额外需要 Qwen3Guard 与 Cosmos Guardrail 仓库。
  • 校验完成后生成过程强制离线。所选平台必须遵守声明的 checkpoint_root.container_path；生成叶会拒绝不同的挂载点。基础检查点参数保持独立：它是包含 checkpoint.json 与 model/ 检查点目录的父级。

使用方式

通过 tao-launch-workflow 调用，先审查确切的镜像、挂载、GPU 形态、运行时与输出目录，然后提交 generate 动作。示例命令：

scripts/generate_od_defects.py \
  --inputs-dir /results/prepared_inputs \
  --base-checkpoint /models/Cosmos3-Nano \
  --checkpoint-root /workspace/paidf-anomalygen/checkpoints \
  --output-dir /temporary/generation \
  --published-root /persistent/generation \
  --no-guardrail \
  --num-gpus 1

当执行使用临时存储时，--published-root 记录将包含保存结果的持久位置。

在适配输入或完成行为时，请阅读 references/execution-contract.md；在提交前请阅读 references/container-runtime.md。

完成与输出

对每个数据集，generated + guardrail_blocked 必须等于请求的行数。使用 --no-guardrail 时，generated 必须等于请求的行数，且 guardrail_blocked 保持为零。校验摘要会记录所选的护栏模式。伪标签数量必须与生成的图像匹配，每张图像都需要一个标注，类别必须声明为异常类型，且每个 COCO 边界框必须为正且位于其图像内。该动作输出：

  • DATASET/raw/
  • DATASET/pseudo_labels/coco_annotations.json
  • pseudo_labels/coco_annotations.json
  • pseudo_labels/coco_annotations_od_defect.json
  • validation_summary.json
  • status.json

原生 COCO 保留 TEXTURE+DEFECT 类别。二值文件将所有标注映射为 defect。只有调用方应用才可将这些输出纳入训练；该叶始终报告 training_pool_mutated=false。

适用场景

适用于需要从原生测试用例或已准备好的生成计划运行 AnomalyGenNext 1.1 推理，并生成经过校验的细粒度与二值 COCO 缺陷标签的场景。

限制

该技能不准备检测器缺口、不运行 AMP、不训练 AnomalyGenNext，也不将输出追加到检测器训练集。输入必须满足图像、掩码、异常类型与行数等一致性要求。所选平台必须遵守声明的检查点挂载路径。启用护栏时需要额外的 Qwen3Guard 与 Cosmos Guardrail 仓库。