Hi
I am reading  an image using both matplotlib imread and imagefolder dataloader.
When plotting the output looks okay for imread but the one with imagefolder output is multiplied to 9 for one image though the shape looks okay (9 of the same images with different channels I suppose as its not color anymore).
TEST_DATA_PATH = “./data”
TRANSFORM_IMG = transforms.Compose([
transforms.ToTensor(),
])
test_data = torchvision.datasets.ImageFolder(root=TEST_DATA_PATH, transform=TRANSFORM_IMG)
test_data_loader  = data.DataLoader(test_data, batch_size=2, shuffle=False)
for i, data in enumerate(test_data_loader):
images,labels = data
plt.imshow(images[0])
plt.show()