加载预先训练的模型时出现Keras导入错误

时间:2020-04-14 15:43:07

标签: python tensorflow keras transfer-learning

我正在尝试在Tensorflow中使用预训练的模型。我正在使用以下代码:

import tensorflow as tf
from tensorflow import keras

from keras.applications import mobilenet_v2

我收到以下错误: ModuleNotFoundError:没有名为“ keras”的模块

但是,以下代码可以工作:

from tensorflow.keras.applications import mobilenet_v2

OR

from keras_applications import mobilenet_v2

上面提到的2种方法都可以,但是第一种方法不起作用。为什么会这样?

1 个答案:

答案 0 :(得分:0)

我已经通过使用以下命令将tensorflow降级到2.0版来解决了这个问题:

pip install tensorflow == 2.0

希望对您有帮助。