本篇内容主要讲解“怎么使用pytorch准备自己的图片数据”，感兴趣的朋友不妨来看看。本文介绍的方法操作简单快捷，实用性强。下面就让小编来带大家学习“怎么使用pytorch准备自己的图片数据”吧!图片数据一般有两种情况：1、所有图片放在一个文件夹内，另外有一个txt文件显示标签。2、不同类别的图片放在不同的文件夹内，文件夹就是图片的类别。针对这两种不同的情况，数据集的准备也不相同，第一种情况可以自定义一个Dataset，第二种情况直接调用torchvision.datasets.ImageFolder来处理。下面分别进行说明：这里以mnist数据集的10000个test为例, 我先把test集的10000个图片保存出来，并生着对应的txt标签文件。先在当前目录创建一个空文件夹mnist_test, 用于保存10000张图片，接着运行代码：经过上面的操作，10000张图片就保存在mnist_test文件夹里了，并在当前目录下生成了一个mnist_test.txt的文件，大致如下：前期工作就装备好了，接着就进入正题了：自定义了一个MyDataset, 继承自torch.utils.data.Dataset。然后利用torch.utils.data.DataLoader将整个数据集分成多个批次。同样先准备数据，这里以flowers数据集为例提取链接: https://pan.baidu.com/s/1dcAsOOZpUfWNYR77JGXPHA?pwd=mwg6花总共有五类，分别放在5个文件夹下。大致如下图：我的路径是d:/flowers/.数据准备好了，就开始准备Datase免费云主机域名t吧，这里直接调用torchvision里面的ImageFolder到此，相信大家对“怎么使用pytorch准备自己的图片数据”有了更深的了解，不妨来实际操作一番吧！这里是百云主机网站，更多相关内容可以进入相关频道进行查询，关注我们，继续学习！
本篇内容介绍了“Android如何自定义加载圈”的有关知识，在实际案例的操作过程中，不少人都会遇到这样的困境，接下来就让小编带领大家学习一下如何处理这些情况吧！希望大家仔细阅读，能够学有所成！“Andro免费云主机域名id如何自定义加载圈”的内容就介绍到这里了…
免责声明：本站发布的图片视频文字，以转载和分享为主，文章观点不代表本站立场，本站不承担相关法律责任；如果涉及侵权请联系邮箱：360163164@qq.com举报，并提供相关证据，经查实将立刻删除涉嫌侵权内容。
微信扫一扫 
