📅  最后修改于: 2022-03-11 14:45:24.766000             🧑  作者: Mango
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)