Lightlines
By nvidia

tao-train-centerpose

nvidia1.6k installs

Train CenterPose to estimate object poses.

What it does

Trains, evaluates, exports, and runs CenterPose models for object keypoints and 6-DoF pose estimation.

When to use it

Use for CenterPose training, evaluation, export, or inference requests involving object keypoints or 6-DoF poses.

How to use it

Give it a CenterPose task and the applicable datasets or model files; it produces training results, evaluations, predictions, or exported models.

What you provide

  • Your files

Uses

Good to know

  • a credential is hard-coded: encryption_key

Latest review

What you need · 5

Requires Docker for the container-based workflow.

Requires NVIDIA Container Toolkit alongside Docker.

Requires at least one GPU with 16GB or more VRAM per GPU; two GPUs are recommended.


Access · 1

This skill

wandb

Write

Logs training runs to wandb