tao-train-pose-classification

Automate ST-GCN pose classification training, evaluation, and export with PyTorch.

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

SYSTEM DOCUMENTATION & REQUIREMENTS

💡 This Skill requires torch, torchvision, torch.nn, and includes scripts (resource) and references (resource) and assets (resource) components.

What problem does it solve?

This Skill streamlines the training and evaluation processes for pose classification using ST-GCN (Spatial Temporal Graph Convolutional Network) models, saving you time and reducing complexity.

Core Features & Use Cases

  • Automated Pose Classification Training: Supports automatic model training with pre-trained checkpoints.
  • Evaluate Models: Conduct performance assessments of the models across multiple metrics.
  • Use Case: For example, train and evaluate an ST-GCN model to classify human poses in action sequences. You can initiate a skill request such as 'train pose classification model for skeleton actions using the dataset'.

Quick Start

To initiate a new pose classification model training task, use the command 'train pose classification with model "pose_classification_model_name" on dataset "dataset_path" with checkpoint "checkpoint_path"'. To evaluate the trained model, use 'evaluate pose classification model "pose_classification_model_name" on dataset "dataset_path" with checkpoint "checkpoint_path"'.

Frequently Asked Questions about tao-train-pose-classification

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

FAQPage Schema
How do I train an ST-GCN model for skeletal action recognition?

Train ST-GCN models for pose classification by initiating a skill request with your dataset path and pre-trained checkpoint. This automates the training workflow, including hyperparameter tuning, to classify human poses in action sequences.

What is the best way to evaluate pose classification models across multiple metrics?

Evaluate pose classification models by running the evaluation command with your trained checkpoint and dataset path. This conducts automated performance benchmarking across multiple metrics to assess ST-GCN model accuracy.

Can I use PyTorch to automate pose recognition workflows with pre-trained checkpoints?

Yes, you can automate pose recognition workflows using PyTorch. The skill utilizes deep learning libraries like torch and torchvision to support automatic model training and evaluation with pre-trained checkpoints.

How does automated pose classification training handle hyperparameter tuning?

Automated pose classification training handles hyperparameter tuning internally during the model training process. It uses your specified pre-trained checkpoint and dataset to optimize the ST-GCN model for skeletal action recognition.

Do I need specific deep learning libraries to run ST-GCN pose classification training?

Yes, you need PyTorch deep learning libraries including torch, torchvision, and torch.nn to run ST-GCN pose classification training and evaluation workflows for skeletal action recognition.