BloodAxe / pytorch-toolbelt
PyTorch extensions for fast R&D prototyping and Kaggle farming
☆1,533Updated 3 weeks ago
Alternatives and similar repositories for pytorch-toolbelt:
Users that are interested in pytorch-toolbelt are comparing it to the libraries listed below
- Image Test Time Augmentation with PyTorch!☆993Updated last year
- A learning rate range test implementation in PyTorch☆948Updated 3 months ago
- A PyTorch impl of EfficientDet faithful to the original Google impl w/ ported weights☆1,606Updated 6 months ago
- Code for the Lovász-Softmax loss (CVPR 2018)☆1,387Updated 10 months ago
- High-level batteries-included neural network training library for Pytorch☆398Updated 3 years ago
- Toolbox of models, callbacks, and datasets for AI/ML researchers.☆1,715Updated last month
- Pretrained EfficientNet, EfficientNet-Lite, MixNet, MobileNetV3 / V2, MNASNet A1 and B1, FBNet, Single-Path NAS☆1,569Updated 8 months ago
- Codebase for Image Classification Research, written in PyTorch.☆2,149Updated 11 months ago
- Accelerated deep learning R&D☆3,328Updated 11 months ago
- PyTorch Boilerplate For Research☆609Updated 3 years ago
- Official Implementation of 'Fast AutoAugment' in PyTorch.☆1,602Updated 3 years ago
- YACS -- Yet Another Configuration System☆1,300Updated 2 years ago
- Over9000 optimizer☆427Updated 2 years ago
- A (PyTorch) imbalanced dataset sampler for oversampling low frequent classes and undersampling high frequent ones.☆2,286Updated last month
- A collection of loss functions for medical image segmentation☆3,876Updated last year
- This repository reproduces the results of the paper: "Fixing the train-test resolution discrepancy" https://arxiv.org/abs/1906.06423☆1,038Updated 3 years ago
- VISSL is FAIR's library of extensible, modular and scalable components for SOTA Self-Supervised Learning with images.☆3,269Updated last year
- pip install antialiased-cnns to improve stability and accuracy☆1,662Updated 10 months ago
- Gradually-Warmup Learning Rate Scheduler for PyTorch☆987Updated 4 months ago
- Simple PyTorch implementations of U-Net/FullyConvNet (FCN) for image segmentation☆852Updated last year
- Augmentations usage examples for albumentations library☆430Updated 3 weeks ago
- Model summary in PyTorch similar to `model.summary()` in Keras☆4,033Updated last year
- Provides everything needed for high performance data loading and augmentation in pytorch.☆319Updated last year
- Machine learning metrics for distributed, scalable PyTorch applications.☆2,215Updated this week
- Implementation of various self-attention mechanisms focused on computer vision. Ongoing repository.☆1,194Updated 3 years ago
- Gradient based receptive field estimation for Convolutional Neural Networks☆338Updated 5 years ago
- UNet model with VGG11 encoder pre-trained on Kaggle Carvana dataset☆1,055Updated 2 years ago
- Official Pytorch implementation of CutMix regularizer☆1,227Updated 4 years ago
- 🛠 Toolbox to extend PyTorch functionalities☆419Updated 9 months ago
- ResNeSt: Split-Attention Networks☆3,249Updated 2 years ago