官术网_书友最值得收藏!

PyTorch dataset loaders

Pytorch includes data loaders for several datasets to help you get started. The torch.dataloader is the class used for loading datasets. The following is a list of the included torch datasets and a brief description:

Here is a typical example of how we load one of these datasets into PyTorch:

CIFAR10 is a torch.utils.dataset object. Here, we are passing it four arguments. We specify a root directory relative to where the code is running, a Boolean, train, indicating if we want the test or training set loaded, a Boolean that, if set to True, will check to see if the dataset has previously been downloaded and if not download it, and a callable transform. In this case, the transform we select is ToTensor(). This is an inbuilt class of torchvision.transforms that makes the class return a tensor. We will discuss transforms in more detail later in the chapter.

The contents of the dataset can be retrieved by a simple index lookup. We can also check the length of the entire dataset with the len function. We can also loop through the dataset in order. The following code demonstrates this:

主站蜘蛛池模板: 普兰店市| 岚皋县| 通河县| 墨脱县| 澜沧| 枞阳县| 嵊泗县| 黄石市| 宜城市| 宁津县| 乌兰浩特市| 甘洛县| 井研县| 上饶市| 和顺县| 云龙县| 遵化市| 临汾市| 贞丰县| 临邑县| 会东县| 北宁市| 霞浦县| 大关县| 怀来县| 民权县| 蕉岭县| 贵定县| 黄陵县| 社会| 江西省| 大庆市| 乐陵市| 肥乡县| 杭锦后旗| 海门市| 如东县| 阿克陶县| 大化| 唐海县| 金川县|