使用TFLite量化模型的参数进行计算操作

时间:2019-11-30 17:08:23

标签: tensorflow tensorflow-lite mobilenet tf-lite

我正在尝试使用取自here的量化Mobilenetv2模型在硬件中实现图像分类。为此,我首先需要从头到尾重现推理过程,以确保我理解对数据执行的计算/运算。

第一个目标是Conv fuction。我可以看到它是如何计算的,但是有几个参数传递给该函数,我想知道它们是如何产生的:output_offset, output_multiplier,output_shift, output_activation_min, output_activation_max。我找不到使用这些参数调用 Conv()函数的先前函数。希望这将使我对这些论点是如何产生的有深刻的了解。有人可以指出我在源代码的右行吗?

源代码中的另一个空白是在interpreter.invoke()函数中。我希望跟踪并查看接下来会发生什么,但是找不到实现 invoke()函数的源代码。帮助将不胜感激!

1 个答案:

答案 0 :(得分:1)

如果您想知道转换参考代码的用法,可以阅读conv operator的代码。

python解释器使用swig来调用C++ intepreter

希望这会有所帮助。