tao-train-pointpillars

Trains, evaluates, exports, and prunes PyTorch-based PointPillars 3D detection models from LiDAR point clouds.

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

SYSTEM DOCUMENTATION & REQUIREMENTS

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

What problem does it solve?

This Skill streamlines the training and evaluation of 3D object detection models from LiDAR point clouds, reducing the complexity and time required for autonomous driving and robotics workflows.

Core Features & Use Cases

  • Model Training: Train PointPillars models from scratch or resume training from a checkpoint.
  • Evaluation: Evaluate model performance using a specified checkpoint.
  • Export: Export models to ONNX format for inference.
  • Pruning: Prune models to reduce size and improve inference speed.
  • Retraining: Retrain pruned models for improved performance.
  • Use Case: Imagine you need to train a 3D object detection model for autonomous driving. This Skill can handle the entire workflow, from data conversion to model evaluation and export.

Quick Start

Use the tao-train-pointpillars skill to train a PointPillars model from scratch. Run the following command: tao-train-pointpillars train -n <num_epochs> -d <data_path> -i <info_path> -c <checkpoint_path>

Frequently Asked Questions about tao-train-pointpillars

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

FAQPage Schema
How do I train a 3D object detection model from LiDAR point clouds for autonomous driving?

To train a 3D object detection model from LiDAR point clouds, you can use this Skill to train PointPillars models from scratch or resume from a checkpoint. It handles point cloud data preprocessing, model training, and evaluation for autonomous driving workflows.

Can I export a trained PointPillars model to ONNX format for inference?

Yes, you can export a trained PointPillars model to ONNX format for inference. This Skill supports exporting models directly to ONNX, enabling deployment of your 3D object detection models in production environments.

How do I prune a PyTorch model to improve inference speed for 3D object detection?

You can prune a PyTorch model to reduce its size and improve inference speed using the built-in pruning feature. After pruning your 3D object detection model, you can also retrain it to recover performance.

Do I need PyTorch and torchvision to train PointPillars models?

Yes, you need PyTorch, torchvision, and torchgeometry installed to train PointPillars models. These dependencies are required for handling point cloud data preprocessing, model training, and evaluation within the workflow.

What is the best way to evaluate a 3D object detection model using a checkpoint?

The best way to evaluate a 3D object detection model is by running the evaluation command with a specified checkpoint. This Skill allows you to assess model performance directly from your saved checkpoints during development.

Can I use this workflow for robotics applications or is it limited to autonomous driving?

You can use this workflow for both robotics applications and autonomous driving. The Skill streamlines training and evaluation of 3D object detection models from LiDAR point clouds, applying to any robotics workflows requiring spatial perception.