Imported from https://github.com/taki0112/SDIT-Tensorflow