Как добавить несколько изображений в коллбэк TensorBoard для Keras - PullRequest
Новая
       8

Как добавить несколько изображений в коллбэк TensorBoard для Keras

0 голосов
/ 03 ноября 2018

Я пытаюсь добавить свои изображения проверочного набора к обратным вызовам TensorBoard Keras. Следующий код работает и даже дает мне ползунок для каждого шага пакета.
Тем не менее, шаг пакета добавляется к тегу, так что он заканчивается отображением каждого изображения в виде отдельного тега (например, predictions_1, predictions_2, predictions_3).

Можно ли как-нибудь помешать добавлению тега? 

class TensorBoardWithImages(K.callbacks.TensorBoard):
    def __init__(self, **kwargs):
        super(TensorBoardWithImages, self).__init__(**kwargs)

    def on_batch_end(self, batch, logs=None):

        logs = logs or {}

        val_data = self.validation_data

        a=np.random.randint(0, val_data[0].shape[0], 10)
        tf.summary.image(name="predictions", tensor=val_data[0][a],
                         max_outputs=3)

        summary_op = tf.summary.merge_all()

        self.writer.add_summary(summary_op.eval(session=self.sess), batch)

        super(TensorBoardWithImages, self).on_batch_end(batch, logs)
...