度量学习识别(MLRecog)Skill tao-train-metric-learning-recognition

该技能封装 NVIDIA TAO 工具包中的 Metric-Learning Recognition (ml-recog) 模型,用于细粒度视觉识别中的检索式匹配(如零售商品识别)。它支持使用三元组/对比损失训练嵌入模型,并支持评估、推理、导出及 TensorRT 部署。关键词:度量学习、嵌入检索、三元组损失、ml-recog、TAO、视觉识别、AutoML、TensorRT 部署

视觉模型训练 0 次安装 0 次浏览 更新于 9/6/2026

ML Recog(度量学习识别)

独立安装? 如果此会话未由 TAO 技能库插件初始化,请先运行 tao-setup 技能(主机预检、凭据、跨技能发现)。

度量学习识别用于细粒度视觉识别。学习嵌入以进行基于检索的匹配(例如零售商品识别)。使用三元组/对比损失。

Dataclass Schemas

生成的 TAO Core schemas 打包在 schemas/<action>.schema.json 中。AutoML 启用状态在模型层通过 references/skill_info.yaml 中的 automl_enabled 声明。

训练操作策略

该模型在模型层启用了 AutoML。默认使用 automl_policy: on。非训练操作(如 evaluate、inference、export 和部署)保留在此模型技能中。

训练要求

  • 数据集类型: ml_recog
  • 格式: default
  • 监控指标: 验证集 Rank-1 精度

各操作数据集要求

操作 数据
train known_classes 的 train/reference/val
evaluate/inference unknown_classes 的 reference/test

重要参数

  • model.backbone:骨干网络(默认 resnet_50)
  • model.feat_dim:嵌入维度(默认 256)
  • train.batch_size:单 GPU 批次大小(默认 4)
  • dataset.num_instance:每个身份实例数(默认 4)
  • train.optim.trunk.base_lrtrain.optim.embedder.base_lr:学习率
  • dataset.train_dataset:训练图像路径
  • dataset.val_dataset:参考集与查询集字典

部署

  • 参见 references/tao-deploy-metric-learning-recognition.md