Функция summary.image

Функция summary.image записывает изображения в лог событий TensorBoard. Первым параметром передается имя тега, под которым изображение будет отображаться на панели Images. Вторым параметром передается тензор с данными изображения. Третьим параметром является номер шага (по умолчанию 0). Четвертым - максимальное количество изображений для отображения (по умолчанию 3). Пятым - описание. Функция возвращает объект Summary, который затем записывается в файл с помощью метода writer.add_summary.

Синтаксис

tf.summary.image(name, data, step=None, max_outputs=3, description=None)

Пример

Давайте создадим простое изображение и запишем его в лог TensorBoard:

import tensorflow as tf

# Create a 4D tensor representing a single image
# Shape: [batch, height, width, channels]
image = tf.constant([[[[1.0], [2.0]], [[3.0], [4.0]]]])

# Create a file writer
writer = tf.summary.create_file_writer('logs/image_example')

with writer.as_default():
	tf.summary.image('my_image', image, step=1)

print('Image written to log')

Результат выполнения кода:

"Image written to log"

Пример

Давайте запишем пакет из нескольких изображений с разными значениями пикселей:

import tensorflow as tf
import numpy as np

# Create a batch of 2 images, each 3x3 with 1 channel
# Values are random but reproducible with seed
tf.random.set_seed(0)
images = tf.random.uniform([2, 3, 3, 1], minval=0, maxval=1)

writer = tf.summary.create_file_writer('logs/batch_images')

with writer.as_default():
	tf.summary.image('batch_images', images, step=2, max_outputs=2)

print('Batch of images written to log')

Результат выполнения кода:

"Batch of images written to log"

Смотрите также

функцию summary.create_file_writer,
которая создает файл для записи событий TensorBoard
функцию summary.scalar,
которая записывает скалярные значения в TensorBoard
функцию summary.histogram,
которая записывает гистограммы в TensorBoard
функцию summary.text,
которая записывает текстовые данные в TensorBoard