OpenCV 이미지 다루기 (읽기, 출력, 저장)

DRF APIView를 통해 업로드 받은 이미지를 다뤄보자

딥러닝을 다루는 프로젝트 진행 중 딥러닝으로 변환 된 이미지를 새롭게 저장하기 위해 이미지 읽고, 저장하기 위한 방법에 대해 공부하고 정리한 내용입니다.

 

먼저 패키지를 인스톨 받은 후에

pip install opencv-python

cv2를 import

import cv2

 

이미지 읽어오기

imread() : 이미지 읽어오기

cv.2imread(이미지 경로)를 해주면 해당 경로의 이미지를 읽어올 수 있다.

def img_transfer():
    image_path = "/media/sample.jpg"
    img = cv2.imread(image_path)

 

이미지 출력하기

imshow() : 이미지 출력하기(화면에 띄우기)

imshow(윈도우 창 이름, imread()의 리턴 값)을 통해 이미지를 출력하며 새로운 윈도우 창을 띄워 출력하게 된다.

def img_transfer():
    image_path = "/media/sample.jpg"
    img = cv2.imread(image_path)
    
    cv2.imshow('image',img)	# 'image'는 윈도우 창 이름
    cv2.waitKey(0)

 

이미지 저장하기

imwrite() : 이미지 저장하기

이미지를 수정하거나 변경한 후에 새롭게 저장하기 위해 imwrite를 활용하여 저장할 수 있다.

imwrite(경로, 저장할 이미지)

def img_transfer():
    image_path = "/media/sample.jpg"
    img = cv2.imread(image_path)
    
    cv2.imshow('image',img)	# 'image'는 윈도우 창 이름
    cv2.waitKey(0)
    
    cv2. imwrite("/media/arter_image/result.jpg", img)