tao-train-reid

Train, evaluate, export, and run inference for TAO person re-identification models.

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

SYSTEM DOCUMENTATION & REQUIREMENTS

💡 This Skill includes scripts (resource) and references (resource) components.

What problem does it solve?

This Skill provides a comprehensive toolkit for training, evaluating, exporting, and running inference on TAO person re-identification models, enabling cross-camera person matching and efficient video analytics.

Core Features & Use Cases

  • Training: Leverage metric learning to train discriminative embeddings for person re-identification across different camera views.
  • Evaluation: Assess the performance of trained models using retrieval-based metrics like CMC and mAP.
  • Export: Generate ONNX models for deployment and inference.
  • Inference: Perform person re-identification on new datasets with trained models.
  • Use Case: Imagine you are developing a video analytics system that requires accurate person re-identification across multiple camera feeds. Use this Skill to train and evaluate your models efficiently.

Quick Start

Run the 'train' action to start training a new ReID model using the provided datasets and configurations.

Frequently Asked Questions about tao-train-reid

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

FAQPage Schema
How do I train a person re-identification model for cross-camera video analytics?

To train a person re-identification model, you use metric learning to build discriminative embeddings for matching individuals across different camera views. This Skill supports training with AutoML and manual model tuning to optimize retrieval metrics.

What metrics are used to evaluate person re-identification models?

Person re-identification models are evaluated using retrieval-based metrics like CMC and mAP. These metrics measure the accuracy of matching individuals across different camera feeds during the evaluation phase.

Do I need Docker to run TAO person re-identification tasks?

Yes, running TAO person re-identification tasks requires Docker and the nvidia-container-toolkit to be installed in your environment. These dependencies are necessary to execute the training, evaluation, and inference workflows.

Can I export trained re-identification models to ONNX for deployment?

Yes, you can export trained re-identification models to the ONNX format. This allows you to generate deployment-ready models for running inference on new person re-identification datasets.

How does metric learning work for cross-camera person matching?

Metric learning for cross-camera person matching trains models to generate discriminative embeddings that cluster images of the same person together. It optimizes retrieval metrics like CMC and mAP to accurately identify individuals across multiple camera feeds.

Can I automate model tuning for person re-identification training?

Yes, you can automate model tuning for person re-identification training using the built-in AutoML support. This allows you to optimize retrieval metrics like CMC and mAP without fully manual hyperparameter configuration.