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,256Oct 17, 2024Updated last year
- PyTorch Implementation of Focal Loss and Lovasz-Softmax Loss☆335Dec 24, 2021Updated 4 years ago
- Code for the Lovász-Softmax loss (CVPR 2018)☆1,410Apr 12, 2024Updated 2 years ago
- RetinaNet in PyTorch☆998Mar 17, 2019Updated 7 years ago
- A PyTorch implementation of EfficientNet☆8,221Apr 8, 2022Updated 4 years ago
- AI Agents on DigitalOcean Gradient AI Platform • AdBuild production-ready AI agents using customizable tools or access multiple LLMs through a single endpoint. Create custom knowledge bases or connect external data.
- Focal loss for Dense Object Detection☆487Dec 20, 2017Updated 8 years ago
- PyTorch implementation of SENet☆2,343Mar 2, 2021Updated 5 years ago
- Class-Balanced Loss Based on Effective Number of Samples. CVPR 2019☆616Aug 29, 2021Updated 4 years ago
- Pretrained ConvNets for pytorch: NASNet, ResNeXt, ResNet, InceptionV4, InceptionResnetV2, Xception, DPN, etc.☆9,097Apr 22, 2022Updated 4 years ago
- A CV toolkit for my papers.☆2,046Dec 21, 2024Updated last year
- A (PyTorch) imbalanced dataset sampler for oversampling low frequent classes and undersampling high frequent ones.☆2,322Jul 6, 2026Updated 3 weeks ago
- DeepLab v3+ model in PyTorch. Support different backbones.☆3,000Aug 4, 2024Updated last year
- ResNeSt: Split-Attention Networks☆3,263Dec 9, 2022Updated 3 years ago
- Pytorch implementation of Center Loss☆992Feb 19, 2023Updated 3 years ago
- Bare Metal GPUs on DigitalOcean Gradient AI • AdPurpose-built for serious AI teams training foundational models, running large-scale inference, and pushing the boundaries of what's possible.
- Pytorch implementation of the paper "Class-Balanced Loss Based on Effective Number of Samples"☆803Feb 18, 2024Updated 2 years ago
- Dual Attention Network for Scene Segmentation (CVPR2019)☆2,461Dec 23, 2024Updated last year
- The largest collection of PyTorch image encoders / backbones. Including train, eval, inference, export scripts, and pretrained weights --…☆37,025Updated this week
- Count the MACs / FLOPs of your PyTorch model.☆5,078Jul 8, 2024Updated 2 years ago
- Object detection, 3D detection, and pose estimation using center point detection:☆7,570Mar 2, 2023Updated 3 years ago
- Implementation of Non-local Block.☆1,564Apr 12, 2023Updated 3 years ago
- Fast and flexible image augmentation library. Paper about the library: https://www.mdpi.com/2078-2489/11/2/125☆15,322Jun 25, 2025Updated last year
- A faster pytorch implementation of faster r-cnn☆7,856May 20, 2022Updated 4 years ago
- A PyTorch implementation of the Transformer model in "Attention is All You Need".☆9,774Apr 16, 2024Updated 2 years ago
- 1-Click AI Models by DigitalOcean Gradient • AdDeploy popular AI models on DigitalOcean Gradient GPU virtual machines with just a single click. Zero configuration with optimized deployments.
- Implementation of the mixup training method☆469Jun 12, 2018Updated 8 years ago
- Deformable ConvNets V2 (DCNv2) in PyTorch☆1,485Nov 18, 2022Updated 3 years ago
- PyTorch implementation of Deformable Convolution☆905Jul 21, 2021Updated 5 years ago
- OpenMMLab Detection Toolbox and Benchmark☆32,855Aug 21, 2024Updated last year
- A PyTorch Extension: Tools for easy mixed precision and distributed training in Pytorch☆8,986Updated this week
- Minimal PyTorch implementation of YOLOv3☆7,438Nov 17, 2024Updated last year
- The implementation of “Gradient Harmonized Single-stage Detector” published on AAAI 2019.☆618Jul 15, 2019Updated 7 years ago
- Advanced AI Explainability for computer vision. Support for CNNs, Vision Transformers, Classification, Object detection, Segmentation, I…☆12,942Jul 10, 2026Updated 3 weeks ago
- Pytorch implementation for Semantic Segmentation/Scene Parsing on MIT ADE20K dataset☆5,076Jan 15, 2024Updated 2 years ago
- AI Agents on DigitalOcean Gradient AI Platform • AdBuild production-ready AI agents using customizable tools or access multiple LLMs through a single endpoint. Create custom knowledge bases or connect external data.
- Synchronized Batch Normalization implementation in PyTorch.☆1,502Apr 8, 2021Updated 5 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,364Feb 16, 2023Updated 3 years ago
- TorchCV: A PyTorch-Based Framework for Deep Learning in Computer Vision☆2,250Nov 19, 2020Updated 5 years ago
- A PyTorch Implementation of Single Shot MultiBox Detector☆5,223Dec 29, 2021Updated 4 years ago
- 3D ResNets for Action Recognition (CVPR 2018)☆4,037Jan 20, 2021Updated 5 years ago
- Siamese and triplet networks with online pair/triplet mining in PyTorch☆3,175Apr 29, 2023Updated 3 years ago