Creating a custom Dataset class in Pytorch

Before creating deep-learning models and training them, the dataset needs to be prepared to be fed into the model training scheme. This article focuses on building a custom dataset class, with the functionality of converting images to tensors. This p...