I’ve never used the built-in transforms system before, but it seemed really useful, so I wanted to give it a try.
My dataloader grabs a 4D batch (batch, channels, x, y) np.ndarray from an HDF5 file, I believe this is a very common setup for many 2D networks.
However, the transforms system can only process 3D (x,y,channel) shaped PIL-images?
So to use this system I’d have to build a for loop in which np.ndarray is converted to PIL image, transpose, transform and convert back to np?
This sounds so ridiculous I can hardly believe it. I would have had expected that this torch module would at least accept torch tensors.