Imported from https://github.com/chenyilun95/tf-cpn