Imported from https://github.com/alterzero/RBPN-PyTorch