torchsummaryX: Improved visualization tool of torchsummary
☆303Mar 17, 2022Updated 3 years ago
Alternatives and similar repositories for torchsummaryX
Users that are interested in torchsummaryX are comparing it to the libraries listed below
Sorting:
- Model summary in PyTorch similar to `model.summary()` in Keras☆4,064Mar 2, 2024Updated 2 years ago
- Count the MACs / FLOPs of your PyTorch model.☆5,081Jul 8, 2024Updated last year
- FCOS: Fully Convolutional One-Stage Object Detection.☆25Oct 7, 2022Updated 3 years ago
- [IROS 2021] ADD: A Fine-grained Dynamic Inference Architecture for Semantic Image Segmentation☆10May 3, 2022Updated 3 years ago
- A python package for Tensorflow. It has the right level of abstraction needed to built state-of-the-art deep learning models. Tensormodel…☆13Jul 5, 2016Updated 9 years ago
- Flops counter for neural networks in pytorch framework☆2,977Aug 20, 2025Updated 6 months ago
- TensorRT half precision inference routine on a API-based TensorRT model☆13Jul 3, 2018Updated 7 years ago
- EdgePose: Pose estimation library for Edge computing using TensorRT, ONNX, Pytorch, C++, CoreML☆12Jul 26, 2019Updated 6 years ago
- (ICML 2021) Implementation for S2SD - Simultaneous Similarity-based Self-Distillation for Deep Metric Learning. Paper Link: https://arxiv…☆44Sep 18, 2020Updated 5 years ago
- PyTorch layer-by-layer model profiler☆606May 23, 2021Updated 4 years ago
- Unofficial implementation for SOLO instance segmentation☆25Mar 29, 2020Updated 5 years ago
- Residual Refine based Pseudo Multi-frame Network for Efficient Single Image Super-Resolution☆14Oct 27, 2018Updated 7 years ago
- PyTorch code for our ECCV 2018 paper "Image Super-Resolution Using Very Deep Residual Channel Attention Networks"☆1,486Apr 1, 2025Updated 11 months ago
- Model analyzer in PyTorch☆1,501Mar 19, 2023Updated 2 years ago
- pytorch implementation of Funnel Activation (FReLU)☆16Aug 16, 2020Updated 5 years ago
- 🛠 Toolbox to extend PyTorch functionalities☆421May 13, 2024Updated last year
- Pytorch implementation of our paper accepted by IEEE TNNLS, 2022 -- Distilling a Powerful Student Model via Online Knowledge Distillation☆31Nov 11, 2021Updated 4 years ago
- Collection of common code that's shared among different research projects in FAIR computer vision team.☆2,225Feb 25, 2026Updated last week
- ☆17Jul 10, 2022Updated 3 years ago
- PyTorch implementation for OD-cheap-convolution.☆20Sep 29, 2019Updated 6 years ago
- Rotate RoI Align and Rotate Position Sensitive RoI Align Operation in Caffe☆15Dec 5, 2018Updated 7 years ago
- ☆17Apr 1, 2020Updated 5 years ago
- An easy to understand and better performance version of CenterNet☆550Apr 12, 2021Updated 4 years ago
- A PyTorch Extension: Tools for easy mixed precision and distributed training in Pytorch☆8,928Updated this week
- Pytorch implementation of our paper accepted by IEEE TNNLS, 2021 -- Network Pruning using Adaptive Exemplar Filters☆24Apr 4, 2021Updated 4 years ago
- Pytorch implementation of newly added convolution☆583Oct 26, 2019Updated 6 years ago
- Pytorch implementation of our paper accepted by TPAMI 2023 — Lottery Jackpots Exist in Pre-trained Models☆35Jun 19, 2023Updated 2 years ago
- ATSS-EfficientDet implemented in PyTorch, outperforming the original EfficientDet. Weights are released.☆67Apr 30, 2020Updated 5 years ago
- unofficial implementation of CondConv: Conditionally Parameterized Convolutions for Efficient Inference in PyTorch.☆166Jan 7, 2024Updated 2 years ago
- View model summaries in PyTorch!☆2,903Mar 2, 2026Updated last week
- (ECCV 2020) Interactive Multi-Dimension Modulation with Dynamic Controllable Residual Learning for Image Restoration☆96Sep 28, 2020Updated 5 years ago
- Generative Adversarial Networks for Image Super-Resolution: A Survey☆17May 5, 2022Updated 3 years ago
- Pytorch code for our paper "Feedback Network for Image Super-Resolution" (CVPR2019)☆552Dec 17, 2019Updated 6 years ago
- PyTorch implementation of Pose Neural Fabrics Search (PNFS)--Part-specific neural architecture search for human pose estimation.☆52Jan 3, 2021Updated 5 years ago
- ☆18Apr 3, 2018Updated 7 years ago
- Experiments based on CenterNet (more backbones, TensorRT deployment and mask head)☆222Jun 16, 2020Updated 5 years ago
- PyTorch code for ICIP 2022 paper "A Unified Framework for Masked and Mask-Free Face Recognition via Feature Rectification"☆35Jul 8, 2024Updated last year
- ResNeSt: Split-Attention Networks☆3,264Dec 9, 2022Updated 3 years ago
- ☆491May 14, 2022Updated 3 years ago