数字IC萌新
我走通过这个流程,分享些实操细节。用PYNQ-Z2板子,软件训练用TensorFlow,量化用TensorFlow Lite的INT8转换,然后用到Xilinx的DNNDK工具链(现在升级为Vitis AI了)。具体:训练后保存为pb文件,用Vitis AI的quantizer校准生成量化模型,再用compiler针对目标板卡编译。部署时把编译好的文件放到PYNQ的/usr/share目录下,在Jupyter里用Python加载DPU驱动和模型文件,调用推理。注意点:板载内存有限,模型层数不能太深;量化校准需要一小部分验证集图片;如果编译出错,通常是某些算子不支持,得简化模型。全流程教程在Xilinx官网有,搜'Vitis AI MNIST tutorial'就行。先按教程复现,再自己改模型结构。
