load dataset

code_3.py

# Image Preprocessing

start_time = time.time()

dataset = Images('flowers', 'flowers', config=['resize=(128,128)', 'store'])

print("Preprocessing: ", int(time.time() - start_time), "secs")
print("Classes", dataset.classes, "Images:", dataset.count)

curr_mem = psutil.virtual_memory().used
print("Memory Used: %.2f GB" % ((curr_mem - start_mem) / GB))

# Extract dataset into memory
dataset.split = 0.0 # use the entire dataset
X_train, X_test, Y_train, Y_test = dataset.split

# Release unused memory
gc.collect()
time.sleep(5)