def show_batch () не показывает мои поезда - PullRequest
Новая
       5

def show_batch () не показывает мои поезда

0 голосов
/ 08 февраля 2020

Я выполняю предварительную обработку данных в TensorFlow, следуя инструкциям на их веб-сайте: https://www.tensorflow.org/tutorials/load_data/images

Однако, превратив изображения в тензоры и присвоив каждому из них соответствующую метку, я не могу построить их.

Я загружаю пару (изображение, метку) параллельно: labeled_ds = list_ds.map(process_path, num_parallel_calls=AUTOTUNE). Затем я проверяю форму изображений и соответствующую метку: 

for image, label in labeled_ds.take(1):
  print("Image shape: ", image.numpy().shape)
  print("Label: ", label.numpy())

И получаю Image shape: (80, 80, 3) Label: [False False True False], как и ожидалось.

Затем я определяю следующую функцию для подготовки набора данных в пакетах для обучения: 

def prepare_for_training(ds, cache=True, shuffle_buffer_size=1000):
  ds = ds.batch(100)
  ds = ds.prefetch(buffer_size=AUTOTUNE)

  if cache:
    if isinstance(cache, str):
      ds = ds.cache(cache)
    else:
      ds = ds.cache()

  ds = ds.shuffle(buffer_size=shuffle_buffer_size)
  ds = ds.repeat()

  return ds



train_ds = prepare_for_training(labeled_ds)
image_batch, label_batch = next(iter(train_ds))

Но когда я хочу отобразить каждое изображение с его меткой, используя plt.show(), изображения не отображаются. Вот как я это делаю: 

def show_batch(image_batch, label_batch):
    plt.figure(figsize=(10,10))
    for n in range(25):
        ax = plt.subplot(5,5,n+1)
        plt.imshow(image_batch[n])
        plt.title(CLASS_NAMES[label_batch[n]==1][0].title())
        plt.axis('off')
        return plt.show()


show_batch(image_batch.numpy(), label_batch.numpy())

Есть какие-нибудь подсказки, почему мои изображения могут не отображаться?

Ответы [ 2 ]

1 голос
/ 08 февраля 2020

Почему вы добавили return plt.show()? Официальный учебник не имеет этой строки (и она работает!) - и, вызывая return на первой итерации цикла for, вы не разрешаете отображать изображения.

0 голосов
/ 08 февраля 2020

return plt.show был внутри для l oop. Это должно быть снаружи так: 

def show_batch(image_batch, label_batch):
    plt.figure(figsize=(10,10))
    for n in range(25):
        ax = plt.subplot(5,5,n+1)
        plt.imshow(image_batch[n])
        plt.title(CLASS_NAMES[label_batch[n]==1][0].title())
        plt.axis('off')

    return plt.show()
...