Keras EfficientNet 迁移学习代码示例不起作用

Question

我的代码几个月来都运行良好，但今天我意识到它不再运行了。在我的代码中，我只是复制粘贴了这个 keras 代码示例：https://keras.io/examples/vision/image_classification_efficientnet_fine_tuning/#example-efficientnetb0-for-stanford-dogs

所以“我的”代码如下所示：

import tensorflow as tf
import keras
from keras.layers import *
from keras import Sequential
from keras.layers.experimental import preprocessing
from keras import layers
from tensorflow.keras.applications import EfficientNetB0

img_augmentation = Sequential(
    [
        preprocessing.RandomRotation(factor=0.15),
        preprocessing.RandomTranslation(height_factor=0.1, width_factor=0.1),
        preprocessing.RandomFlip(),
        preprocessing.RandomContrast(factor=0.1),
    ],
    name="img_augmentation",
)

inputs = layers.Input(shape=(224, 224, 3))
x = img_augmentation(inputs)
outputs = EfficientNetB0(include_top=True, weights=None, classes=5)(x)

model = tf.keras.Model(inputs, outputs)
model.compile(
    optimizer="adam", loss="categorical_crossentropy", metrics=["accuracy"]
)

然而，今天当我运行我的 colab 中的这个单元格时，我收到了很多这样的警告：

WARNING:tensorflow:
The following Variables were used a Lambda layer's call (tf.compat.v1.nn.fused_batch_norm_422), but
are not present in its tracked objects:
  <tf.Variable 'top_bn/gamma:0' shape=(1280,) dtype=float32>
  <tf.Variable 'top_bn/beta:0' shape=(1280,) dtype=float32>
It is possible that this is intended behavior, but it is more likely
an omission. This is a strong indication that this layer should be
formulated as a subclassed Layer rather than a Lambda layer

这个错误：

TypeError: Cannot convert a symbolic Keras input/output to a numpy array. This error may indicate that you're trying to pass a symbolic value to a NumPy call, which is not supported. Or, you may be trying to pass Keras symbolic inputs/outputs to a TF API that does not register dispatching, preventing Keras from automatically converting the API call to a lambda layer in the Functional Model.

我认为google colab更新了keras和tensorflow，现在它们都是2.5.0版本

我怎样才能让我的代码再次工作？

Answer 1

所以经过一些研究，我意识到它来自进口：

from tensorflow.keras.applications import EfficientNetB0

我不知道为什么，但它没有抛出任何错误，而是破坏了整个代码。相反，我必须导入：

from keras.applications.efficientnet import EfficientNetB0

而且效果很好。

Answer 2

您不应混合使用 tf 2.x 和独立 keras。您应该按如下方式导入您的库，这样您就不会遇到任何问题。

import tensorflow as tf
from tensorflow.keras.layers import *
from tensorflow.keras import Sequential
from tensorflow.keras.layers.experimental import preprocessing
from tensorflow.keras import layers
from tensorflow.keras.applications import EfficientNetB0

img_augmentation = Sequential(
    [
        preprocessing.RandomRotation(factor=0.15),
        preprocessing.RandomTranslation(height_factor=0.1, width_factor=0.1),
        preprocessing.RandomFlip(),
        preprocessing.RandomContrast(factor=0.1),
    ],
    name="img_augmentation",
)

inputs = layers.Input(shape=(224, 224, 3))
x = img_augmentation(inputs)
outputs = EfficientNetB0(include_top=True, weights=None, classes=5)(x)

model = tf.keras.Model(inputs, outputs)

Keras EfficientNet 迁移学习代码示例不起作用

Keras EfficientNet transfer learning code example not working

keras

tensorflow

google-colaboratory

transfer-learning