我正在使用tensorflow 2.2.0中的Keras。我有一个带有双线性插值的UpSampling2D层的模型。当尝试使用CoreML4.0转换模型时,我得到了以下错误:
ValueError:"ResizeBiliner";op:第二个输入,即输出大小,必须具有int32 类型的元素
UpSampling2D插值最接近导出时。我仍然希望插值是"双线性">
有解决办法吗?
看起来像是coremltools中的一个bug。在这种情况下,我会(暂时(将模型更改为使用";最近的";只是为了进行转换,然后修补mlmodel文件,将该层的类型更改为";双线性";。您可以使用coremltools和protobuf API来实现这一点(请参阅我的书(。