Tensorflow2.0中利用keras读取保存的模型时, 产生错误 [ValueError: Unknown activation function:relu6]

在使用已经训练好的mobilenet时候,keras产生错误 [ValueError: Unknown activation function:relu6]

目前博主查到了两种解决办法:

1、查看在建立模型过程中,定义 relu6 激活函数时是否使用了 tf.keras.layers.Activation(tf.nn.relu6),如果有的话,将其更改为:(记住,是所有的,更换前可以按 ctrl + F 键搜索一下当前的代码中是否还有未替换的)

tf.keras.layers.ReLU(6.)

2、如果已经使用了tf.keras.layers.Activation(tf.nn.relu6) 训练好了模型 , 那么在使用时需要用如下方法导入模型:(即在导入时备注之)

model = tf.keras.models.load_model(filepath='my_mobilenet.h5',
                                   custom_objects={'relu6': tf.keras.layers.ReLU(6.),'DepthwiseConv2D':tf.keras.layers.DepthwiseConv2D})

其他版本的keras可参考:
https://blog.csdn.net/Will_Ye/article/details/106554196

  • 2
    点赞
  • 6
    收藏
    觉得还不错? 一键收藏
  • 1
    评论
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值