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

9 votes275 views1 answer
Jupyter notebook extensions
Python
1 vote330 views1 answer
Python fme logger
Python
2 votes382 views1 answer
Daphne heroku
Python
4 votes324 views3 answers
Write a list into csv python
Python
1 vote354 views6 answers
Python tuple methods
Python
2 votes331 views2 answers
How to find the longest string in a list in python
Python
-1 vote348 views1 answer
Smma python
Python
0 vote349 views1 answer
Flask sqlalchemy query specific columns
Python
0 vote300 views1 answer
Data don't retrieve from database view in django
Python
0 vote349 views1 answer
Simple Splash screen in pyqt5
Python