时间戳

首页 > 解决方案 > Tensorflow Datasets“so2sat”数据集仅提供黑色 rgb 图像

python - Tensorflow Datasets“so2sat”数据集仅提供黑色 rgb 图像

问题描述

我成功下载了带有 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()

这显示了几乎完全黑色的图片

安装：

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

这种行为是出乎意料的，因为我预计至少会有一些白天正常的卫星 rgb 图片。我在这里错过了什么吗？

标签： pythontensorflowtensorflow-datasets

解决方案

Tensorflow Datasets一旦升级到4.0和 ，问题就可以解决Tensorflow 2.4。

推荐阅读