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

Question

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

import tensorflow as tf
from tensorflow import keras

from keras.applications import mobilenet_v2

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

但是，以下代码确实有效：

from tensorflow.keras.applications import mobilenet_v2

或

from keras_applications import mobilenet_v2

上面提到的 2 种方法有效，但第一种方法无效。为什么会这样？

Answer 1

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

pip install tensorflow==2.0

希望对你有帮助。