본문 바로가기
OpenCV

[OpenCV] 2. 이미지의 shape/type/color

by pulluper 2024. 4. 10.
반응형

이번에는 cv2 이미지의 shape/type/color 에 대하여 알아보겠습니다. 

 

1) 이미지의 shape 확인하기

h, w, c = image.shape

torch 와 다르게 h, w, c 의 순서로 shape을 가진다.  

 

2) 이미지의 type 확인

이미지 타입은 default 로 uint8이다. 따라서 range 는 0~255

print(image.dtype)

 

    2-1) 이미지의 type 변경하기 

    만약 float32로 타입을 변환 시키는 부분은 다음과 같다.  

    주의할 점은 255 곱하고 나누는 순서가 다른 부분이다.  

# uint8 --> float32 / 64
image = image.astype(np.float32) / 255

# float 32 --> uint8
image = (image * 255).astype(np.uint8)

 

3) color 포맷

opencv의 default color 포맷은 bgr이다. 

초기 개발이 카메라 산업에서 많이 쓰이던 bgr이라 이를 따라갔다. 

 

    3-1) bgr 에서 rgb 로 변환하기

image = cv2.cvtColor(image, cv2.COLOR_BGR2RGB)

 

------------------------------------------------- review ------------------------------------------------- 

 

1. 이미지 shape(h, w, c)얻기 

2. 이미지의 type 변경 (uint8 --> float32)

3. 이미지의 type 변경 (float32 --> uint8)

4. bgr 에서 rgb 로 변환하기

반응형

댓글