Yes just as i thought. Thank You.
I have one  ore issue. How to skip if an image is not available and fetch another image
Here is my code
def __getitem__(self, index):
    imagePath = self.listImagePaths[index]
    
    imageData = Image.open(imagePath).convert('RGB')
    imageLabel= torch.FloatTensor(self.listImageLabels[index])
    
    if self.transform != None: imageData = self.transform(imageData)
    
    return imageData, imageLabel`

some image names in csv are not available which rises error.