如何检查tflite文件中的量化权重

时间:2018-10-26 03:16:56

标签: tensorflow tensorflow-lite

当我通过tensorflow中的精简模块对模型进行量化时,无法检查具有quantized的权重值。有没有办法查看.tflite文件中的这些值?还是有什么方法可以解析.tflite文件?

2 个答案:

答案 0 :(得分:1)

有些神经网络可视化工具还可以提供检查文件的界面。我一直在使用Netron。您可以单击感兴趣的图层的“权重”选项卡以查看数据。我还没有尝试过,但是当您在右侧栏中查看权重/偏差时,似乎有一个软盘保存图标。

数据格式是Google FlatBuffer,由架构文件here定义。如果您想对数据做某事(例如以其他格式输出),则可能更喜欢这样做。我发现自己使用schema.fbs文件对输出进行了解析,以匹配传入的CNN的Netron的输出。您可以查看FlatBuffers文档here

答案 1 :(得分:1)

第一个答案中的

here是指导,如何创建.tflite模型的json视图。在那里您可以看到量化值