1. Motivation
在往期的教程中,地平线工具链的环境安装都绕不开 天工开物工具包 OR Docker镜像。这两者作为地平线【训练】【量化】【部署】【教程】全功能的集大成者,难免显得有些臃肿,从下图的直观数字中读者可以发现,想要完成X3的上板,如果按照官方教程走,至少需要准备20G的空间。一个朴素的问题是:我们真的需要这20G的所有内容吗?-
2. Method
答案显然是否定的,地平线工程师根据长期实践删繁就简,总结出如下最轻量化的工具链使用方式:
2.1 最简模型转换
为了方便与系统其他环境隔离,我们推荐使用anaconda配置模型转换所需要的环境。
conda create -n test.py3.6 python=3.6 -y
conda activate test.py3.6
git clone https://github.com/xingchensong/toolchain_pkg.git
cd toolchain_pkg && tar -xzf pkg.tar.gz
pip install pkg/* torch numpy
是的你没看错,安装模型转换所需要的环境只需要这五句!
接下来跑一个只有一层conv2d的极简模型从导出onnx到转换为bin的demo:
cd toolchain_pkg/demo
conda activate test.py3.6
python demo.py
demo.py示例
运行结果示例
2.2 最简模型perf
环境安装:
conda activate test.py3.6
conda install graphviz
执行perf:
cd toolchain_pkg/demo
conda activate test.py3.6
python demo.py
hb_perf hb_makertbin_output/demo.bin
perf结果
perf结果详解1
perf结果详解2
perf结果详解3