Deep Learning Project Four: Create a custom dataset dataset, DataLoader

Deep Learning Project Four: Create a custom dataset dataset, DataLoader



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

Guess you like

Origin blog.csdn.net/m0_60890175/article/details/131749936