Tensorflow .pb 文件到 coreml 模型：“不支持的操作类型：AddV2”

Question

我正在尝试将通过重新训练 mobilenet 架构获得的 tensorflow 模型（.pb 文件）转换为 coreml 模型。 最初我使用 tensorflow 1.14.0一切正常。 但是当我将 tensorflow 升级到2.0.0并重新训练模型时，由于弃用，python 正在添加“AddV2”操作而不是“Add”。 因此，我在转换过程中遇到错误

不支持的操作类型：AddV2

我该怎么办？ 我是否必须更改操作或调整转换代码以支持该操作。

Answer 1

尽管 tfcoreml==1.1.0 说它已经添加了 AddV2 操作来支持 tf 2.0.0，但它并没有。

https://github.com/tf-coreml/tf-coreml/releases/tag/1.1.0

证明：

https://github.com/tf-coreml/tf-coreml/blob/master/tfcoreml/_ops_to_layers.py

这可以通过添加 AddV2 来解决，它在上述脚本中执行元素操作。