Creating a custom dataset along with data loaders