keras中使用tensorboard可视化

只需要在fit或fit_generator前加上这几句代码即可,log_filepath前要加r解析为路径,否则会报错

log_filepath = r'logs'
tb_cb = TensorBoard(log_dir=log_filepath, write_images=1, histogram_freq=1)
cbks = [tb_cb]
model.fit_generator(generate_one_epoch(train_X, train_y, batch_size=batch_size),
                    steps_per_epoch=len(train_X)//batch_size,
                    epochs=100, callbacks=cbks)

同时在当前目录下新建一个空的logs文件夹
在这里插入图片描述
网络开始训练后进入logs文件夹,如下图,在该页面开启终端窗口(shift+鼠标右键,“在此处打开PowerShell窗口”),输入命令 tensorboard --logdir=./
在这里插入图片描述
出现下图,复制网址到浏览器打开即可
在这里插入图片描述
在这里插入图片描述

发布了83 篇原创文章 · 获赞 4 · 访问量 5363

猜你喜欢

转载自blog.csdn.net/weixin_43486780/article/details/105379235