Proyecto de aprendizaje profundo cuatro: crear un conjunto de datos personalizado, DataLoader
Directorio de artículos
prefacio
Después de construir el modelo de estructura de red neuronal, es necesario realizar iteraciones de entrenamiento de entrada en los datos, y es necesario construir el conjunto de datos y el cargador de datos del conjunto de datos. Si el conjunto de datos utilizado es el conjunto de datos incorporado de torch, puede heredar directamente el conjunto de datos y el cargador de datos de torch.utils.data para completar el entrenamiento y la evaluación.
Este proyecto explica principalmente cómo crear un conjunto de datos personalizado y DataLoader para su propio conjunto de datos.
1. Cree la clase básica de conjunto de datos
- Para construir la clase básica de conjunto de datos