tao-run-deft-aoi
nvidia
Improve PCB inspection models toward your target.
Optical inspection
GitHub publisher217 Skills358.1k installs
Organization · NVIDIA Corporation · nvidia.com · On GitHub since 2012
tao-run-deft-aoi
nvidia
Improve PCB inspection models toward your target.
Optical inspection
tao-train-centerpose
nvidia
Train CenterPose to estimate object poses.
Object pose estimation
tao-train-segformer
nvidia
Train SegFormer for image segmentation.
Semantic segmentation
tao-train-pointpillars
nvidia
Train 3D object detectors on LiDAR scans.
3D object detection
tao-train-mask-auto-encoder
nvidia
Train vision models by rebuilding hidden patches.
Computer vision
tao-train-mask-auto-label
nvidia
Turn image points and boxes into precise masks.
Image segmentation
tao-train-ocdnet
nvidia
A trained detector for text in natural images.
Scene text detection
tao-train-action-recognition
nvidia
Train TAO models to recognize actions in video.
Video classification
tao-train-bevfusion
nvidia
Train 3D detectors with LiDAR and camera fusion.
3D perception
tao-train-mask-grounding-dino
nvidia
Train models that outline objects from text prompts.
Computer vision
tao-train-oneformer
nvidia
Train one model for three image segmentation tasks.
Image segmentation
tao-validate-dataset-format
nvidia
Check your DAFT datasets before training.
Dataset validation
tilegym-monkey-patch-kernels-to-transformers
nvidia
Integrate and validate faster kernels for your language model.
GPU kernel optimization
tao-train-deformable-detr
nvidia
Train Deformable DETR to find objects in images.
Object detection
tao-train-nvdinov2
nvidia
Train NVDINOv2 on images without labels.
Computer vision
tao-train-single-step
nvidia
Fine-tune a TAO model on your labeled dataset.
Model training
tao-mine-aoi-images
nvidia
Find fresh training images like your weak examples.
Training data selection
tao-run-on-slurm
nvidia
Your TAO jobs, running on a remote GPU cluster.
Machine learning operations
tao-train-sparse4d
nvidia
Train 3D detectors from multi-camera images.
Computer vision
tao-train-foundation-stereo
nvidia
Turn stereo image pairs into depth estimates.
Computer vision
tao-train-fast-foundation-stereo
nvidia
Real-time depth estimates from stereo images.
Computer vision
tao-route-visual-changenet-samples
nvidia
Know which augmentation modules can handle each weak sample.
Data augmentation
tilegym-improve-cutile-kernel-perf
nvidia
Faster GPU kernels, checked one optimization at a time.
GPU kernel optimization
tilegym-cutile-autotuning
nvidia
Tune CuTile GPU kernels without repeat searches.
GPU kernel optimization