Solved by Younggee in post #12 
              
              
                Thanks alot the Custom data worked but i i had to pre-process the image manually using a loop to crop the required images in the range of 50 then i applied the transform and my custom dataset is as follows: 
num_classes = 2 
Class that reads a sequence of image paths from a directory and creates a d…