Tensorboard、TensorboardX保存多通道图像(1或3通道除外)

注意，Tensorboard、TensorboardX本身可以直接存储1通道和3通道的图像，如果不是1或3通道才需要本方法把所有的通道分成1通道

Python环境安装Tensorboard、TensorboardX

通过以下命令安装

pip install tensorboard
pip install tensorboardX

基础使用方法可以查看下面的链接

我之前写过的基础教程：如何利用Tensorboard以可视化的方式查看训练的“过程资产”（目标函数、结果、AUC等等）

具体步骤：将多通道图像存入Events(日志文件)

1 引入第三方库

from tensorboardX import SummaryWriter
import datetime # 是为了让保存结果按照日期时间命名，方便区分每次运行的结果

2 设置保存路径

也就是结果保存的路径，就是那个结果文件。

# 路径你可以写个现在不存在的，summary会自己创建的。
save_path = './logdir/' + datetime.datetime.now().strftime("%Y%m%dT%H%M%S") + '/' #注意，时间部分不要改成用“:”相连的方式，因为格式不合适。
writer = SummaryWriter(save_path);

3 处理 Batch 和 Channel

以下代码只是个样例，只保留了重要的部分。

while