我正在尝试确定是否需要通过Tensorflow模型传播批处理输入图像的元数据,或者是否可以将其保存在模型外部。
对于以下代码,是否保证predictions
与batch_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)