cifar10_dataset: Cifar datasets

Description Usage Arguments

Description

CIFAR10 Dataset.

Downloads and prepares the CIFAR100 dataset.

Usage

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
cifar10_dataset(
  root,
  train = TRUE,
  transform = NULL,
  target_transform = NULL,
  download = FALSE
)

cifar100_dataset(
  root,
  train = TRUE,
  transform = NULL,
  target_transform = NULL,
  download = FALSE
)

Arguments

root

(string): Root directory of dataset where directory cifar-10-batches-bin exists or will be saved to if download is set to TRUE.

train

(bool, optional): If TRUE, creates dataset from training set, otherwise creates from test set.

transform

(callable, optional): A function/transform that takes in an PIL image and returns a transformed version. E.g, transforms.RandomCrop

target_transform

(callable, optional): A function/transform that takes in the target and transforms it.

download

(bool, optional): If true, downloads the dataset from the internet and puts it in root directory. If dataset is already downloaded, it is not downloaded again.


torchvision documentation built on Aug. 17, 2021, 5:06 p.m.