Custom-Dataset-and-Dataloader-in-Torch

Sample code for creating a modified dataset class in torch that can be used for training.

Requirements:

1. python2 / 3
2. opencv -> pip3 install opencv-python
3. numpy -> pip3 install numpy
4. torch -> pip3 install torch

A detailed description of the code can be found here: https://vineeth2309.medium.com/creating-a-custom-dataset-and-dataloader-in-pytorch-76f210a1df5d