OpenCV

Python(파이썬)에서 OpenCV를 이용해 이미지 불러오기 imread(), imshow()

HooSL 2021. 12. 27. 17:47

먼저 필요한 numpy와 opencv를 import 해줍니다.

import cv2
import numpy as np
img_file = '사진경로' # 예시) data/images/sample.jpg

#opencv 로 이미지 열기 컬러이미지
image = cv2.imread(img_file,cv2.IMREAD_COLOR)

#이미지가 정상인지 체크하는 코드
if image is None:
    print('이미지를 불러올 수 없습니다.') # image 변수가 없다면 출력
else:
    print(image.shape) # 있다면 image의 shape를 터미널에 출력
    
    
gray_image = cv2.cvtColor(image, cv2.COLOR_BGR2GRAY)

#컬러 이미지 출력
cv2.imshow('color',image)

#흑백 이미지 출력
cv2.imshow('gray scale',gray_image)

위 코딩중 imshow함수는 화면에 표시하는 함수인데 실행되었다가 바로 종료되는걸 볼 수 있습니다.

cpu가 imshow를 실행하고 그 다음 라인을 실행하는데 아래라인은 아무것도 없어서 종료가 됩니다.

그래서 우리 눈으로 확인하기 위해서는 cpu의 코드 실행을 잠시 멈추게 해줘야 합니다.

cv2.waitKey(0) # 0은 특정키를 입력하지 않는 이상 계속 기다려라
cv2.destroyAllWindows() # 아무키나 누르면 cv2가 실행시킨 모든창들 종료해라