ufoym / imbalanced-dataset-sampler
A (PyTorch) imbalanced dataset sampler for oversampling low frequent classes and undersampling high frequent ones.
☆2,251Updated 2 months ago
Related projects: ⓘ
- label-smooth, amsoftmax, partial-fc, focal-loss, triplet-loss, lovasz-softmax. Maybe useful☆2,159Updated last year
- The easiest way to use deep metric learning in your application. Modular, flexible, and extensible. Written in PyTorch.☆5,949Updated last week
- mixup: Beyond Empirical Risk Minimization☆1,156Updated 2 years ago
- SimCLRv2 - Big Self-Supervised Models are Strong Semi-Supervised Learners☆4,049Updated last year
- PyTorch implementation of "Supervised Contrastive Learning" (and SimCLR incidentally)☆3,056Updated 8 months ago
- PyTorch implementation of SimCLR: A Simple Framework for Contrastive Learning of Visual Representations☆2,216Updated 6 months ago
- A curated list of resources for Learning with Noisy Labels☆2,621Updated 4 months ago
- Model summary in PyTorch similar to `model.summary()` in Keras☆4,001Updated 6 months ago
- PyTorch implementation of Contrastive Learning methods☆1,932Updated 11 months ago
- SAM: Sharpness-Aware Minimization (PyTorch)☆1,742Updated 6 months ago
- Siamese and triplet networks with online pair/triplet mining in PyTorch☆3,088Updated last year
- PyTorch implementation of SwAV https//arxiv.org/abs/2006.09882☆1,989Updated last year
- A learning rate range test implementation in PyTorch☆912Updated 3 weeks ago
- torch-optimizer -- collection of optimizers for Pytorch☆3,012Updated 5 months ago
- VISSL is FAIR's library of extensible, modular and scalable components for SOTA Self-Supervised Learning with images.☆3,249Updated 6 months ago
- Neural network graphs and training metrics for PyTorch, Tensorflow, and Keras.☆1,789Updated 7 months ago
- Usable Implementation of "Bootstrap Your Own Latent" self-supervised learning, from Deepmind, in Pytorch☆1,732Updated 2 months ago
- A collection of AWESOME things about domian adaptation☆5,045Updated 3 weeks ago
- A PyTorch Implementation of Focal Loss.☆951Updated 4 years ago
- PyTorch implementation of MoCo: https://arxiv.org/abs/1911.05722☆4,721Updated last month
- Model interpretability and understanding for PyTorch☆4,810Updated this week
- Gradually-Warmup Learning Rate Scheduler for PyTorch☆971Updated 3 years ago
- A PyTorch implementation for exploring deep and shallow knowledge distillation (KD) experiments with flexibility☆1,834Updated last year
- Class activation maps for your PyTorch models (CAM, Grad-CAM, Grad-CAM++, Smooth Grad-CAM++, Score-CAM, SS-CAM, IS-CAM, XGrad-CAM, Layer-…☆1,982Updated this week
- A small package to create visualizations of PyTorch execution graphs☆3,168Updated 5 months ago
- Codebase for Image Classification Research, written in PyTorch.☆2,133Updated 5 months ago
- Unofficial implementation of the ImageNet, CIFAR 10 and SVHN Augmentation Policies learned by AutoAugment using pillow☆1,458Updated last year
- A simple way to calibrate your neural network.☆1,081Updated 3 years ago
- 😎 An up-to-date & curated list of awesome semi-supervised learning papers, methods & resources.☆1,770Updated 3 months ago
- OpenMMLab Self-Supervised Learning Toolbox and Benchmark☆3,167Updated last year