我试图运行基于Keras / TensorFlow的机器学习代码。在tensorflow
环境中运行时,遇到以下错误:
from keras_applications.mobilenet import relu6
ImportError: cannot import name 'relu6'
我该如何解决?
答案 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")