Tensorflow批处理预测是否以与批处理输入相同的顺序返回?

时间:2018-08-23 22:46:26

标签: python tensorflow

我正在尝试确定是否需要通过Tensorflow模型传播批处理输入图像的元数据,或者是否可以将其保存在模型外部。

对于以下代码,是否保证predictionsbatch_imgs的顺序相同?

session = tf.Session()
inputs = tf.placeholder(dtype=tf.float32, shape=(None, height, width, channel_size), name='input_img')
model = SomeModel(self.inputs, num_classes=num_classes, is_training=False)
logits, _ = model.build_model()
pred = tf.nn.softmax(logits, name='logits_to_softmax')
feed_dict = {inputs: batch_imgs}
predictions = TFSessionsession.run(pred, feed_dict=feed_dict)

0 个答案:

没有答案