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 vote144 views1 answer
Python crop string
Python
7 votes141 views4 answers
How to create 2d list in python
Python
0 vote147 views2 answers
Comment faire un long commentaire en python
Python
8 votes156 views4 answers
Set index to column pandas
Python
0 vote151 views1 answer
Bash check if python package is installed
Python
13 votes139 views2 answers
Sys get current pythonpath
Python
0 vote219 views2 answers
Pd is not defined python error
Python
26 votes105 views2 answers
Pil image crop
Python
0 vote140 views2 answers
How to update sklearn
Python
0 vote142 views1 answer
KMeans
Python