Imported from https://github.com/Schlumberger/pixel-constrained-cnn-tf