img = Image.open(‘C:/Users/rukhm/nodule_classification/data/train/1/LIDC-IDRI-0003_74_1.png’)
im = (transforms.ToTensor()(img))
im.shape
Output:
torch.Size([1, 64, 64])
For a single image it is showing as a grayscale image.
Is it possible to check the shape of train_dataset/test_dataset?
I need to check the line where it is converted into a 3 channel image.
Thanks for the help!!