numpy和tensor的转换

Posted 2023-05-09

参考技术A numpy转tensor

tensor转numpy

（数组张量相互转换）numpy2tensor and tensor2numpy

import tensorflow as tf
import numpy as np
 
path2='results/HXresults/1.png'
path3='test_imgs/mri/1.png'

MRI=(imread(path3)/255.0).astype(np.float32)
X=(imread(path2)/255.0).astype(np.float32)

MRI = np.expand_dims(MRI, axis = 0)
MRI = np.expand_dims(MRI, axis = -1)
X = np.expand_dims(X, axis = 0)
# numpy2tensor
X = tf.constant(X)

X_YCBCR = rgb2ycbcr(X)

# numpy2tensor
sess=tf.Session()
MRI =MRI.eval(session=sess)
X_YCBCR=X_YCBCR.eval(session=sess)

np_img = vifp_tf(MRI, X_YCBCR[:,:,:,0])
print(np_img)

# tensor2numpy
tf_img = tf.convert_to_tensor(a)
print(tf_img)
with tf.Session() as sess: # 查看张量内容
    print(sess.run(tf_img))