TensorBoardX add_image()输出图片与torchvision.transforms.Normalize()标准化需要注意的地方

TensorBoardX add_image()输出图片与torchvision.transforms.Normalize()标准化

版权归属：

• https://blog.csdn.net/halchan
• chanhal@outlook.com

更多关注：

• https://github.com/chanhal
• https://www.zhihu.com/people/chanhal

1、TensorBoardX add_image()输出图片需要注意的地方

pytorch中调用TensorBoardX显示图片的方法如下：

from tensorboardX import SummaryWriter

writer = SummaryWriter(log_dir)
# display image
writer.add_image('label_name', img, global_step=total_step)

首先，label_name必须是一个唯一的名字，特别需要注意的是防止与add_scale()中的名字重复，这样子会出现莫名的错误；

其次，参数img可以是numpy or torch.Tensor，需是一张图像的三维矩阵，格式是[C, H, W]。如果是 torch.Tensor时，通常是选择batch中的一张图片img =