image_colorizer_alpha_version_train.py

def train_model(image_name, model):
    image = img_to_array(load_img(image_name))
    image = np.array(image, dtype=np.float32)

    X = rgb2lab(1.0/255*image)[:, :, 0]
    Y = rgb2lab(1.0/255*image)[:, :, 1:]
    Y /= 128
    X = X.reshape(1, 400, 400, 1)
    Y = Y.reshape(1, 400, 400, 2)

    model.fit(x=X,
              y=Y,
              batch_size=1,
              epochs=1000)