[PYTHON/TENSORFLOW] transpose 함수 : 이미지 회전하기

import matplotlib.image as img
import matplotlib.pyplot as pp
import tensorflow as tf

fileName = "d:\\sample.png"

ndarray = img.imread(fileName)
variable = tf.Variable(ndarray, name = 'x')
model = tf.global_variables_initializer()

with tf.Session() as sess:
    variable = tf.transpose(variable, perm = [1, 0, 2])
    sess.run(model)
    transposeNDArray = sess.run(variable)
    pp.imshow(transposeNDArray)
    pp.show()

 

728x90

 

[결과]