首页 > 解决方案 > 为 maskRcnn 导入 Tensorboard(Matterport - Mask RCNN)

问题描述

我目前正在尝试通过关注 Matterport repo 来实现 Mask RCNN。我对张量板的实施有疑问。

该数据集类似于 coco 数据集。在def train 下的 model.py 中,提到了tensorboard

回调 = [ keras.callbacks.TensorBoard(log_dir=self.log_dir,histogram_freq=0, write_graph=True, write_images=False)

但是对于使用 tensorboard,我还应该提到什么?当我尝试运行张量板时,它说找不到日志文件。我知道我在某些地方遗漏了一些东西!请帮帮我!

标签: python-3.xkerasdeep-learningtensorflow2.0tensorboard

解决方案


在你model.train()确保你设置custom_callbacks = callbacks参数。

如果您完全像这样指定这些参数,那么这意味着您的问题是您没有正确打开日志目录。

打开(在 Anaconda/PyCharm 内)或单独的 Python 终端并放置绝对路径(以确保它有效):

tensorboard --logdir = my_absolute_path/logs/

推荐阅读