无法保存Keras模型,这是一个错误吗?

帕特里克

我正在尝试对resnet50模型进行简单保存,但出现错误。我的代码来重现错误:

from tensorflow import keras
import keras_resnet

inputs = keras.layers.Input(shape=(None, None, 3))
resnet = keras_resnet.models.ResNet50(inputs, include_top=False, freeze_bn=True)

resnet.save("my-model")

我收到错误:“ KeyError:'输入'”。这是Bug还是keras save命令缺少我的东西?我在MacOS和ubuntu容器中尝试了该命令。结果相同。

编辑:它与resnet的官方keras实现一起工作。通过这种实现,尽管我必须更改keras-retinanet的最简单实现的resnet.py文件中的代码。具体来说,使用以下方法定义resnet:

from keras.applications.resnet import ResNet50,ResNet101,ResNet152
resnet = ResNet50(input_tensor=inputs, include_top=False)

我必须从以下位置更改骨干层的代码:

backbone_layers = {
    'C2': resnet.outputs[0],
    'C3': resnet.outputs[1],
    'C4': resnet.outputs[2],
    'C5': resnet.outputs[3]
}

到:

backbone_layers = {
    'C2': resnet.layers[-137].output,
    'C3': resnet.layers[-95].output,
    'C4': resnet.layers[-33].output,
    'C5': resnet.outputs[0]
}

我尚未对其进行测试,但我认为它应该可以工作。

我看到的唯一警告是,我不再具有freeze_bn参数。有关此参数的原因,请参见https://github.com/fizyr/keras-retinanet/issues/974我希望它不会对我的网络培训产生不利影响。

戴维·塞里

您需要使用h5等格式保存模型。我转载了您的错误,并通过以下方式修复了该错误:

resnet.save("mymodel.h5")

本文收集自互联网,转载请注明来源。

如有侵权,请联系 [email protected] 删除。

编辑于
0

我来说两句

0 条评论
登录 后参与评论

相关文章

无法完全限定枚举,这是一个错误吗?

无法在Sublime Text 3中的新标签中打开。这是一个错误吗?

Xcode 8.2中的xib文件无法正常工作,这是一个错误吗?

这是一个错误吗?Python属性无法在+ =下正常工作

我必须保存一个 json 文件来重新构建 Keras 模型吗?

BOOL属性KVC:这是一个错误吗?

numpy:正确的sqrt:这是一个错误吗?

DateTime :: createFromFormat这是一个错误吗?

错误处理-这是一个不错的模式吗?

Python 多继承。这是一个错误吗?

Kotlin RecyclerView删除项目。这是一个错误吗?

为什么bindAttr无法与残疾人一起使用?这是一个错误吗?

这是一个诺言错误吗?提交成功登录后,Selenium-webdriver测试无法找到元素

这是一个热编码吗?

这是一个纯函数吗?

这是一个封闭的例子吗?

这是一个块参数吗?

这是一个Java BUG吗?

Mutate以错误的顺序返回数据dplyr这是一个错误吗?

GMail API错误:路径“ {userId} / messages / send”缺少“ userId”参数。这是一个错误吗?

这是一个顽皮的错误吗?数值错误:对象未对齐

ToFixed in Javascript ..这是一个错误

C ++:这是一个语义错误

Java Nashorn的绑定行为不一致。这是一个错误吗?

Qt无法通过一个通道保存图像吗?

使SKScene的背景透明不起作用...这是一个错误吗?

如何控制 Android 中的可见性,这是一个错误吗?

这是一个像Learncpp.com所说的GCC错误吗?

不经意删除.ssh / authorized_keys,这是一个严重的错误吗?