Hi,
I would like to train a Resnet-18 network on GPU using different sized images, and I do not want to resize the images nor pad. The network allows the task, as it contains an AdaptiveAvgPooling layer.
The problem I’m stuck with is how to create a DataLoader that loads different sized images.
It appears from a number of posts on this that a custom collate_fn function should be used instead of the default one in the DataLoader.
Can anybody direct me in the right direction in how to create such a DataLoader for different sized images that can be used for training ?
Thank you very much