next(iter(dataloader))的详解

        经常在自己建立好数据集后需要验证一下，自己的图片和标签能不能对上，经常用到next(iter(dataloader))命令。iter(dataloader)的结果是一个generator生成器，我们需要用循环或者next()来获取其中的数据。

       分别打印出返回值的大小、类别。

        可知其返的是一个batch的图片和标签数据，数据类型均为tensor，图片维度为（c，h，w）。

        image[0].squeeze()表示去掉维度为1的 。

        其中在plt.imshow()中需要传入的参数为数组类型，格式为RGB，因此在显示图片的时候需要将图片格式维度转换为数组的RGB，输入命令：

sample = sample.permute((1, 2, 0)).numpy()

最终输出图片结果：