predicted_outputs = ann.predict(training_inputs)predicted_image_array = np.zeros_like(image_array)i = 0for row,rgbs in enumerate(predicted_image_array): for column in range(len(rgbs)): r,g,b = predicted_outputs[i] predicted_image_array[row][column] = [r*255,g*255,b*255] i += 1Image.fromarray(predicted_image_array)