[python] OpenCV로 이미지 저장하기

07 Dec 2023

python

이미지 처리와 컴퓨터 비전 작업을 위해 널리 사용되는 OpenCV는 Python에서도 많이 사용됩니다. OpenCV를 사용하여 이미지를 로드하고 수정한 후 저장하는 방법을 알아보겠습니다.

이미지 로드하기

먼저, OpenCV를 사용하여 이미지를 로드해야 합니다. imread 함수는 이미지 파일을 읽어들여 numpy 배열로 반환합니다. 다음은 예시입니다.

import cv2

# 이미지 파일 로드
image = cv2.imread('image.jpg')

# 이미지 파일이 존재하지 않는 경우 예외 처리
if image is None:
    print("이미지 파일을 찾을 수 없습니다.")

이미지 수정하기

이미지를 로드한 후, 필요한 작업을 수행하여 이미지를 수정할 수 있습니다. 다양한 작업을 수행할 수 있지만, 이 예시에서는 이미지를 그레이스케일로 변환하는 방법을 다루겠습니다.

# 이미지를 그레이스케일로 변환
gray_image = cv2.cvtColor(image, cv2.COLOR_BGR2GRAY)

이미지 저장하기

이미지를 수정한 후, imwrite 함수를 사용하여 이미지를 저장할 수 있습니다. 다음은 예시입니다.

# 수정된 이미지를 저장
cv2.imwrite('gray_image.jpg', gray_image)

전체 코드

import cv2

# 이미지 파일 로드
image = cv2.imread('image.jpg')

# 이미지 파일이 존재하지 않는 경우 예외 처리
if image is None:
    print("이미지 파일을 찾을 수 없습니다.")

# 이미지를 그레이스케일로 변환
gray_image = cv2.cvtColor(image, cv2.COLOR_BGR2GRAY)

# 수정된 이미지를 저장
cv2.imwrite('gray_image.jpg', gray_image)

이제 OpenCV를 사용하여 이미지를 로드하고 수정한 후 저장하는 방법을 알게 되었습니다. 이를 통해 이미지 처리 작업을 더욱 효율적으로 수행할 수 있습니다.

참고 문서:

OpenCV 공식 문서: https://docs.opencv.org/
OpenCV Python 공식 문서: https://docs.opencv.org/4.4.0/d3/dc1/tutorial_basic_linear_transform.html