在训练的过程中为了更好地了解训练情况,包括损失曲线、输入图片、输出图片、卷积核的参数分布等信息,我们可以使用Tensorboard对训练过程进行可视化。这些信息能帮助我们更好地监督网络的训练过程,并为参数优化提供方向和依据。
我们需要安装的是tensorboardX,因此,需要先安装其依赖,TensorFlow
1. 安装TensorFlow
pip3 install tensorflow
2 安装tensorboard
pip install tensorboard
3 安装tensorboardX
pip install tensorboardX
tensorboardX 的使用非常简单。首先用如下命令,在终端启动tensorboard:
tensorboard --logdir <your/running/dir> --port <your_bind_port>
使用过程中有以下注意事项:
1. log所在目录不能有中文
2. 建议直接使用chrome浏览器查看,其他浏览器可能会出现数据无法显示的情况