Is it necessary to Create a Custom dataset? what are the pros?

While some public datasets follow a specific pattern (and thus can be loaded using a torchvision dataset), a custom Dataset gives you the ability to load your custom dataset using any logic you want. I.e. you don’t have to follow a specific folder structure, can apply transformations using different conditions, could create the target using a custom logic etc.

1 Like