pytorch可视化的常用方法

pytorch可视化的常用方法

1. 保存为图片

假设一个tensor的维度为[N,C,H,W],我们要把这个mini-batch格式的tensor保存到图片，那么可以这样：

• 先作一些处理工作（一般是unnormalize）
• 然后用make_grid()方法把mini-batch变换成网格状的一张大图片，结果是一个3D的tensor
• 最后用save_image()将make_grid()的3D结果保存到图片文件中去

总结起来就是

save_image(make_grid(postprocess_image(batch_tensor), nrow=8),"XXX.jpg")

2. 训练过程中保存到tensorboardX中

同样假设有一个tensor的维度为[N,C,H,W],我们要把这个mini-batch格式的tensor保存到tensorboardX中，那么可以这样：

• 先作一些处理工作（一般是unnormalize）
• 然后用make_grid()方法把mini-batch变换成网格状的一张大图片，结果是一个3D的tensor
• 最后用SummaryWriter.add_image()将make_grid()的3D结果保存到训练的中间结果中去。

总结起来就是：

SummaryWriter.add_image("image", make_grid(postprocess_image(batch_tensor), nrow=8), step)

注意add_image()方法默认3D张量的维度的CHW的格式，如果是HWC，需要显式地指名。