将 AutoML TFLite 编译到 Edge TPU

Posted

技术标签:

【中文标题】将 AutoML TFLite 编译到 Edge TPU【英文标题】:Compile AutoML TFLite to Edge TPU 【发布时间】:2021-07-04 00:35:05 【问题描述】:

我使用 Vision AutoML 训练了一个专用模型。我的数据集包含全景图片(3000x1500px)。 该模型在 tensorflow.lite (2.5.0) 上运行良好。但我想在 Coral Edge TPU 上加载它。

当我使用 edge_compiler 时,出现此错误:

Edge TPU 编译器版本 15.0.340273435 loc("reshape_output"): error: 'tfl.reshape' op 量化参数违反了相同的尺度约束:!quant.uniform vs. !quant.uniform

内部编译器错误。中止!

有什么想法吗?

【问题讨论】:

【参考方案1】:

只需要降级到edgetpu_compiler 14.1

【讨论】:

以上是关于将 AutoML TFLite 编译到 Edge TPU的主要内容,如果未能解决你的问题,请参考以下文章

Google AutoML 对象检测数据拆分错误

具有委托 gpu 的 TFlite 给出错误的结果

将 .tflite 转换为 .pb

云极DataCanvas 发布 2021 AutoML 自动机器学习白皮书|附完整下载

将 keras 模型从 pb 文件转换为 tflite 文件

无法将 Keras 模型转换为 tflite