Keras导入Colab

问题描述

我正在尝试使用keras和_obtain_input_shape函数,这似乎绝对是一团糟。现在,以下错误不断出现。

ImportError:您需要先import keras才能使用keras_applications。例如,您可以执行以下操作:

import keras
from keras_applications import vgg16

或者,最好是等价的公式:

from keras import applications

该消息非常方便且有用,但是,如果我按照指示执行该操作(任一种选择),ImportError仍然存在。有人遇到过这种情况么? 我的Keras版本是2.4.0。

解决方法

可以直接从tensorflow导入Keras:

from tensorflow import keras

然后

app = tensorflow.keras.applications.VGG16(
    include_top=True,weights="imagenet",input_tensor=None,input_shape=None,pooling=None,classes=1000,classifier_activation="softmax",)