본문 바로가기

지식생활/A.I.

카페2(caffe2) 튜토리얼 Image Pre-Processing (Rotation Mirroring)

## 카페2 공부를 위해 튜토리얼 확인

## 공식 페이지에 있는 소스는 일부분만 표시되어있어 실행시 문제없도록

예제 부분을 완성해보았습니다.

caffe2 사이트에서는 로테이션과 미러 각각의 이미지를 사용하였지만

확인하기 쉽게 하나의 이미지에서 

90도회전해서 출력하고 다시 좌우반전해서 출력하도록 하였습니다.


############################################################################


import skimage

import skimage.io as io

import numpy as np

import matplotlib.pyplot as pyplot


ROTATED_IMAGE = "https://upload.wikimedia.org/wikipedia/commons/e/e1/Ananas.jpg"


## 90도회전 시킴

imgRotated = skimage.img_as_float(skimage.io.imread(ROTATED_IMAGE)).astype(np.float32)

imgRotated = np.rot90(imgRotated)


pyplot.figure()

pyplot.imshow(imgRotated)

pyplot.axis('on')

pyplot.title('Rotated image\nbig-sun.tistory.com')


MIRROR_IMAGE = "https://upload.wikimedia.org/wikipedia/commons/e/e1/Ananas.jpg"


## 90도 회전된 이미지를 사용하여 좌우반전

# imgMirror = skimage.img_as_float(skimage.io.imread(MIRROR_IMAGE)).astype(np.float32)

imgMirror = np.fliplr(imgRotated)


pyplot.figure()

pyplot.imshow(imgMirror)

pyplot.axis('on')

pyplot.title('Mirror image\nbig-sun.tistory.com')


pyplot.show()



############################################################################