FrancescoSaverioZuppichini / non-max-suppression-in-pytorch
How to implement Non Max Suppression (NMS) in PyTorch
☆13Updated last year
Related projects: ⓘ
- Heatmap Learner Convolutional Neural Network for Object Counting and Localization☆42Updated 7 months ago
- ☆24Updated 2 years ago
- Implementing DropPath/StochasticDepth in PyTorch☆16Updated 2 years ago
- Implementation of CaiT models in TensorFlow and ImageNet-1k checkpoints. Includes code for inference and fine-tuning.☆12Updated last year
- Implementing RepVGG in PyTorch☆18Updated 2 years ago
- TensortRT installation and Conversion from PyTorch Models☆25Updated 4 years ago
- PyTorch and TensorFlow/Keras image models with automatic weight conversions and equal API/implementations - Vision Transformer (ViT), Res…☆33Updated last year
- Tiny ResNet inspired FPN network (<2M params) for Rotated Object Detection using 5-parameter Modulated Rotation Loss☆18Updated 3 years ago
- Official PyTorch implementation of RIO☆18Updated 3 years ago
- Implementation of CenterNet and FairMOT with PyTorch Lightning☆69Updated 2 years ago
- Comparing four automatic image augmentation techniques in PyTorch: AutoAugment, RandAugment, AugMix, and TrivialAugment☆26Updated last year
- Knock your images before you get stressed.☆10Updated 2 years ago
- Adversarial examples to the new ConvNeXt architecture☆20Updated 2 years ago
- LightCollections⚡️: Ready-to-use implementations such as `LightningModules` for various computer vision papers.☆23Updated last year
- An easy to use evaluator for object detection performance metrics, such as mAP and AR☆18Updated 2 years ago
- ☆14Updated 2 years ago
- Accompanying notebook and sources to "A Guide to Pseudolabelling: How to get a Kaggle medal with only one model" (Dec. 2020 PyData Boston…☆27Updated 2 years ago
- A simple wrapper library for binding timm models as detectron2 backbones☆38Updated last year
- Adds timm pretrained backbone to pytorch's FasterRcnn model☆12Updated 7 months ago
- Includes PyTorch -> Keras model porting code for DeiT models with fine-tuning and inference notebooks.☆41Updated 2 years ago
- Implementation of Deep Orthogonal Fusion of Local and Global Features in TensorFlow 2☆25Updated last year
- TF 2 implementation Learning to Resize Images for Computer Vision Tasks (https://arxiv.org/abs/2103.09950v1).☆51Updated 2 years ago
- TF2 implementation of knowledge distillation using the "function matching" hypothesis from https://arxiv.org/abs/2106.05237.☆87Updated 2 years ago
- Simplify Your Visual Data Ops. Find and visualize issues with your computer vision datasets such as duplicates, anomalies, data leakage, …☆65Updated 11 months ago
- Examples of using PyTorch hooks, as covered in my YouTube tutorial video.☆32Updated 10 months ago
- QUick and DIrty Domain Adaptation☆20Updated 8 months ago
- State-of-the-art data augmentation search algorithms in PyTorch☆47Updated last year
- A Keras implementation of hybrid efficientnet swin transformer model.☆33Updated 11 months ago
- Attempting to build YOLOX from scratch☆28Updated last year
- An implementation of GhostNet for Tensorflow 2.0+ (From the paper "GhostNet: More Features from Cheap Operations")☆37Updated last year