Have a look at the Dataloading tutorial 19. In this tutorial you’ll find an example of the  Dataset class and how to create your own Dataset.
Let me know, if this helps or if you need more information.