import tensorflow as tf

import tensorlayer as tl

from tensorlayer.prepro import *

import numpy as np

import time

X_train, y_train, X_test, y_test = tl.files.load_cifar10_dataset(

shape=(-1, 32, 32, 3), plotable=False)

def distort_img(x):

x = flip_axis(x, axis=1, is_random=True)

x = crop(x, wrg=28, hrg=28, is_random=True)

return x

s = time.time()

results = threading_data(X_train[0:100], distort_img)

print("took %.3fs" % (time.time()-s))

print(results.shape)

tl.visualize.images2d(images=results[0:9], second=0.01, saveable=True, name='cifar10_distort', dtype=np.uint8)