执行 vai_q_tensorflow 前,请确保您的浮点格式的冻结推断 TensorFlow 模型和校准集都已准备就绪,其中应包括下表中列出的文件。
编号 | 名称 | 描述 |
---|---|---|
1 | frozen_graph.pb | 浮点冻结推断计算图。确保此计算图为推断计算图,而非训练计算图。 |
2 | 校准数据集 | 训练数据集的子集,包含 100 到 1000 张图像。 |
3 | input_fn | 这是输入函数,用于在训练后量化期间将校准数据集转换为 frozen_graph 的输入数据。通常用于执行数据预处理和数据增广。 |