针对pytorch模型的自动化模型结构分析和修改工具集，包含自动分析模型结构的模型压缩算法库
☆256Apr 19, 2023Updated 2 years ago
Alternatives and similar repositories for torch-model-compression
Users that are interested in torch-model-compression are comparing it to the libraries listed below
Sorting:
- 高效部署：YOLO X, V3, V4, V5, V6, V7, V8, EdgeYOLO TRT推理 ™️ ,前后处理均由CUDA核函数实现 CPP/CUDA🚀☆53Feb 23, 2023Updated 3 years ago
- micronet, a model compression and deploy lib. compression: 1、quantization: quantization-aware-training(QAT), High-Bit(>2b)(DoReFa/Quantiz…☆2,269May 6, 2025Updated 9 months ago
- cpp rotation album，基于cpp eigen实现的3d旋转相册，GAMES101复现内容☆12Jul 25, 2022Updated 3 years ago
- Various test models in WNNX format. It can view with `pip install wnetron && wnetron`☆12Jun 22, 2022Updated 3 years ago
- DEYOv1.5☆29Jul 22, 2024Updated last year
- An onnx-based quantitation tool.☆71Jan 8, 2024Updated 2 years ago
- TinyNeuralNetwork is an efficient and easy-to-use deep learning model compression framework.☆866Dec 24, 2025Updated 2 months ago
- [CVPR 2023] DepGraph: Towards Any Structural Pruning; LLMs, Vision Foundation Models, etc.☆3,262Sep 7, 2025Updated 5 months ago
- yolov3 yolov4 channel and layer pruning, Knowledge Distillation 层剪枝，通道剪枝，知识蒸馏☆1,519Sep 3, 2020Updated 5 years ago
- ☆79May 16, 2023Updated 2 years ago
- The c++ implement of Multi class ByteTrack☆14Mar 24, 2023Updated 2 years ago
- More readable and flexible yolov5 with more backbone(gcn, resnet, shufflenet, moblienet, efficientnet, hrnet, swin-transformer, etc) and …☆686Aug 19, 2024Updated last year
- mobilev2-yolov5s剪枝、蒸馏，支持ncnn，tensorRT部署。ultra-light but better performence！☆857Sep 26, 2022Updated 3 years ago
- This repo is implemented based on detectron2 and centernet☆558Dec 31, 2020Updated 5 years ago
- YOLO ModelCompression MultidatasetTraining☆445Jun 21, 2022Updated 3 years ago
- Quantize yolov5 using pytorch_quantization.🚀🚀🚀☆14Oct 24, 2023Updated 2 years ago
- PPL Quantization Tool (PPQ) is a powerful offline neural network quantization tool.☆1,785Mar 28, 2024Updated last year
- TensorRT ONNX Plugin、Inference、Compile☆464Jul 26, 2021Updated 4 years ago
- yolov5 prune，Support V2, V3, V4 and V6 versions of yolov5☆583Jan 6, 2022Updated 4 years ago
- Support Yolov5(4.0)/Yolov5(5.0)/YoloR/YoloX/Yolov4/Yolov3/CenterNet/CenterFace/RetinaFace/Classify/Unet. use darknet/libtorch/pytorch/mxn…☆210Aug 2, 2021Updated 4 years ago
- pytorch implementation of Funnel Activation (FReLU)☆16Aug 16, 2020Updated 5 years ago
- Deploy your model with TensorRT quickly.☆764Nov 21, 2023Updated 2 years ago
- Experiments based on CenterNet (more backbones, TensorRT deployment and mask head)☆222Jun 16, 2020Updated 5 years ago
- Samples code for world class Artificial Intelligence SoCs for computer vision applications.☆284Updated this week
- 自监督目标检测。针对目标检测任务，提出无需标签数据的自监督算法预训练 backbone，检测性能优于有标签的预训练。☆29Apr 28, 2022Updated 3 years ago
- OpenMMLab Model Compression Toolbox and Benchmark.☆1,662Jun 11, 2024Updated last year
- quantize aware training package for NCNN on pytorch☆68Jul 27, 2021Updated 4 years ago
- 本项目支持对剪枝后的yolov5模型进行知识蒸馏训练(This project supports knowledge distillation training for the pruned YOLOv5 model)☆108Jan 28, 2024Updated 2 years ago
- An implementation of ResNet with mixup and cutout regularizations and soft filter pruning.☆17Feb 23, 2026Updated last week
- PaddleSlim is an open-source library for deep model compression and architecture search.☆1,612Jan 4, 2026Updated 2 months ago
- YOLOv3、YOLOv4、YOLOv5、YOLOv5-Lite、YOLOv6-v1、YOLOv6-v2、YOLOv7、YOLOX、YOLOX-Lite、PP-YOLOE、PP-PicoDet-Plus、YOLO-Fastest v2、FastestDet、YOLOv5-S…☆765Oct 25, 2022Updated 3 years ago
- On-the-fly Structured Pruning for PyTorch models. This library implements several attributions metrics and structured pruning utils for n…☆167Jun 4, 2020Updated 5 years ago
- mobileNet SSD 基于caffe的前向检测☆10Nov 30, 2018Updated 7 years ago
- ☆241Nov 29, 2022Updated 3 years ago
- provide some new architecture, channel pruning and quantization methods for yolov5☆31Oct 13, 2025Updated 4 months ago
- A model compression and acceleration toolbox based on pytorch.☆333Jan 12, 2024Updated 2 years ago
- This is a collection of our zero-cost NAS and efficient vision applications.☆449Aug 21, 2023Updated 2 years ago
- YOLOv5 Quantization Aware Training with TensorRT☆19Jan 10, 2023Updated 3 years ago
- ☆19May 28, 2020Updated 5 years ago