Hsuxu / Loss_ToolBox-PyTorch
PyTorch Implementation of Focal Loss and Lovasz-Softmax Loss
☆333Updated 2 years ago
Alternatives and similar repositories for Loss_ToolBox-PyTorch:
Users that are interested in Loss_ToolBox-PyTorch are comparing it to the libraries listed below
- DiceLoss for PyTorch, both binary and multi-class.☆150Updated 5 years ago
- Implementation of DropBlock: A regularization method for convolutional networks in PyTorch.☆588Updated 4 years ago
- Pytorch implementation of the paper "Class-Balanced Loss Based on Effective Number of Samples"☆788Updated 9 months ago
- A PyTorch Implementation of Focal Loss.☆969Updated 5 years ago
- Image Test Time Augmentation with PyTorch!☆982Updated last year
- Code for our paper "Multi-scale Guided Attention for Medical Image Segmentation"☆461Updated 4 years ago
- 🛠 Toolbox to extend PyTorch functionalities☆416Updated 7 months ago
- Some classical and cutting-edge loss in semantic segmentation task☆119Updated 5 years ago
- Summarize the paper and code in AI(Semantic Segmentation, Medical Segmentation,REID,Super-Resolution,Registration,CVPR,ECCV,ICCV,AAAI,MIC…☆375Updated 4 years ago
- PyTorch Implementation of 2D and 3D 'squeeze and excitation' blocks for Fully Convolutional Neural Networks☆316Updated 4 years ago
- This is the code for the NeurIPS 2019 paper Region Mutual Information Loss for Semantic Segmentation.☆270Updated last year
- This is a PyTorch re-implementation of Axial-DeepLab (ECCV 2020 Spotlight)☆450Updated 3 years ago
- CRF-RNN PyTorch version http://crfasrnn.torr.vision☆197Updated 5 years ago
- The code for Expectation-Maximization Attention Networks for Semantic Segmentation (ICCV'2019 Oral)☆682Updated 4 years ago
- Official implementation of 'FMix: Enhancing Mixed Sample Data Augmentation'☆333Updated 3 years ago
- Semi-Supervised Semantic Segmentation with Cross-Consistency Training (CVPR 2020).☆395Updated 2 years ago
- Self-supervised Equivariant Attention Mechanism for Weakly Supervised Semantic Segmentation, CVPR 2020 (Oral)☆542Updated last year
- Code for the Lovász-Softmax loss (CVPR 2018)☆1,384Updated 8 months ago
- lookahead optimizer (Lookahead Optimizer: k steps forward, 1 step back) for pytorch☆333Updated 5 years ago
- Official code for "Boundary loss for highly unbalanced segmentation", runner-up for best paper award at MIDL 2019. Extended version in Me…☆662Updated 8 months ago
- PyTorch implementation of DeepLabv3☆253Updated 4 years ago
- Adversarial Learning for Semi-supervised Semantic Segmentation, BMVC 2018☆504Updated 3 years ago
- ☆196Updated 4 years ago
- This repository contains the reference implementation for our proposed Convolutional CRFs.☆564Updated 4 years ago
- PyTorch implementation of AutoAugment.☆158Updated 4 years ago
- Implementing Attention Augmented Convolutional Networks using Pytorch☆645Updated 2 years ago
- Pytorch implementation of newly added convolution☆584Updated 5 years ago
- Source code and model GALD net (BMVC-2019) and Dual-Seg Net (BMVC-2019)☆345Updated 3 years ago
- Code for our CVPR2020 paper "Strip Pooling: Rethinking Spatial Pooling for Scene Parsing"☆390Updated 4 years ago
- Weakly Supervised Learning of Instance Segmentation with Inter-pixel Relations, CVPR 2019 (Oral)☆529Updated last year