← all repositories

qubvel/ttach

PyTorch library for test-time augmentation applied to computer vision model inference.

1k stars Python Computer VisionML Frameworks
ttach
Velocity · 7d
+0.4
★ / day
Trend
steady
star history

TTAch provides test-time augmentation techniques for deep learning models in computer vision tasks. During inference, it applies augmentations like flips, rotations, and scales to input images, passes them through the model, reverses the transformations, and merges predictions (via mean, max, or gmean) to improve accuracy. It offers specialized wrappers for segmentation, classification, and keypoint detection models, all integrated with PyTorch.

heatdrop uses Google Analytics to see which pages get read — nothing else. Your call. How we handle data.