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

2 votes183 views1 answer
How to make file to curent dir python
Python
4 votes209 views3 answers
Django Change Password
Python Django
0 vote168 views1 answer
Python list pop equivalent
Python
14 votes197 views2 answers
Check dictionary is empty or not in python
Python
24 votes184 views1 answer
Creating a new enviroment in conda
Python
26 votes165 views6 answers
Python push to list
Python
0 vote161 views1 answer
Pandas new column with loc
Python
6 votes158 views2 answers
How to press enter python keyboard
Python
10 votes194 views4 answers
Add row to dataframe with index
Python
1 vote196 views1 answer
Python selenium partial class name
Python