TFLite Conversion更改模型权重

时间:2018-10-09 17:43:55

标签: android tensorflow machine-learning tensorflow-lite toco

我有一个自定义的构建张量流图,该图实现了我自己实现的MobileNetV2-SSDLite。在PC上运行正常。

但是,当我将模型转换为TFLite(全部浮动,没有量化)时,模型权重将急剧变化。

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

成为- 4.165565013885498, -2.3981268405914307, -1.1919032335281372

较大的权重值完全摆脱了我在设备上的推断。需要帮忙! :(

1 个答案:

答案 0 :(得分:0)

您使用什么命令转换为tflite?例如,您使用的是toco,如果使用的是什么参数?虽然我没有查看过滤器,但是here are my default instructions用于微调MobileNetV2-SSD和SSDLite图形,并且模型运行良好。