Deeplite / deeplite-torch-zoo
Collection of SOTA efficient computer vision models for embedded applications, with pre-trained weights and training recipes
☆74Updated last week
Related projects: ⓘ
- PyTorch Pruning Example☆46Updated last year
- ☆56Updated last year
- Using ideas from product quantization for state-of-the-art neural network compression.☆145Updated 3 years ago
- PyTorch implementation of soft-nms☆26Updated last month
- Count number of parameters / MACs / FLOPS for ONNX models.☆84Updated 2 years ago
- Refactored implementation of CenterNet (Objects as Points - Zhou, Xingyi et. al.) shipping with PyTorch Lightning modules☆59Updated last year
- Official codes of ICCV2023 paper: <<FemtoDet: an object detection baseline for energy versus performance tradeoffs>>☆62Updated 3 months ago
- Attempting to build YOLOX from scratch☆28Updated last year
- Easily benchmark PyTorch model FLOPs, latency, throughput, allocated gpu memory and energy consumption☆87Updated last year
- ☆27Updated last year
- Implementation of CenterNet and FairMOT with PyTorch Lightning☆69Updated 2 years ago
- A collection of metrics to profile a single deep learning model or compare two different deep learning models☆24Updated 10 months ago
- ☆52Updated 3 years ago
- ☆47Updated this week
- Just another yolo variant.☆24Updated 2 years ago
- Official PyTorch Implementation of Mask-aware IoU and maYOLACT Detector [BMVC2021]☆52Updated 2 years ago
- PyTorch Static Quantization Example☆39Updated 3 years ago
- A package to make do Network Slimming a little easier☆47Updated 2 years ago
- A PyTorch implementation of SSDLite on COCO☆84Updated 3 years ago
- Implementation of YOLOv9 QAT optimized for deployment on TensorRT platforms.☆71Updated 2 months ago
- Implementation of knapsack pruning☆28Updated 4 years ago
- PyTorch Quantization Aware Training Example☆119Updated 4 months ago
- Public documentation of Deeplite Neutrino™ engine.☆63Updated last year
- Implementation for the paper 'YOLO-ReT: Towards High Accuracy Real-time Object Detection on Edge GPUs'☆91Updated last year
- Pytorch implementation of EfficientNet-lite. ImageNet pre-trained models are provided.☆92Updated 3 years ago
- Scailable ONNX python tools☆96Updated this week
- Tiny ResNet inspired FPN network (<2M params) for Rotated Object Detection using 5-parameter Modulated Rotation Loss☆18Updated 3 years ago
- TF2 implementation of knowledge distillation using the "function matching" hypothesis from https://arxiv.org/abs/2106.05237.☆87Updated 2 years ago
- A multibranch neural net architecture. (PyTorch)☆82Updated 4 years ago
- Implementation of the Confluence and Confluence NMS algorithms☆23Updated 2 years ago