在inception/inception_eval.py中,我想打印标签和logits。
我尝试了一些方法,但他们不能工作 我如何获得这些张量的数量?
with tf.Graph().as_default():
# Get images and labels from the dataset.
# with tf.Session() as sess:
# images, labels = image_processing.inputs(dataset)
# print ("lable")
# sess.run (labels.eval())
images, labels = image_processing.inputs(dataset)
# sess = tf.Session()
# print("lable")
# print (labels)
# Number of classes in the Dataset label set plus 1.
# Label 0 is reserved for an (unused) background class.
num_classes = dataset.num_classes() + 1
# Build a Graph that computes the logits predictions from the
# inference model.
logits, _ = inception.inference(images, num_classes)
print ("logits")
# print (tf.cast(logits,tf.float32).eval())
# print ("_")
# print (tf.cast(_,tf.float32).eval())
答案 0 :(得分:0)
如果您要检查inception_model文件,您会看到inference function返回两个张量:
Logits。二维浮子张量。
辅助Logits。 2-D float Tensor of 方头。仅用于训练。
所以你只需要创建一个会话并在这个会话中运行这个张量。