qubvel / ttach
Image Test Time Augmentation with PyTorch!
☆978Updated last year
Related projects ⓘ
Alternatives and complementary repositories for ttach
- PyTorch extensions for fast R&D prototyping and Kaggle farming☆1,520Updated last month
- Set of methods to ensemble boxes from different object detection models, including implementation of "Weighted boxes fusion (WBF)" method…☆1,735Updated last year
- PyTorch implementation of EfficientNetV2 family☆460Updated 2 years ago
- Official Pytorch implementation of CutMix regularizer☆1,221Updated 4 years ago
- Augmentations usage examples for albumentations library☆423Updated last week
- This repository reproduces the results of the paper: "Fixing the train-test resolution discrepancy" https://arxiv.org/abs/1906.06423☆1,033Updated 3 years ago
- A PyTorch impl of EfficientDet faithful to the original Google impl w/ ported weights☆1,588Updated 2 months ago
- 🛠 Toolbox to extend PyTorch functionalities☆417Updated 6 months ago
- Copy-paste augmentation for segmentation and detection tasks☆546Updated 3 years ago
- Gradually-Warmup Learning Rate Scheduler for PyTorch☆977Updated last month
- PyTorch implementations of recent Computer Vision tricks (ReXNet, RepVGG, Unet3p, YOLOv4, CIoU loss, AdaBelief, PolyLoss, MobileOne). Oth…☆318Updated this week
- Official Implementation of 'Fast AutoAugment' in PyTorch.☆1,596Updated 3 years ago
- Unofficial PyTorch Reimplementation of RandAugment.☆628Updated last year
- This is a PyTorch re-implementation of Axial-DeepLab (ECCV 2020 Spotlight)☆450Updated 3 years ago
- PyTorch Implementation of Focal Loss and Lovasz-Softmax Loss☆333Updated 2 years ago
- Implementation of DropBlock: A regularization method for convolutional networks in PyTorch.☆588Updated 4 years ago
- Code for the Lovász-Softmax loss (CVPR 2018)☆1,382Updated 7 months ago
- Pretrained EfficientNet, EfficientNet-Lite, MixNet, MobileNetV3 / V2, MNASNet A1 and B1, FBNet, Single-Path NAS☆1,566Updated 5 months ago
- Implementation of the Swin Transformer in PyTorch.☆794Updated 3 years ago
- The official Pytorch implementation of OCNet, OCRNet, and SegFix.☆1,190Updated 3 months ago
- This Repository is implementation of majority of Semantic Segmentation Loss Functions☆555Updated last year
- Official implementation of PVT series☆1,729Updated 2 years ago
- This is an official implementation for "Swin Transformer: Hierarchical Vision Transformer using Shifted Windows" on Semantic Segmentation…☆1,180Updated 2 years ago
- DetectoRS: Detecting Objects with Recursive Feature Pyramid and Switchable Atrous Convolution☆1,137Updated 2 years ago
- Implementation of various self-attention mechanisms focused on computer vision. Ongoing repository.☆1,182Updated 3 years ago
- Semi-Supervised Learning, Object Detection, ICCV2021☆904Updated 5 months ago
- A Simple pytorch implementation of GradCAM and GradCAM++☆359Updated 5 years ago
- label-smooth, amsoftmax, partial-fc, focal-loss, triplet-loss, lovasz-softmax. Maybe useful☆2,183Updated last month
- Official Pytorch Implementation of "TResNet: High-Performance GPU-Dedicated Architecture" (WACV 2021)☆471Updated this week
- NFNets and Adaptive Gradient Clipping for SGD implemented in PyTorch. Find explanation at tourdeml.github.io/blog/☆345Updated 10 months ago