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

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:

主站蜘蛛池模板: 白银市| 贵德县| 哈巴河县| 鸡西市| 鄂托克前旗| 义乌市| 额尔古纳市| 依安县| 桑日县| 准格尔旗| 大兴区| 临沂市| 兴仁县| 清原| 盐城市| 浏阳市| 平原县| 玉林市| 克拉玛依市| 科尔| 阳新县| 沙河市| 綦江县| 尚志市| 海门市| 资阳市| 湾仔区| 平阴县| 彭阳县| 南京市| 道孚县| 阜新| 铁岭县| 兴国县| 安阳县| 元朗区| 嘉义市| 平泉县| 大兴区| 河北区| 东丽区|