Stars
Self-supervised Equivariant Attention Mechanism for Weakly Supervised Semantic Segmentation, CVPR 2020 (Oral)
2nd place solution for SIIM-OSIC Melanoma Classification
[ICIP 2021] Puzzle-CAM: Improved localization via matching partial and full features.
The largest collection of PyTorch image encoders / backbones. Including train, eval, inference, export scripts, and pretrained weights -- ResNet, ResNeXT, EfficientNet, NFNet, Vision Transformer (V…
This is to help people get forward signal of their inverse volatility allocation strategy. https://www.portfoliovisualizer.com/ used to provide this for free, but now it requires a subscription.
Gradually-Warmup Learning Rate Scheduler for PyTorch
PyTorch implementation of image classification models for CIFAR-10/CIFAR-100/MNIST/FashionMNIST/Kuzushiji-MNIST/ImageNet
The 4th Place Solution of SIIM-ACR Pneumothorax Segmentation
SIIM-ACR Pneumothorax Segmentation first place solution
Winning algorithms for the SpaceNet 4: Off-Nadir Building Footprint Detection Challenge
https://www.kaggle.com/c/siim-acr-pneumothorax-segmentation/leaderboard
Semantic segmentation models with 500+ pretrained convolutional and transformer-based backbones.
XLNet: Generalized Autoregressive Pretraining for Language Understanding
Deep learning based alignment-free method for protein family modeling and prediction
Basic Utilities for PyTorch Natural Language Processing (NLP)
Super easy library for BERT based NLP models
A PyTorch implementation of EfficientNet
OpenCV implementation of Torchvision's image augmentations
mixup: Beyond Empirical Risk Minimization
Image augmentation for machine learning experiments.
Dataset with 5 million images depicting human-made and natural landmarks spanning 200 thousand classes.
Fast and flexible image augmentation library. Paper about the library: https://www.mdpi.com/2078-2489/11/2/125