准备浮点模型和相关输入文件 - 3.5 简体中文

Vitis AI 用户指南 (UG1414)

Document ID
UG1414
Release Date
2023-09-28
Version
3.5 简体中文

执行 vai_q_tensorflow 前,请确保您的浮点格式的冻结推断 TensorFlow 模型和校准集都已准备就绪,其中应包括下表中列出的文件。

表 1. vai_q_tensorflow 的输入文件
编号 名称 描述
1 frozen_graph.pb 浮点冻结推断计算图。确保此计算图为推断计算图,而非训练计算图。
2 校准数据集 训练数据集的子集,包含 100 到 1000 张图像。
3 input_fn 这是输入函数,用于在训练后量化期间将校准数据集转换为 frozen_graph 的输入数据。通常用于执行数据预处理和数据增广。