BloodAxe / pytorch-toolbelt
PyTorch extensions for fast R&D prototyping and Kaggle farming
☆1,536Updated last month
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!☆1,009Updated last year
- A collection of loss functions for medical image segmentation☆3,903Updated last year
- Code for the Lovász-Softmax loss (CVPR 2018)☆1,389Updated last year
- A learning rate range test implementation in PyTorch☆954Updated 4 months ago
- Set of methods to ensemble boxes from different object detection models, including implementation of "Weighted boxes fusion (WBF)" method…☆1,760Updated 2 years ago
- Pretrained EfficientNet, EfficientNet-Lite, MixNet, MobileNetV3 / V2, MNASNet A1 and B1, FBNet, Single-Path NAS☆1,569Updated 10 months ago
- Accelerated deep learning R&D☆3,347Updated last year
- PyTorch Boilerplate For Research☆609Updated 3 years ago
- High-level batteries-included neural network training library for Pytorch☆400Updated 3 years ago
- This repository reproduces the results of the paper: "Fixing the train-test resolution discrepancy" https://arxiv.org/abs/1906.06423☆1,041Updated 3 years ago
- A PyTorch impl of EfficientDet faithful to the original Google impl w/ ported weights☆1,620Updated 7 months ago
- Over9000 optimizer☆426Updated 2 years ago
- This Repository is implementation of majority of Semantic Segmentation Loss Functions☆567Updated 2 years ago
- Toolbox of models, callbacks, and datasets for AI/ML researchers.☆1,718Updated 2 weeks ago
- Codebase for Image Classification Research, written in PyTorch.☆2,154Updated last year
- Implementation of various self-attention mechanisms focused on computer vision. Ongoing repository.☆1,201Updated 3 years ago
- YACS -- Yet Another Configuration System☆1,307Updated 3 years ago
- Ranger - a synergistic optimizer using RAdam (Rectified Adam), Gradient Centralization and LookAhead in one codebase☆1,199Updated last year
- VISSL is FAIR's library of extensible, modular and scalable components for SOTA Self-Supervised Learning with images.☆3,279Updated last year
- Official Implementation of 'Fast AutoAugment' in PyTorch.☆1,602Updated 3 years ago
- The official Pytorch implementation of OCNet, OCRNet, and SegFix.☆1,204Updated 9 months ago
- Semantic segmentation models, datasets and losses implemented in PyTorch.☆1,762Updated 2 months ago
- UNet model with VGG11 encoder pre-trained on Kaggle Carvana dataset☆1,058Updated 2 years ago
- pip install antialiased-cnns to improve stability and accuracy☆1,671Updated last year
- Gradually-Warmup Learning Rate Scheduler for PyTorch☆988Updated 6 months ago
- Pytorch Lightning code guideline for conferences☆1,261Updated last year
- Train AI models efficiently on medical images using any framework☆1,874Updated 10 months ago
- scikit-learn cross validators for iterative stratification of multilabel data☆863Updated 6 months ago
- Collection of common code that's shared among different research projects in FAIR computer vision team.☆2,112Updated 5 months ago
- A (PyTorch) imbalanced dataset sampler for oversampling low frequent classes and undersampling high frequent ones.☆2,297Updated 2 weeks ago