dogs_vs_cats_dataset: Dog vs cats dataset

In mlverse/torchdatasets: Ready to Use Extra Datasets for Torch

Dog vs cats dataset

Description

Prepares the dog vs cats dataset available in Kaggle here

Usage

dogs_vs_cats_dataset(
  root,
  split = "train",
  download = FALSE,
  ...,
  transform = NULL,
  target_transform = NULL
)

Arguments

root	path to the data location
split	string. 'train' or 'submission'
download	whether to download or not
...	Currently unused.
transform	function that takes a torch tensor representing an image and return another tensor, transformed.
target_transform	function that takes a scalar torch tensor and returns another tensor, transformed.

Value

A torch::dataset() ready to be used with dataloaders.

Examples

if (torch::torch_is_installed() && FALSE) {
dogs_cats <- dogs_vs_cats_dataset("./data", token = "path/to/kaggle.json",
                                  download = TRUE)
length(dogs_cats)
}

mlverse/torchdatasets documentation built on April 5, 2025, 3:12 a.m.