Skip to content

Glossary: Computer Vision Basics

ImageFolder

Also written: datasets.ImageFolder

A torchvision dataset that treats each subfolder name as a class label. It sorts classes alphabetically, so you should save `classes` with a trained model.

Where the book teaches it

Related terms

DatasetDataLoaderclass index