In the case of an image enhancement application, I would like to normalize each image of the batch independently before entering the network and then inverse normalize the output images using the statistics of their corresponding input image.
I tried to use InstanceNorm2d() but I’m not sure to understand how to use it in my case.
Any idea ?