pytorch读取图像数据的方法
本篇内容主要讲解“pytorch读取图像数据的方法”,感兴趣的朋友不妨来看看。本文介绍的方法操作简单快捷,实用性强。下面就让小编来带大家学习“pytorch读取图像数据的方法”吧!
成都创新互联专注于企业成都全网营销推广、网站重做改版、七台河网站定制设计、自适应品牌网站建设、H5开发、商城系统网站开发、集团公司官网建设、成都外贸网站建设公司、高端网站制作、响应式网页设计等建站业务,价格优惠性价比高,为七台河等各大城市提供网站开发制作服务。
pytoech读取图像数据的方法和步骤是固定的,一般是第一步先设置预处理方法,第二步是设置数据类,也就是确定要读取的数据是什么,第三步是读取数据,就是按照设置的要求把数据读取出来。
举个常见例子:
# 第一步:预处理 transform = transforms.Compose([transforms.ToTensor(), transforms.Normalize(mean=[0.5], std=[0.5])]) # 第二步:设置数据类 train_dataset = datasets.MNIST(root='D:\code\mnist', train=True, download=False, transform=transform) test_dataset = datasets.MNIST(root='D:\code\mnist', train=False, download=False, transform=transform) # 第三步:读取数据 train_loader = DataLoader(train_dataset, shuffle=True, batch_size=batch_size) test_loader = DataLoader(test_dataset, shuffle=False, batch_size=batch_size)
到此,相信大家对“pytorch读取图像数据的方法”有了更深的了解,不妨来实际操作一番吧!这里是创新互联网站,更多相关内容可以进入相关频道进行查询,关注我们,继续学习!
网页题目:pytorch读取图像数据的方法
新闻来源:http://lswzjz.com/article/pojoeh.html