
image_x = 50
image_y = 50
train_y = np_utils.to_categorical(y_train)
test_y = np_utils.to_categorical(y_test)
train_y = train_y.reshape(train_y.shape[1], train_y.shape[2])
test_y = test_y.reshape(test_y.shape[1], test_y.shape[2])
x_train = x_train.reshape(x_train.shape[0], 50, 50, 1)
x_test = x_test.reshape(x_test.shape[0], 50, 50, 1)
print("x_train shape: "+ str(x_train.shape))
