rassilon712 / Neural_Color_Transfer

Implementation of Neural Color Transfer between Images by PyTorch.
41 stars 8 forks source link

Instance Normalization #2

Open nieguangyang opened 5 years ago

nieguangyang commented 5 years ago

I have notice that the value range of each feature map (i.e. channel) from certain layer of VGG19 are quite different, maximum of some feature maps are hundreds and others are less than 1. Therefore quite a few feature maps are useless because of small value.

To handle this problem, I tried to apply instance normalization (instead of the original normalization method used in the paper) to each channel (feature map), and it seems work well.

rassilon712 commented 5 years ago

Sorry for the late reply, our people are busy doing other projects. We also have the concern on value of each feature map. Normalizing by its instance could be more better than usual normalization. We will also try instance normalization and report some results.