[python] OpenCV를 사용하여 이미지 저장하기

22 Nov 2023

python

이미지 처리 및 컴퓨터 비전 작업을 할 때 Python에서 많이 사용되는 라이브러리 중 하나가 OpenCV입니다. 이 라이브러리를 사용하여 이미지를 저장하는 방법을 알아보겠습니다.

1. OpenCV 설치하기

OpenCV를 사용하기 위해 먼저 설치해야 합니다.

pip install opencv-python

위 명령어를 사용하여 OpenCV를 설치할 수 있습니다.

2. 이미지 저장하기

이미지를 저장하기 위해서는 cv2.imwrite() 함수를 사용해야 합니다.

import cv2

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

# 이미지 저장
cv2.imwrite('saved_image.jpg', image)

imread() 함수를 사용하여 이미지를 읽어온 후, imwrite() 함수를 사용하여 저장할 수 있습니다. 첫 번째 파라미터는 저장할 이미지 파일의 이름이며, 두 번째 파라미터는 이미지 객체입니다.

3. 이미지 형식 지정하기

imwrite() 함수를 사용하여 이미지를 저장할 때, 이미지 형식을 지정할 수도 있습니다. 형식을 지정하지 않으면 기본적으로 JPEG 형식으로 저장됩니다.

import cv2

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

# 이미지를 PNG 형식으로 저장
cv2.imwrite('saved_image.png', image, [cv2.IMWRITE_PNG_COMPRESSION, 0])

위 코드에서 cv2.IMWRITE_PNG_COMPRESSION은 PNG 형식으로 저장할 때 압축 비율을 설정하는 상수입니다. 0은 압축률이 없다는 의미입니다.

4. 이미지 저장 성공 여부 확인하기

imwrite() 함수를 사용하여 이미지를 저장하면, 반환값으로 성공 여부를 확인할 수 있습니다.

import cv2

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

# 이미지 저장
result = cv2.imwrite('saved_image.jpg', image)

if result:
    print("이미지 저장 성공!")
else:
    print("이미지 저장 실패!")

imwrite() 함수는 저장에 성공하면 True를, 실패하면 False를 반환합니다.

참고 자료

OpenCV 공식 문서: https://docs.opencv.org/
코딩 튜토리얼: https://www.opencv-python-tutroals.readthedocs.io/