时间戳

首页 > 解决方案 > 使用 Keras OxfordPets 代码时的 CustomMaskWarning

python - 使用 Keras OxfordPets 代码时的 CustomMaskWarning

问题描述

我正在进行一个项目，该项目采用 Oxford Pets 代码https://keras.io/examples/vision/oxford_pets_image_segmentation/并以各种方式对其进行修改。我们收到以下警告（在 Google Colab 上运行时），原来 Oxford Pets 代码也收到了警告（也在 Google Colab 上）。有没有办法更改代码以不引起警告？

这是我们在保存和训练模型时收到的警告：

/usr/local/lib/python3.7/dist-packages/tensorflow/python/keras/utils/generic_utils.py:497：CustomMaskWarning：自定义遮罩层需要配置并且必须覆盖 get_config。加载时，必须将自定义遮罩层传递给 custom_objects 参数。类别=自定义掩码警告）

标签： pythonkerastf.keras

解决方案

正如@Henrique Mendonça 所述，当我们在 Tfv2.5 上使用 Resnet 模型时会出现此警告。

这些警告不会妨碍您的代码执行，您仍然可以使用以下代码抑制这些警告

import logging, os
logging.disable(logging.WARNING)
os.environ["TF_CPP_MIN_LOG_LEVEL"] = "3"
import tensorflow as tf

推荐阅读