PPL Quantization Tool 0.6.5(PPL 量化工具)

PPL QuantTool (PPQ) is a highly efficient neural network quantization tool with custimized IR, cuda based executor, automatic dispacher and powerful optimization passes. Together with OpenPPL ecosystem, we offer you this industrial-grade network deploy tool that empowers AI developers to unleash the full potential of AI hardware. With quantization and other optimizations, nerual network model can run 5~10x faster than ever.

PPL QuantTool 是一个高效的工业级神经网络量化工具。 PPQ 被设计为一个灵活而全面的神经网络离线量化工具，我们允许你控制对量化进行细致入微的控制，同时严格控制硬件模拟误差。即便在网络极度复杂的情况下，我们依然能够得到正确的网络量化结果。PPQ 有着自定义的量化算子库、网络执行器、神经网络调度器、量化计算图等独特设计，这些为量化而设计的组件相互协作，共同构成了这一先进神经网络量化框架。借助 PPQ, OpenPPL, TensorRT, Tengine，ncnn等框架，你可以将神经网络模型提速 10 ~ 100 倍，并部署到多种多样的目标硬件平台，我们期待你将人工智慧带到千家万户之间。

目前 PPQ 使用 onnx(opset 11 ~ 13) 模型文件作为输入，覆盖常用的 90 余种 onnx 算子类型。如果你是 Pytorch, tensorflow 的用户，你可以使用这些框架提供的方法将模型转换到 onnx。PPQ 支持向 TensorRT, OpenPPL, Openvino, ncnn, Onnxruntime, Tengine, Snpe 等多个推理引擎生成目标文件并完成部署。借助算子自定义与平台自定义功能，你还可以将 PPQ 的量化能力扩展到其他可能的硬件上。

Code Example

Video Tutorial(Bilibili 视频教程)

Watch video tutorial series on www.bilibili.com, following are links of PPQ tutorial links(Only Chinese version).

• 安装教程: https://www.bilibili.com/video/BV1WS4y1N7Kn
• 使用教程: https://www.bilibili.com/video/BV1934y147p2
• 基础知识：https://www.bilibili.com/video/BV1gS4y1Y7KR
• 网络性能分析：https://www.bilibili.com/video/BV1oT4y1h73e
• 量化计算原理(Part 1)：https://www.bilibili.com/video/BV1fB4y1m7fJ
• 量化计算原理(Part 2)：https://www.bilibili.com/video/BV1qA4y1Q7Uh
• 图优化与量化模拟：https://www.bilibili.com/video/BV1Kr4y1n7cy
• 图调度与模式匹配：https://www.bilibili.com/video/BV1xY411A7ea
• 神经网络部署: https://www.bilibili.com/video/BV1t34y1E7Fz
• TensorRT 部署: https://www.bilibili.com/video/BV1AU4y127Uo
• 量化参数选择: https://www.bilibili.com/video/BV1QF41157aM
• 其他教程: 等待缓慢更新...

Installation

To release the power of this advanced quantization tool, at least one CUDA computing device is required. Install CUDA from CUDA Toolkit, PPL Quantization Tool will use CUDA compiler to compile cuda kernels at runtime.

ATTENTION: For users of PyTorch, PyTorch might bring you a minimized CUDA libraries, which will not satisfy the requirement of this tool, you have to install CUDA from NVIDIA manually.

ATTENTION: Make sure your Python version is >= 3.6.0. PPL Quantization Tool is written with dialects that only supported by Python >= 3.6.0.

• Install dependencies:

  • For Linux User, use following command to install ninja:

  sudo apt install ninja-build

  • For Windows User:
    • Download ninja.exe from https://github.com/ninja-build/ninja/releases, add it to Windows PATH Environment
    • Download Visual Studio 2019 from https://visualstudio.microsoft.com, if you already got a c++ compiler, you can skip this step.
    • Add your c++ compiler to Windows PATH Environment, if you are using Visual Studio, it should be something like "C:\Program Files\Microsoft Visual Studio\2019\Community\VC\Tools\MSVC\14.16.27023\bin\Hostx86\x86"
    • Update pytorch to 1.10+.

• Install PPQ from source:

1. Run following code with your terminal(For windows user, use command line instead).

git clone https://github.com/openppl-public/ppq.git
cd ppq
pip install -r requirements.txt
python setup.py install

2. Wait for Python finish its installation and pray for bug free.

• Install PPQ from Pip:

1. pre-built wheels are maintained in PPQ, you could simply install ppq with the following command(You should notice that install from pypi might get an outdated version ...)

python3 -m pip install ppq

Contact Us

WeChat Official Account	QQ Group
OpenPPL	627853444

Email: [email protected]

Other Resources

• Sensetime Parrots
• Sensetime Parrots Primitive Libraries
• Sensetime mmlab

Contributions

We appreciate all contributions. If you are planning to contribute back bug-fixes, please do so without any further discussion.

If you plan to contribute new features, utility functions, or extensions to the core, please first open an issue and discuss the feature with us. Sending a PR without discussion might end up resulting in a rejected PR because we might be taking the core in a different direction than you might be aware of.

Benchmark

PPQ is tested with models from mmlab-classification, mmlab-detection, mmlab-segamentation, mmlab-editing, here we listed part of out testing result.

• No quantization optimization procedure is applied with following models.

• PPQ(sim) stands for PPQ quantization simulator's result.
• Dispatcher stands for dispatching policy of PPQ.
• Classification models are evaluated with ImageNet, Detection and Segamentation models are evaluated with COCO dataset, Editing models are evaluated with DIV2K dataset.
• All calibration datasets are randomly picked from training data.

License

This project is distributed under the Apache License, Version 2.0.