Imported from https://github.com/xiaogangLi/tensorflow-VTN