프로그래밍/Python

[python] OpenCV - 이미지의 폭과 높이 구하기

채윤아빠 2021. 5. 31. 09:19
728x90
반응형

개요

OpenCV에서 이미지에 대한 폭(width)과 높이(height)를 구하는 방법을 알아 보겠습니다.


이미지의 폭과 높이 구하는 방법

이미지에 대한 폭과 높이는 다음과 같이 img.shape로부터 구할 수 있습니다.

from cv2 import imread

img = imread('image.jpg')

# 이미지 크기 구하기 ; 높이, 폭 순서 주의
height, width = img.shape[:2]

print(f"이미지 폭: {width} / 이미지 높이: {height}")

# 컬러/흑백 구분
if len(img.shape) == 2:
    print("흑백 이미지입니다.")
else:
    print("컬러 이미지입니다.")

위 예제 코드와 같이 "height, width = img.shape[:2]" 이렇게 수행하는 것이 보다 안정적입니다.
흑백 이미지의 경우 shape에 3번째 channel 정보가 포함되어 있지 않기 때문입니다.


참고자료