分类：tensor类型

2021
07-08

pytorch 带batch的tensor类型图像显示操作

项目场景pytorch训练时我们一般把数据集放到数据加载器里，然后分批拿出来训练。训练前我们一般还要看一下训练数据长啥样，也就是训练数据集可视化。那么如何显示dataloader里面带batch的tensor类型的图像呢？显示图像绘图最常用的库就是matplotlib：pipinstallmatplotlib显示图像会用到matplotlib.pyplot.imshow方法。查阅官方文档可知，该方法接收的图像的通道数要放到后面：数据加载器中数据的维度是[B,C,H,W]，我们每次只... 继续阅读 >