test_batch = tf.stack([img_tf1, img_tf2]) # or test_batch = np.vstack([imgs_tf1, img_tf2]) test_batch_size = 32 predictions = model.predict(test_batch, batch_size=test_batch_size)