Проиллюстрировать партию в Tensorboard с помощью генератора

Question

Я использую керасы и fit_generator для тренировки моей модели.Текущая модель, которую я делаю, - это авто-кодировщик, который не дает желаемых результатов.Поэтому я хотел бы создать обратный вызов, который иллюстрирует тренировочный образ и изображение истинности в каждой 500-й партии или около того.Поэтому я хочу использовать on_batch_begin Однако я не уверен, как получить доступ к текущему пакету для создания tf.Summary.Image.

Может кто-нибудь направить мне информацию об этом или знает, как получить текущий пакет.Или это будет сделано в генераторе?Я просто не вижу, как прикрепить к этому обратный вызов.

Answer 1

Мне не удалось найти элегантное решение.Я добавил массив к обратному вызову с файлами, которые я хочу проанализировать.Затем я случайным образом выбираю одно изображение только для иллюстрации, использую текущую модель и иллюстрирую его на тензорной доске.Это работает, но я надеялся, что это будет более элегантно:)