Изменение размера входного изображения в слое Keras Lambda

Question

Я хотел бы, чтобы моя модель keras изменила размер входного изображения с помощью cv2 или аналогичного.

Я видел использование ImageGenerator, но я бы предпочел написать собственный генератор и просто изменить размер изображения в первом слое с помощью keras.layers.core.Lambda.

Answer 1

Если вы используете бэкэндер tensorflow, вы можете использовать функцию tf.image.resize_images(), чтобы изменить размер изображений в слое Lambda.

Вот небольшой пример, чтобы продемонстрировать то же самое:

import numpy as np 
import scipy.ndimage 
import matplotlib.pyplot as plt 

from keras.layers import Lambda, Input 
from keras.models import Model 
from keras.backend import tf as ktf 


# 3 channel images of arbitrary shape 
inp = Input(shape=(None, None, 3)) 
try: 
    out = Lambda(lambda image: ktf.image.resize_images(image, (128, 128)))(inp) 
except : 
    # if you have older version of tensorflow 
    out = Lambda(lambda image: ktf.image.resize_images(image, 128, 128))(inp) 

model = Model(input=inp, output=out) 
model.summary() 

X = scipy.ndimage.imread('test.jpg') 

out = model.predict(X[np.newaxis, ...]) 

fig, Axes = plt.subplots(nrows=1, ncols=2) 
Axes[0].imshow(X) 
Axes[1].imshow(np.int8(out[0,...])) 

plt.show()