CNN批处理具有不同大小的图像

Question

我恢复了一个用于面部检测的预训练模型，该模型一次拍摄一张图像并返回边界框。如果这些图像具有不同的尺寸，该如何拍摄一批图像？

Answer 1

您可以使用tf.image.resize_images方法来实现。根据{{​​3}} tf.image.resize_images：

  

使用指定的方法将图像调整大小。

     

如果未调整尺寸的图像的原始宽高比将失真   与尺寸相同。为避免变形，请参见   tf.image.resize_image_with_pad。

如何使用它？

import tensorflow as tf
from tensorflow.python.keras.models import Model

x = Input(shape=(None, None, 3), name='image_input')
resize_x = tf.image.resize_images(x, [32,32])
vgg_model = load_vgg()(resize_x)
model = Model(inputs=x, outputs=vgg_model.output)
model.compile(...)
model.predict(...)