值错误: "ResizeBilinear" op: 第二个输入(即输出大小)必须具有 int32 类型的元素



我正在使用tensorflow 2.2.0中的Keras。我有一个带有双线性插值的UpSampling2D层的模型。当尝试使用CoreML4.0转换模型时,我得到了以下错误:

ValueError:"ResizeBiliner";op:第二个输入,即输出大小,必须具有int32 类型的元素

UpSampling2D插值最接近导出时。我仍然希望插值是"双线性">

有解决办法吗?

看起来像是coremltools中的一个bug。在这种情况下,我会(暂时(将模型更改为使用";最近的";只是为了进行转换,然后修补mlmodel文件,将该层的类型更改为";双线性";。您可以使用coremltools和protobuf API来实现这一点(请参阅我的书(。

相关内容

最新更新