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

0 vote136 views1 answer
Travis deployment script for django applications to heroku
Python Django
14 votes107 views4 answers
How to detect keypress in python
Python
1 vote98 views1 answer
Como unir dos listas python
Python
0 vote151 views1 answer
Python get item from set
Python
3 votes53 views1 answer
Parquet to csv
Python
0 vote96 views1 answer
How do you make plot show with matplotlib ion method
Python
0 vote140 views1 answer
Setting up a django prostgress project
Python Django
7 votes132 views2 answers
Add image to jupyter notebook in markdown
Python
0 vote100 views1 answer
Pls work
Python
0 vote102 views1 answer
Concat and Append DFs Python
Python