본문 바로가기
  • On the ball
Data/TensorFlow

Image 읽고 특정 영역자르기 / 회전

by pub-lican-ai 2017. 3. 15.
반응형

[필요 라이브러리]

matplotlib(plot), pillow(다양한 포멧의 이미지 핸들링)


아래와 같이 pip을 통해 설치 가능하다

>pip install matplotlib

>pip istall pillow


네트워크 환경에 따라 아래 옵션 사용

--proxy http://PROXYSERVER:PROXYPORT --trusted-host pypi.python.org



pip을 새로 설치한 경우 

pip3.6 install matplotlib


[이미지 읽기 및 자르기]

import tensorflow as tf

import matplotlib.image as mp_image

filename ="/home/sds/Desktop/Penguins.jpg"

input_image = mp_image.imread(filename)

print(input_image.ndim) #3차원의

print(input_image.shape) #사이즈

3

(768, 1024, 3)

import matplotlib.pyplot as plt

plt.imshow(input_image)

plt.show()

penguins.jpg 나타남


my_image = tf.placeholder("uint8",[None,None,3])

slice = tf.slice(my_image,[10,0,0],[200,-1,-1])

with tf.Session() as session:

    result = session.run(slice,feed_dict={my_image:input_image})

    print(result.shape)

plt.imshow(result)

plt.show()

(200, 1024, 3)


[이미지 회전]

#이미지 변환 x = tf.Variable(input_image,name='x') model = tf.global_variables_initializer() with tf.Session() as session: x= tf.transpose(x, perm=[1,0,2]) #perm에서 x,y를 바꿈 0,1,2 하면 기존과 동일하게 나타남 session.run(model) result=session.run(x) print(result.shape) plt.imshow(result) plt.show()

(1024, 768, 3)


출처: 책 텐서플로 입문 _ 잔카를로 자코네

반응형