android - TFLite 转换改变模型权重

Question

我有一个自定义构建的 tensorflow 图，实现了我自己实现的 MobileNetV2-SSDLite。它在 PC 上运行良好。

但是，当我将模型转换为 TFLite（全浮点，无量化）时，模型权重发生了巨大变化。

举个例子，一个过滤器最初是 - 0.13172674179077148, 2.3185202252437188e-32, -0.003990101162344217

变为- 4.165565013885498，-2.3981268405914307，-1.1919032335281372

较大的权重值完全颠覆了我在设备上的推断。需要帮忙！:(

标签： androidtensorflowmachine-learningtensorflow-litetoco

Answer 1

您使用什么命令转换为 tflite？例如，您使用的是 toco，如果是，您使用的是什么参数？虽然我没有查看过滤器，但这里是我对 MobileNetV2-SSD 和 SSDLite 图进行微调的默认说明，并且模型一直表现良好。