*: Equal Contribution
The repo contains TensorFlow and PyTorch Implementation of Positional Normalization (PONO) and Moment Shortcut (MS), described in the paper Positional Normalization.
This image illustrates how PONO differs from some popular nomalization methods:
In the figure, each subplot shows a feature map tensor, with B as the batch axis, C as the channel axis, and (H, W) as the spatial axis. The entries colored in green or blue (ours) are normalized by the same mean and standard deviation. Unlike previous methods, our method processes each position independently.
It's suggested to put PONOs right after convolution operations in the encoder (former layers) and MSs right before convolution operations in the decoder (latter layers). One may use a ConvNet taking μ and σ extracted from PONO as inputs to generate β and γ for MS, which we refer as Dynamic Moment Shortcut (DMS). Here we show the illustration.
Improved AOD-Net Pytorch Version
If you find this repo useful, please cite:
@inproceedings{li2019positional,
title={Positional Normalization},
author={Li, Boyi and Wu, Felix and Weinberger, Kilian Q and Belongie, Serge},
booktitle={Advances in Neural Information Processing Systems},
pages={1620--1632},
year={2019}
}