方法一,输入一个常量,然后在tensorboard中查看。
方法二
sess = tf.Session(config=tfconfig)
import cv2
import numpy as np
img = cv2.imread('./image.jpg')
print('img shape =', img.shape)
operations = tf.get_default_graph().get_operations()
op_names = ['Placeholder']
for op in operations:
if op.name.split('/')[-1] == 'Relu':
# print(op.name)
op_names.append(op.name)
#op_names = ['Placeholder',
# 'vgg_16/conv1/conv1_2/Relu',
# 'vgg_16/conv2/conv2_2/Relu',
# 'vgg_16/conv3/conv3_2/Relu',
# 'vgg_16/conv4/conv4_3/Relu',
# 'vgg_16/conv5/conv5_3/Relu',
# 'vgg_16_text/deconv_1/Relu',
# 'vgg_16_text/deconv_2/Relu',
# 'vgg_16_text/deconv_3/Relu',
# 'vgg_16_text/conv2-4/conv2-4_3/Relu']
for op in op_names:
aa = sess.run(tf.get_default_graph().get_tensor_by_name(op + ':0'), feed_dict={image_tensor:img[np.newaxis,...]}) # 单张图是3维的,加一个轴
print(op, aa.shape)
sess.close()