python - 使用 Keras OxfordPets 代码时的 CustomMaskWarning

Question

我正在一个项目中使用牛津宠物代码 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。加载时，自定义 mask 层必须传递给 custom_objects 参数。
类别=自定义掩码警告)

Answer 1

正如@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