Keras Tensorboard 回调不写入图像答案

【问题标题】：Keras Tensorboard callback not writing imagesKeras Tensorboard 回调不写入图像
【发布时间】：2017-07-28 03:45:51
【问题描述】：

我正在尝试使用 Tensorboard 可视化我的 Keras 模型的权重。这是我正在使用的模型：

model = Sequential([
    Conv2D(filters=32, kernel_size=(3,3), padding="same", activation='relu', input_shape=(40,40,3)),
    MaxPooling2D(pool_size=(2, 2)),
    Conv2D(filters=64, kernel_size=(5,5), padding="same", activation='relu'),
    MaxPooling2D(pool_size=(2, 2)),
    Flatten(),
    Dense(1024, activation='relu'),
    Dropout(0.5),
    Dense(43, activation='softmax'),
])
model.compile(optimizer='sgd',
              loss='categorical_crossentropy',
              metrics=['accuracy'])

我正在用这个电话进行训练：

model.fit_generator(
    ...
    callbacks = [
        ModelCheckpoint('models/gtsrb1-{epoch}.hdf5', verbose=1, save_best_only = True),
        TensorBoard(log_dir='tblogs/', write_graph=True, write_grads=True, write_images=True),
        EarlyStopping(patience=5, verbose=1),
    ],)

但是，当我启动 TensorBoard 时，我得到的是：

标量和图形看起来不错，所以这不是错误logdir 的问题。我在这里做错了什么？

【问题讨论】：

同样的问题。你有没有得到任何地方？

标签： keras tensorboard

【解决方案1】：

您需要添加histogram_freq=x，其中x应该不为零，以便启用图像的写入。

但是如果你这样做，它可能仍然会失败，这取决于 Keras 的版本（参见https://github.com/fchollet/keras/issues/6096）

【讨论】：