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



我正在尝试在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

上面提到的两种方法有效,但第一种方法无效。为什么会发生这种情况?

我使用以下命令将tensorflow降级到2.0版,从而解决了这个问题:

pip安装tensorflow==2.0

我希望它能帮助你。

最新更新