tao-train-metric-learning-recognition

Train, evaluate, export, and infer TAO metric-learning recognition models.

83|20|Updated Apr 16, 2026
One-click install
npx skills add https://github.com/NVIDIA-TAO/tao-skill-bank --skill tao-train-metric-learning-recognition
Or copy as Structured Prompt for Agent
Please help me install this Agent Skill.
Skill: tao-train-metric-learning-recognition
Source: https://github.com/NVIDIA-TAO/tao-skill-bank/tree/main/skills/models/tao-train-metric-learning-recognition
Command: npx skills add https://github.com/NVIDIA-TAO/tao-skill-bank --skill tao-train-metric-learning-recognition

SYSTEM DOCUMENTATION & REQUIREMENTS

💡 This Skill requires docker, nvidia-container-toolkit, and includes scripts (resource) and references (resource) and assets (resource) components.

What problem does it solve?

This Skill simplifies the process of training, evaluating, exporting, and running inference on TAO metric-learning recognition models, making it easier to deploy these models for various applications.

Core Features & Use Cases

  • Model Training: Facilitates training TAO metric-learning recognition models with customizable parameters and settings.
  • Model Evaluation: Provides a benchmark for evaluating model performance with detailed metrics.
  • Model Export: Allows exporting trained models for inference or further processing.
  • Model Inference: Supports running inference on trained models with various input formats.
  • Use Case: A developer could use this Skill to train a model for recognizing fine-grained visual recognition tasks like retail product recognition and then export the model for deployment in a product.

Quick Start

Run the tao-train-metric-learning-recognition skill to train a model using the train action with the specified dataset.

Frequently Asked Questions about tao-train-metric-learning-recognition

High-intent search queries and answers about installing and using this skill.

FAQPage Schema
How do I train metric learning models for fine-grained visual recognition?

Train metric learning models for fine-grained visual recognition by running the skill with your specified dataset. It manages the training pipeline and outputs model checkpoints alongside performance metrics.

What is metric learning used for in machine learning workflows?

Metric learning in machine learning workflows is used for fine-grained visual recognition tasks like retail product recognition. It trains models to differentiate highly similar visual classes by learning distance metrics.

Do I need Docker and nvidia-container-toolkit to run TAO metric learning recognition?

Yes, Docker and nvidia-container-toolkit are required to run TAO metric learning recognition. These dependencies execute the Python libraries and manage the GPU environment for model training and inference.

How do I evaluate and export trained metric learning recognition models?

Evaluate and export trained metric learning recognition models by using the evaluate and export actions. Evaluation provides detailed performance benchmark metrics, while export prepares the model for inference.

Can I run inference on trained TAO metric learning models with various input formats?

Yes, you can run inference on trained TAO metric learning models with various input formats. The inference action processes these inputs using the exported model to generate recognition predictions for your visual data.