QTool: A low-bit quantization toolbox for deep neural networks in computer vision

This project provides aboundant choices of quantization strategies (such as the quantization algorithms, training schedules and empirical tricks) for quantizing the image classification neural networks into low-bit counterparts. Associated projects demonstrate that this project can also act as a flexible plugin and benefit other computer vision tasks, such as object detection, segmentation and text parsing. Pretrained models are provided to show high standard of the code on achieving appealing quantization performance.

Instructions for different tasks

Update History

2020.12.12 Text parsing
2020.11.01 Super Resolution
2020.07.08 Instance Segmentation
2020.07.08 Object Detection
2020.06.23 Add classification quantization

Citation

Please cite the following work if you find the project helpful.

@misc{chen2020qtool,
author = {Peng Chen, Bohan Zhuang, Jing Liu and Chunlei Liu},
title = {{QTool: A low-bit quantization toolbox for deep neural networks in computer vision}},
year = {2020},
howpublished = {\url{https://github.com/aim-uofa/model-quantization}},
note = {Accessed: [Insert date here]}
}

For quantized object detection, please cite

@misc{liu2020aqd,
    title={AQD: Towards Accurate Quantized Object Detection},
    author={Peng Chen, Jing Liu, Bohan Zhuang, Mingkui Tan and Chunhua Shen},
    year={2020},
    eprint={2007.06919},
    archivePrefix={arXiv}
}

Also cite the corresponding publications when you choose dedicated algorithms.

Contribute

To contribute, PR is appreciated and suggestions are welcome to discuss with.

License