ImportError:无法导入名称“ relu6”

时间:2018-08-08 21:47:56

标签: python tensorflow machine-learning keras pip

我试图运行基于Keras / TensorFlow的机器学习代码。在tensorflow环境中运行时,遇到以下错误:

from keras_applications.mobilenet import relu6
ImportError: cannot import name 'relu6'

我该如何解决?

3 个答案:

答案 0 :(得分:2)

从GitHub上查看此answer

您需要使用CustomObjectScope来导入relu6

答案 1 :(得分:1)

MobileNet已移至keras-applications

对于keras 2.2.4

from keras.layers import ReLU
from keras.layers import DepthwiseConv2D

答案 2 :(得分:0)

你可以像这样创建你的 relu6:

relu6 = keras.layers.ReLU(max_value=6, name="ReLU6")