Imported from https://github.com/kingcheng2000/3D-IWGAN