332 Open source projects that are alternatives of or similar to torch-model-compression

micronet, a model compression and deploy lib. compression: 1、quantization: quantization-aware-training(QAT), High-Bit(>2b)(DoReFa/Quantization and Training of Neural Networks for Efficient Integer-Arithmetic-Only Inference)、Low-Bit(≤2b)/Ternary and Binary(TWN/BNN/XNOR-Net); post-training-quantization(PTQ), 8-bit(tensorrt); 2、 pruning: normal、regular and group convolutional channel pruning; 3、 group convolution structure; 4、batch-normalization fuse for quantization. deploy: tensorrt, fp32/fp16/int8(ptq-calibration)、op-adapt(upsample)、dynamic_shape

Easy-to-use UI for automatically sparsifying neural networks and creating sparsification recipes for better inference performance and a smaller footprint

InsightFace REST API for easy deployment of face recognition services with TensorRT in Docker.

A Pytorch Knowledge Distillation library for benchmarking and extending works in the domains of Knowledge Distillation, Pruning, and Quantization.

PyTorch Project Specification.

Infrastructures™ for Machine Learning Training/Inference in Production.

PaddleSlim is an open-source library for deep model compression and architecture search.

Awesome machine learning model compression research papers, tools, and learning material.

[NeurIPS'2019] Shupeng Gui, Haotao Wang, Haichuan Yang, Chen Yu, Zhangyang Wang, Ji Liu, “Model Compression with Adversarial Robustness: A Unified Optimization Framework”

Neural Network Distiller by Intel AI Lab: a Python package for neural network compression research. https://intellabs.github.io/distiller

Intel® Neural Compressor (formerly known as Intel® Low Precision Optimization Tool), targeting to provide unified APIs for network compression technologies, such as low precision quantization, sparsity, pruning, knowledge distillation, across different deep learning frameworks to pursue optimal inference performance.

A toolkit to optimize ML models for deployment for Keras and TensorFlow, including quantization and pruning.

Filter Pruning via Geometric Median for Deep Convolutional Neural Networks Acceleration (CVPR 2019 Oral)

QAT(quantize aware training) for classification with MQBench

This is a highly separated deployment project based on Deepstream , including the full range of Yolo and continuously expanding deployment projects such as Ocr.

(CVPR 2021, Oral) Dynamic Slimmable Network

The purpose of this project is to apply mediapipe to more AI chips.

tensorrt int8 量化yolov5 onnx模型

Neural network model repository for highly sparse and sparse-quantized models with matching sparsification recipes

PyTorch*-based Neural Network Compression Framework for enhanced OpenVINO™ inference

This repository contains a docker image that I use to develop my artificial intelligence applications in an uncomplicated fashion. Python, TensorFlow, PyTorch, ONNX, Keras, OpenCV, TensorRT, Numpy, Jupyter notebook... 🐋🔥

A pytorch pruning toolkit for structured neural network pruning and layer dependency maintaining.

PPL Quantization Tool (PPQ) is a powerful offline neural network quantization tool.

YOLOX is a high-performance anchor-free YOLO, exceeding yolov3~v5 with MegEngine, ONNX, TensorRT, ncnn, and OpenVINO supported. Documentation: https://yolox.readthedocs.io/

PyTorch ,ONNX and TensorRT implementation of YOLOv4

🏎️ Accelerate training and inference of 🤗 Transformers with easy to use hardware optimization tools

BitPack is a practical tool to efficiently save ultra-low precision/mixed-precision quantized models.

CVPR 2021 : Zero-shot Adversarial Quantization (ZAQ)

PyTorch Model Compression

A curated list of awesome edge machine learning resources, including research papers, inference engines, challenges, books, meetups and others.

⚡ boost inference speed of T5 models by 5x & reduce the model size by 3x.

Multiple types of NN model optimization environments. It is possible to directly access the host PC GUI and the camera to verify the operation. Intel iHD GPU (iGPU) support. NVIDIA GPU (dGPU) support.

Efficient ML Filter Runtimes for VapourSynth (with built-in support for waifu2x, DPIR, RealESRGANv2, and Real-CUGAN)

🛠️ Tools for Transformers compression using PyTorch Lightning ⚡

[ICLR'21] PyTorch code for our paper "Neural Pruning via Growing Regularization"

Pruning and quantization for SSD. Model compression.

TensorRT person tracking RFBNet300

tensorRt-inference darknet2onnx pytorch2onnx mxnet2onnx python version

Docker scripts for building ONNX Runtime with TensorRT and OpenVINO in manylinux environment

Tengine is a lite, high performance, modular inference engine for embedded device

reference pytorch code for named entity tagging

Soft Filter Pruning for Accelerating Deep Convolutional Neural Networks

A curated list of neural network pruning resources.

An Open Source Deep Learning Inference Engine Based on FPGA

Quantization library for PyTorch. Support low-precision and mixed-precision quantization, with hardware implementation through TVM.

A list of high-quality (newest) AutoML works and lightweight models including 1.) Neural Architecture Search, 2.) Lightweight Structures, 3.) Model Compression, Quantization and Acceleration, 4.) Hyperparameter Optimization, 5.) Automated Feature Engineering.

A repository that shares tuning results of trained models generated by TensorFlow / Keras. Post-training quantization (Weight Quantization, Integer Quantization, Full Integer Quantization, Float16 Quantization), Quantization-aware training. TensorFlow Lite. OpenVINO. CoreML. TensorFlow.js. TF-TRT. MediaPipe. ONNX. [.tflite,.h5,.pb,saved_model,tfjs,tftrt,mlmodel,.xml/.bin, .onnx]

Pretrained language model and its related optimization techniques developed by Huawei Noah's Ark Lab.

[NeurIPS'21] "Chasing Sparsity in Vision Transformers: An End-to-End Exploration" by Tianlong Chen, Yu Cheng, Zhe Gan, Lu Yuan, Lei Zhang, Zhangyang Wang

AIMET is a library that provides advanced quantization and compression techniques for trained neural network models.

Embedded and mobile deep learning research resources

nanodet int8 量化，实测推理2ms一帧！

Tengine Convert Tool supports converting multi framworks' models into tmfile that suitable for Tengine-Lite AI framework.

Automatic DNN compression tool with various model compression and neural architecture search techniques

A research library for pytorch-based neural network pruning, compression, and more.

一个多语言支持、易使用的 OCR 项目。An easy-to-use OCR project with multilingual support.

This is the implementation that supports yolov5s, yolov5m, yolov5l, yolov5x.

TF2 Deep FloorPlan Recognition using a Multi-task Network with Room-boundary-Guided Attention. Enable tensorboard, quantization, flask, tflite, docker, github actions and google colab.

Reference ImageNet implementation of SelecSLS CNN architecture proposed in the SIGGRAPH 2020 paper "XNect: Real-time Multi-Person 3D Motion Capture with a Single RGB Camera". The repository also includes code for pruning the model based on implicit sparsity emerging from adaptive gradient descent methods, as detailed in the CVPR 2019 paper "On implicit filter level sparsity in Convolutional Neural Networks".

[ICLR 2021] "GANs Can Play Lottery Too" by Xuxi Chen, Zhenyu Zhang, Yongduo Sui, Tianlong Chen