Image loader RGB transform

While loading your images, you could use Image.open(path).convert('RGB') on all images.
If you are using ImageFolder, this functionality should be already there using the default loader.

def myloader(filename):
    return Image.open(filename).convert('RGB')


dog_test = datasets.ImageFolder('/data/dog_images/test',transform = valid_transform,loader=myloader)
test = torch.utils.data.DataLoader(dataset=dog_test,batch_size=64)

Similar codes

8 votes170 views1 answer
Delete certain characters from a string python
Python
110 votes308 views6 answers
Python while
Python
0 vote170 views1 answer
K-Means Clustering in Python – 3 clusters
Python
1 vote123 views1 answer
Python ignore return value
Python
69 votes208 views6 answers
Python online compiler 3.8
Python
0 vote162 views1 answer
Python django if key exists
Python Django
2 votes159 views3 answers
Weather today
Python
1 vote357 views1 answer
Replace space with . pyhton
Python
0 vote169 views1 answer
Get or create django
Python
5 votes165 views3 answers
One hot encoding
Python