Skip to content

Glossary: Computer Vision Basics

Dataset

Also written: torch.utils.data.Dataset, custom Dataset

A PyTorch class with `__len__` and `__getitem__` that returns one example at a time. You write your own, like `DamageDataset`, when labels live in a file instead of folder names.

Where the book teaches it

Related terms

ImageFolderDataLoader