gan_test.py 文件源码

def __init__(self, batch_size, rangex=1.0, z_dim=10,img_h=64, img_w=64, imgs_dir='./'):

        self.batch_size = batch_size
        self.img_h = img_h
        self.img_w = img_w

        self.rangex = rangex
        self.z_dim = z_dim

        #self.imgs_dir = imgs_dir
        self.images_ = glob.glob('{}/*.jpg'.format(imgs_dir))
        self.perm_ = np.random.permutation(np.range(len(self.images_)))
        self.cur_ = 0