使用 Keras OxfordPets 代码时的 CustomMaskWarning

Question

我正在进行一个项目，该项目正在采用 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 参数。类别=自定义掩码警告）

Answer 1

正如 @Henrique Mendon\xc3\xa7a 提到的，当我们在 Tfv2.5 上使用 Resnet 模型时，会出现此警告。

\n

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

\n

import logging, os\nlogging.disable(logging.WARNING)\nos.environ["TF_CPP_MIN_LOG_LEVEL"] = "3"\nimport tensorflow as tf\n

\n