Deep Learning Project Four: Create a custom dataset dataset, DataLoader
Article Directory
foreword
After building the neural network structure model, it is necessary to perform input training iterations on the data, and it is necessary to build the dataset and DataLoader of the data set. If the dataset used is torch's built-in dataset, you can directly inherit the dataset and DataLoader of torch.utils.data to complete training and evaluation.
This project mainly explains how to build a custom dataset dataset and DataLoader for your own dataset.
1. Build the basic class of dataset
- To build the basic class of dataset