def call(self, samples): out_arrays = [ self.preprocess(o) for o in samples.values() if o.shape[1:] == (1, 64, 64) ] return zip_tile(*out_arrays)
def call(self, samples): out_arrays = [self.preprocess(o) for o in samples.values() if o.shape[1:] == (1, 64, 64)] return zip_tile(*out_arrays)
def zip_and_save(fname, *args): clipped = list(map(lambda x: np.clip(x, 0, 1)[:, 0], args)) print(clipped[0].shape) tiled = zip_tile(*clipped) print(tiled.shape) scipy.misc.imsave(fname, tiled)