Tensorflow“decode_png”不断打印“清理调用...”

Question

我正在使用 tensorflow 打开一些 .png 图像，并且它打开的每个图像都会打印一条烦人的消息。

def open_img(path):
    img = tf.io.read_file(path)
    img = tf.io.decode_png(img)
    return tf.image.resize(img, [IMG_HEIGHT, IMG_WIDTH])

每次我尝试打开一张图片时，它都会显示“清理已调用...”，即使是在训练时：

（此代码在 Kaggle 上运行）

张量流版本：2.6.3

请问我该如何解决这个烦人的事情？

Answer 1

将我的 TensorFlow 安装更新到 2.8 版解决了我的问题。

Answer 2

如果您在 Kaggle 上运行代码，将 Tensorflow 升级到 2.8 将破坏 GPU 的 CuDNN 依赖性。我发现将 Tensorflow 降级到 2.4.1 将删除调试消息，同时能够使用 GPU。