qubvel/ttach

Health v2 · maintenance only

Adoption not part of the score

1030 stars · 69 forks observed · 2026-08-28

What it is AI-extracted, prompt v1, taxonomy v1, 2026-08-30, confidence not recorded

TTAch is a Python library for image test time augmentation (TTA) with PyTorch. It wraps existing models to apply augmentations like flips, rotations, and scales at inference time, then merges the predictions to improve accuracy.

Use cases

• improve image classification accuracy with test time augmentation
• apply TTA to a segmentation model in PyTorch
• boost keypoint detection predictions with flips and scales
• average model predictions over augmented versions of test images
• wrap a PyTorch model with test-time augmentation transforms

When to choose

• you have a trained PyTorch model for classification, segmentation, or keypoint detection and want a quick accuracy boost at inference
• you want configurable TTA transforms and merge modes without writing the augmentation loop yourself

When to avoid

• you need TTA for frameworks other than PyTorch
• inference latency is critical, since TTA multiplies the number of forward passes
• you need a library under active development with frequent updates

Facets

library · maturity maintenance

machine-learning deep-learning image-processing computer-vision deep-learning computer-vision machine-learning image-processing python test-time-augmentation pytorch tta segmentation classification keypoint-detection inference

Member repositories

For agents

markdown · JSON · MCP: product_card(name="qubvel/ttach")