torchvision.datasets
时间: 2023-11-08 11:06:26 浏览: 166
关于PyTorch源码解读之torchvision.models
5星 · 资源好评率100%
torchvision.datasets is a module in the PyTorch library that provides access to popular image datasets for machine learning tasks.
It includes datasets such as:
1. MNIST: a dataset of handwritten digits
2. CIFAR10 and CIFAR100: datasets of small color images
3. ImageNet: a dataset of over a million labeled images in 1000 categories
4. COCO: a large-scale object detection, segmentation, and captioning dataset
These datasets can be easily downloaded and used for training and testing deep learning models. The module also provides tools for data augmentation and preprocessing, making it easier to work with these datasets in PyTorch.
阅读全文