如何使用PyTorch加载自定义的图像数据集实例？

本文共计1420个文字，预计阅读时间需要6分钟。

深度学习算法，如卷积神经网络，通常依赖现成的数据集和代码。但当自己开始撰写论文并使用自己的图像数据集时，才会发现许多新手的挑战。

之前学习深度学习算法，都是使用网上现成的数据集，而且都有相应的代码。到了自己开始写论文做实验，用到自己的图像数据集的时候，才发现无从下手 ，相信很多新手都会遇到这样的问题。

参考文章www.jb51.net/article/177613.htm

下面代码实现了从文件夹内读取所有图片，进行归一化和标准化操作并将图片转化为tensor。最后读取第一张图片并显示。