ufoym / imbalanced-dataset-samplerLinks
A (PyTorch) imbalanced dataset sampler for oversampling low frequent classes and undersampling high frequent ones.
☆2,326Updated 3 weeks ago
Alternatives and similar repositories for imbalanced-dataset-sampler
Users that are interested in imbalanced-dataset-sampler are comparing it to the libraries listed below
Sorting:
- A PyTorch Implementation of Focal Loss.☆992Updated 6 years ago
- Gradually-Warmup Learning Rate Scheduler for PyTorch☆992Updated last year
- A learning rate range test implementation in PyTorch☆997Updated 7 months ago
- Pytorch implementation of the paper "Class-Balanced Loss Based on Effective Number of Samples"☆805Updated last year
- label-smooth, amsoftmax, partial-fc, focal-loss, triplet-loss, lovasz-softmax. Maybe useful☆2,259Updated last year
- Model summary in PyTorch similar to `model.summary()` in Keras☆4,064Updated last year
- Early stopping for PyTorch☆1,270Updated last year
- mixup: Beyond Empirical Risk Minimization☆1,199Updated 4 years ago
- Official Implementation of 'Fast AutoAugment' in PyTorch.☆1,614Updated 4 years ago
- Siamese and triplet networks with online pair/triplet mining in PyTorch☆3,170Updated 2 years ago
- List of useful data augmentation resources. You will find here some not common techniques, libraries, links to GitHub repos, papers, and …☆1,640Updated last year
- Ranger - a synergistic optimizer using RAdam (Rectified Adam), Gradient Centralization and LookAhead in one codebase☆1,208Updated 2 years ago
- Image Test Time Augmentation with PyTorch!☆1,029Updated 2 years ago
- A curated list of resources for Learning with Noisy Labels☆2,720Updated 8 months ago
- PyTorch extensions for fast R&D prototyping and Kaggle farming☆1,563Updated 3 months ago
- Unofficial implementation of the ImageNet, CIFAR 10 and SVHN Augmentation Policies learned by AutoAugment using pillow☆1,495Updated 2 years ago
- Some custom dataset examples for PyTorch☆874Updated 5 years ago
- A simple way to calibrate your neural network.☆1,171Updated 6 months ago
- Neural network graphs and training metrics for PyTorch, Tensorflow, and Keras.☆1,859Updated last year
- torch-optimizer -- collection of optimizers for Pytorch☆3,162Updated last year
- PyTorch implementation of Contrastive Learning methods☆1,996Updated 2 years ago
- A scalable template for PyTorch projects, with examples in Image Segmentation, Object classification, GANs and Reinforcement Learning.☆913Updated 2 years ago
- Train AI models efficiently on medical images using any framework☆1,877Updated last year
- The easiest way to use deep metric learning in your application. Modular, flexible, and extensible. Written in PyTorch.☆6,300Updated 5 months ago
- Official Pytorch implementation of CutMix regularizer☆1,249Updated 5 years ago
- Pytorch Lightning code guideline for conferences☆1,287Updated 2 years ago
- PyTorch re-implementation of Grad-CAM (+ vanilla/guided backpropagation, deconvnet, and occlusion sensitivity maps)☆803Updated 6 years ago
- SAM: Sharpness-Aware Minimization (PyTorch)☆1,959Updated last year
- PyTorch implementation of "Supervised Contrastive Learning" (and SimCLR incidentally)☆3,404Updated 2 years ago
- PyTorch implementation of SimCLR: A Simple Framework for Contrastive Learning of Visual Representations☆2,483Updated last year