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，需要顯式地指名。
