we normalize general RGB images using mean and std of imagenet data.
Similarly precompute mean and std of each channel across the whole dataset - just once - dont need to save the normalize data, just save the mean and std.
And in your dataset’s __getitem__ sub and divide the current sample by precomputed mean and std.
I hope this helps