解决由于历史原因解析tflite失败的问题

Zhao Dongyu

于 2024-03-19 10:39:37 发布

阅读量1.4k

点赞数 12

CC 4.0 BY-SA版权

分类专栏： tensorflow 文章标签： tensorflow opcode 历史遗留

本文链接：https://blog.csdn.net/ZhaoDongyu_AK47/article/details/136830036

文章讲述了作者在解析特定TensorFlowLite模型时遇到的问题，原因是schema版本与TensorFlow版本不匹配。解决过程包括升级TensorFlow版本、下载并使用匹配的schema.fbs文件，以及在解析过程中正确处理opcode和builtin_code。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

文章目录

0. 背景
1. tflite 历史遗留问题
2. schema
3. flatbuffers 编译器
4 问题未解决
终极解决方案

写在最前面：解决方法是升级tensorflow版本，重新生成tflite。

0. 背景

今天遇到了这么一个问题：

  File "***/scripts/tflite/schema.py", line 9366, in Tensors
    x = self._tab.Indirect(x)
  File "***/anaconda3/lib/python3.8/site-packages/flatbuffers/table.py", line 46, in Indirect
    return off + encode.Get(N.UOffsetTFlags.packer_type, self.Bytes, off)
  File "***/anaconda3/lib/python3.8/site-packages/flatbuffers/encode.py", line 26, in Get
    return packer_type.unpack_from(memoryview_type(buf), head)[0]
struct.error: unpack_from requires a buffer of at least 1574732 bytes for unpacking