小编Kri*_*ses的帖子

从 tf.keras.applications 导入后,Tensorflow 增强层不起作用

我目前正在使用 tf.keras.applications 中的模型进行训练。还有一个数据增强层。奇怪的是,当我从应用程序导入模型后,增强层不起作用。在导入之前,增强层确实可以工作。到底是怎么回事?

而且,这种情况直到最近新版本 TF 2.8.0 发布后才开始发生。之前一切正常。

增强层的代码是

data_augmentation = tf.keras.Sequential([
  tf.keras.layers.RandomFlip("horizontal_and_vertical"),
  tf.keras.layers.RandomRotation(0.5),
])
Run Code Online (Sandbox Code Playgroud)

我正在使用导入模型

base_model = tf.keras.applications.MobileNetV3Small(
                 input_shape=(75, 50, 3), alpha=1.0, 
                 weights='imagenet', pooling='avg', include_top=False,
                 dropout_rate=0.1, include_preprocessing=False)
Run Code Online (Sandbox Code Playgroud)

请帮助我了解发生了什么事。您可以在此笔记本上重现代码https://colab.research.google.com/drive/13Jd3l2CxbvIWQv3Y7CtryOdrv2IdKNxD?usp=sharing

python keras tensorflow

6
推荐指数
1
解决办法
1480
查看次数

标签 统计

keras ×1

python ×1

tensorflow ×1