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

23 votes191 views2 answers
How to check in which directory python in running
Python
0 vote221 views1 answer
Create requirements file and load it in new envirnment.
Python
0 vote210 views1 answer
Parquet to dataframe
Python
4 votes218 views2 answers
How to print a word in different line in python
Python
5 votes191 views2 answers
Insert function in list
Python
54 votes196 views3 answers
Getting size of list in python
Python
0 vote201 views1 answer
Get random line from file python
Python
0 vote232 views1 answer
__pycache__
Python
0 vote198 views1 answer
Create a django and react readonly web app
Python
0 vote234 views1 answer
"setFlag(QGraphicsItem.ItemIsMovable)" crash
Python