qubvel / ttachLinks
Image Test Time Augmentation with PyTorch!
☆1,026Updated 2 years ago
Alternatives and similar repositories for ttach
Users that are interested in ttach are comparing it to the libraries listed below
Sorting:
- PyTorch extensions for fast R&D prototyping and Kaggle farming☆1,562Updated 2 months ago
- Augmentations usage examples for albumentations library☆443Updated 2 months ago
- Set of methods to ensemble boxes from different object detection models, including implementation of "Weighted boxes fusion (WBF)" method…☆1,808Updated 2 years ago
- Official Pytorch implementation of CutMix regularizer☆1,246Updated 5 years ago
- A PyTorch impl of EfficientDet faithful to the original Google impl w/ ported weights☆1,649Updated last year
- Code for the Lovász-Softmax loss (CVPR 2018)☆1,408Updated last year
- This repository reproduces the results of the paper: "Fixing the train-test resolution discrepancy" https://arxiv.org/abs/1906.06423☆1,044Updated 4 years ago
- This Repository is implementation of majority of Semantic Segmentation Loss Functions☆583Updated 2 years ago
- Copy-paste augmentation for segmentation and detection tasks☆568Updated 4 years ago
- 🛠 Toolbox to extend PyTorch functionalities☆420Updated last year
- Implementation of various self-attention mechanisms focused on computer vision. Ongoing repository.☆1,212Updated 4 years ago
- The official Pytorch implementation of OCNet, OCRNet, and SegFix.☆1,227Updated last year
- Gradually-Warmup Learning Rate Scheduler for PyTorch☆992Updated last year
- PyTorch implementations of recent Computer Vision tricks (ReXNet, RepVGG, Unet3p, YOLOv4, CIoU loss, AdaBelief, PolyLoss, MobileOne). Oth…☆328Updated last week
- PyTorch Implementation of Focal Loss and Lovasz-Softmax Loss☆337Updated 4 years ago
- Semi-Supervised Learning, Object Detection, ICCV2021☆919Updated last year
- PyTorch implementation of EfficientNetV2 family☆482Updated 3 years ago
- Lightweight models for real-time semantic segmentationon PyTorch (include SQNet, LinkNet, SegNet, UNet, ENet, ERFNet, EDANet, ESPNet, ESP…☆1,000Updated last year
- This is a list of awesome methods about data augmentation.☆795Updated 4 years ago
- This is a PyTorch re-implementation of Axial-DeepLab (ECCV 2020 Spotlight)☆460Updated 4 years ago
- [ICCV2021] Official PyTorch implementation of Segmenter: Transformer for Semantic Segmentation☆902Updated last year
- This is an official implementation for "Swin Transformer: Hierarchical Vision Transformer using Shifted Windows" on Semantic Segmentation…☆1,252Updated 3 years ago
- label-smooth, amsoftmax, partial-fc, focal-loss, triplet-loss, lovasz-softmax. Maybe useful☆2,260Updated last year
- Compute CNN receptive field size in pytorch in one line☆366Updated last year
- Per-Pixel Classification is Not All You Need for Semantic Segmentation (NeurIPS 2021, spotlight)☆1,445Updated 3 years ago
- [CVPR 2021 & IJCV 2024] Rethinking Semantic Segmentation from a Sequence-to-Sequence Perspective with Transformers☆1,104Updated last year
- A PyTorch Implementation of Focal Loss.☆992Updated 6 years ago
- Pretrained EfficientNet, EfficientNet-Lite, MixNet, MobileNetV3 / V2, MNASNet A1 and B1, FBNet, Single-Path NAS☆1,584Updated last year
- Official Implementation of 'Fast AutoAugment' in PyTorch.☆1,613Updated 4 years ago
- Code for Noisy Student Training. https://arxiv.org/abs/1911.04252☆765Updated 4 years ago