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

31 votes103 views3 answers
Pandas order dataframe by column of other dataframe
Python
0 vote112 views1 answer
If equal to key return value python
Python
2 votes103 views2 answers
Get title attribute beautiful soup
Python
-1 vote107 views1 answer
Rewrite the equation shown in Figure 2.4 as a Python expression and get the result of the equation: Pay special attention to the order of operations.
Python
1 vote145 views1 answer
Python remove one element from numpy array
Python
0 vote102 views2 answers
Fatal error: Python.h: No such file or directory 45 | #include <Python.h> | ^~~~~~~~~~ compilation terminated. error: command 'gcc' failed with exit status 1
Python
1 vote99 views1 answer
Python load pandas from pickle
Python
3 votes109 views3 answers
How to take matrix input in python
Python
0 vote140 views1 answer
Renpy quickstart
Python
0 vote142 views1 answer
How to extract field values in list from queryset in django
Python