/unsplash-iterable-dataset

An iterable dataset that downloads image-text pairs from Unsplash.

Primary LanguagePythonMIT LicenseMIT

unsplash-iterable-dataset

An iterable dataset that downloads image-text pairs from Unsplash.

Installation

Install with pip.

pip install git+https://github.com/oelin/unsplash-iterable-dataset

Usage

import torch
from unsplash_iterable_dataset import UnsplashIterableDataset

dataset = UnsplashIterableDataset(
  topic='animals',
  limit=1_000_000,
)

dataloder = torch.utils.data.DataLoader(dataset=dataset, batch_size=4, shuffle=False)