application_vnd.jupyter.stderr_output_1.png
X_train.shape

X_cv.shape

X_test.shape

img_width = img_height = 224
if K.image_data_format() == 'channels_first':
    input_shape = (3, img_width, img_height)
    X_train = X_train.reshape(X_train.shape[0], 3, img_width, img_height)
    X_cv = X_cv.reshape(X_cv.shape[0], 3, img_width, img_height)
    X_test = X_test.reshape(X_test.shape[0], 3, img_width, img_height)
else:
    input_shape = (img_width, img_height, 3)
    X_train = X_train.reshape(X_train.shape[0], img_width, img_height, 3)
    X_cv = X_cv.reshape(X_cv.shape[0], img_width, img_height, 3)
    X_test = X_test.reshape(X_test.shape[0], img_width, img_height, 3)

X_train[0]