Context On the Pytorch website, you'll find a bunch of tutorials. This Dataset is the one used for the dataloading one. Content It contains some faces images and a CSV file with their respective landmarks points. Acknowledgements Thanks for pytorch to provide comprehensive material to get a grasp about their framework. https://pytorch.org/tutorials/beginner/data_loading_tutorial.html Inspiration Play with pytorch or check how much simpler it is to accomplish this kind of task with a different framework.