Tensorflow数据集"so2sat“数据集仅提供黑色rgb图像。

Question

我成功地下载了带有tensorflow数据集的so2sat LCZ42数据集。然而，当采样rgb图像(预期范围为0-255)时，它们几乎都是完全黑色的。重现：

import tensorflow_datasets as tfds
import numpy as np
import matplotlib.pyplot as plt

train_ds, test_ds = tfds.load('so2sat', split=['train', 'validation'])
images = []
for i, example in enumerate(train_ds.take(10000)):
  image, label = example["image"], example["label"]
  # only show when perhaps not black
  if image.numpy().max()>150:
    print(i)
    print(image.numpy().max())
    images.append(image)
    plt.imshow(image.numpy())
    plt.show()

This shows allmost entirely black pictures

已安装：

tensoflow-datasets==1.2.0
tensorflow==2.1.0
matplotlib==3.3.2
numpy==1.19.2

这种行为是意想不到的，因为我预计至少会有一些白天正常的卫星rgb照片。我是不是漏掉了什么？

Answer 1

将Tensorflow Datasets升级到4.0和Tensorflow 2.4后，即可解决此问题。