A PyTorch Implementation of Focal Loss.
☆991Oct 16, 2019Updated 6 years ago
Alternatives and similar repositories for focal_loss_pytorch
Users that are interested in focal_loss_pytorch are comparing it to the libraries listed below. We may earn a commission when you buy through links labeled 'Ad' on this page.
Sorting:
- label-smooth, amsoftmax, partial-fc, focal-loss, triplet-loss, lovasz-softmax. Maybe useful☆2,262Oct 17, 2024Updated last year
- 全中文注释.(The loss function of retinanet based on pytorch).(You can use it on one-stage detection task or classifical task, to solve data …☆495Oct 9, 2025Updated 5 months ago
- PyTorch Implementation of Focal Loss and Lovasz-Softmax Loss☆338Dec 24, 2021Updated 4 years ago
- Code for the Lovász-Softmax loss (CVPR 2018)☆1,410Apr 12, 2024Updated last year
- RetinaNet in PyTorch☆999Mar 17, 2019Updated 7 years ago
- A PyTorch implementation of EfficientNet☆8,219Apr 8, 2022Updated 3 years ago
- Focal loss for Dense Object Detection☆488Dec 20, 2017Updated 8 years ago
- PyTorch implementation of SENet☆2,345Mar 2, 2021Updated 5 years ago
- Class-Balanced Loss Based on Effective Number of Samples. CVPR 2019☆616Aug 29, 2021Updated 4 years ago
- A CV toolkit for my papers.☆2,048Dec 21, 2024Updated last year
- Pretrained ConvNets for pytorch: NASNet, ResNeXt, ResNet, InceptionV4, InceptionResnetV2, Xception, DPN, etc.☆9,109Apr 22, 2022Updated 3 years ago
- A (PyTorch) imbalanced dataset sampler for oversampling low frequent classes and undersampling high frequent ones.☆2,324Jan 5, 2026Updated 2 months ago
- DeepLab v3+ model in PyTorch. Support different backbones.☆3,006Aug 4, 2024Updated last year
- ResNeSt: Split-Attention Networks☆3,264Dec 9, 2022Updated 3 years ago
- Pytorch implementation of Center Loss☆995Feb 19, 2023Updated 3 years ago
- Pytorch implementation of the paper "Class-Balanced Loss Based on Effective Number of Samples"☆806Feb 18, 2024Updated 2 years ago
- Dual Attention Network for Scene Segmentation (CVPR2019)☆2,457Dec 23, 2024Updated last year
- The largest collection of PyTorch image encoders / backbones. Including train, eval, inference, export scripts, and pretrained weights --…☆36,538Updated this week
- Count the MACs / FLOPs of your PyTorch model.☆5,080Jul 8, 2024Updated last year
- Object detection, 3D detection, and pose estimation using center point detection:☆7,551Mar 2, 2023Updated 3 years ago
- Implementation of Non-local Block.☆1,568Apr 12, 2023Updated 2 years ago
- Fast and flexible image augmentation library. Paper about the library: https://www.mdpi.com/2078-2489/11/2/125☆15,286Jun 25, 2025Updated 8 months ago
- A faster pytorch implementation of faster r-cnn☆7,863May 20, 2022Updated 3 years ago
- A PyTorch implementation of the Transformer model in "Attention is All You Need".☆9,654Apr 16, 2024Updated last year
- Deformable ConvNets V2 (DCNv2) in PyTorch☆1,489Nov 18, 2022Updated 3 years ago
- Implementation of the mixup training method☆469Jun 12, 2018Updated 7 years ago
- PyTorch implementation of Deformable Convolution☆911Jul 21, 2021Updated 4 years ago
- OpenMMLab Detection Toolbox and Benchmark☆32,507Aug 21, 2024Updated last year
- A PyTorch Extension: Tools for easy mixed precision and distributed training in Pytorch☆8,936Mar 17, 2026Updated last week
- Minimal PyTorch implementation of YOLOv3☆7,438Nov 17, 2024Updated last year
- Synchronized Batch Normalization implementation in PyTorch.☆1,503Apr 8, 2021Updated 4 years ago
- Advanced AI Explainability for computer vision. Support for CNNs, Vision Transformers, Classification, Object detection, Segmentation, I…☆12,701Apr 7, 2025Updated 11 months ago
- Pytorch implementation for Semantic Segmentation/Scene Parsing on MIT ADE20K dataset☆5,064Jan 15, 2024Updated 2 years ago
- Pytorch implementation of Feature Pyramid Network (FPN) for Object Detection☆970Jan 11, 2019Updated 7 years ago
- Fast, modular reference implementation of Instance Segmentation and Object Detection algorithms in PyTorch.☆9,386Feb 16, 2023Updated 3 years ago
- TorchCV: A PyTorch-Based Framework for Deep Learning in Computer Vision☆2,256Nov 19, 2020Updated 5 years ago
- A PyTorch Implementation of Single Shot MultiBox Detector☆5,231Dec 29, 2021Updated 4 years ago
- Siamese and triplet networks with online pair/triplet mining in PyTorch☆3,169Apr 29, 2023Updated 2 years ago
- 3D ResNets for Action Recognition (CVPR 2018)☆4,042Jan 20, 2021Updated 5 years ago