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

2 votes125 views4 answers
Python how to draw a circle
Python
32 votes120 views6 answers
If and else in python
Python
1 vote133 views1 answer
Python find in largest 3 numbers in an array
Python
0 vote137 views2 answers
How to kill python process started by excel
Python
82 votes124 views5 answers
Python requirements.txt
Python
1 vote112 views1 answer
Openpyxl delete column by name
Python
2 votes127 views4 answers
How to count special values in data in python
Python
0 vote124 views1 answer
Python how to add a string to the beginning of a list
Python
1 vote126 views1 answer
Sort and remove duplicates list python
Python
0 vote125 views1 answer
Ist und ein satzglied
Python