[DIP] cv2.imwrite

cv2.imwrite는 OpenCV 라이브러리에서 제공하는 "이미지 데이터를 지정한 파일로 저장하는 기능"을 수행.

 

OpenCV 는 다양한 image format을 지원하며, 저장할 파일명의 확장자에 따라 알아서 encoding을 수행.

---

사용법

cv2.imwrite(filepath, img, params=None) -> bool

• filepath:
  • 저장할 파일의 경로와 이름을 나타내는 str 객체.
  • 파일 extension을 통해 저장될 이미지의 format이 결정됨.
  • 예: 'image.png', 'image.jpg'
• img:
  • 저장할 이미지 데이터.
  • 일반적으로 NumPy 배열 형태로 제공 (dtype를 np.uint8로 주로 사용.)
• params (optional):
  • 이미지 저장 시 추가적인 파라미터를 설정 담당.
  • key와 value를 가진 list객체로 제공.
  • 이미지 포맷에 따라 압축률 등을 지정하는데 사용됨.

• return value 확인:
  • 함수는 저장 성공 시 True,
  • 실패 시 False를 반환

---

예제

from skimage import data
import cv2

# 이미지 생성
cat = data.chelsea()
print(f"{type(cat) = }")

#Color Space 변경
cat_cv = cv2.cvtColor(cat, cv2.COLOR_RGB2BGR)
print(f"{type(cat_cv) = }")

# 결과 이미지 저장
success = cv2.imwrite('output.png', cat_cv)
# 저장 성공 여부 확인
if success:
    print('이미지 저장에 성공했습니다.')
else:
    print('이미지 저장에 실패했습니다.')

---

추가 파라미터 사용

params를 사용하여 이미지 저장 옵션을 설정 가능.

JPEG 품질 설정 (0 ~ 100, 기본값: 95)

cv2.imwrite('output.jpg', img, [cv2.IMWRITE_JPEG_QUALITY, 90])

PNG 압축 레벨 설정 (0 ~ 9, 기본값: 3)

cv2.imwrite('output.png', img, [cv2.IMWRITE_PNG_COMPRESSION, 5])

---

주의사항

• file path 유효성:
  • 저장하려는 path(디렉토리)가 존재하지 않으면 파일 저장에 실패.
  • 저장 전에 유효한 path인지 확인할 것.
• return value 확인:
  • 함수는 저장 성공 시 True,
  • 실패 시 False를 반환

---

같이보면 좋은 자료

2022.12.05 - [Programming/DIP] - [DIP] Image Format (summary)

[DIP] Image Format (summary)

 

https://dsaint31.me/mkdocs_site/OS/file/

BME

 

https://ds31x.tistory.com/25

[Python] os 모듈의 함수들 : file과 directory 관련

https://ds31x.tistory.com/464

[torchvision] torchvision.utils.save_image and torchvision.io.encode_jpeg, torchvision.io.encode_png

---

2024.09.22 - [Programming/DIP] - [DIP] Image 다루기: cv2-기본편 1 (Summary)

[DIP] Image 다루기: cv2-기본편 1 (Summary)

---