Programming/python-computer vision

[python] OpenCV 이미지 시각화 코드 모음 (cv2.imshow) - vscode, jupyter, colab

방황하는 데이터불도저 2023. 12. 24. 00:59

1. 비주얼 스튜디오 코드와 같은 환경에서 .py 파일을 터미널에 실행시키는 방식

def print_image(image):
    if isinstance(image, str):
        image = cv2.imread(image)

    cv2.imshow("image", image)
    cv2.waitKey(0)
    cv2.destroyAllWindows()

 

2. 주피터 환경에서 (.ipynb 파일의 한 셀에서) 이미지를 보고싶은 경우

 

  1) OpenCV : viewer (window)가 뜨는 방식

def print_image(image):
    if isinstance(image, str):
        image = cv2.imread(image)
        
    cv2.namedWindow("Sample Image", cv2.WINDOW_NORMAL)
    cv2.resizeWindow('Sample Image', 800, 800)
    cv2.imshow('Sample Image', image)
    cv2.waitKey()
    cv2.destroyAllWindows()

 

  2) matplotlib 활용하기 : execution cell에 보이는 방식

from matplotlib import pyplot as plt

def print_image(image):
    if isinstance(image, str):
        image = cv2.imread(image)

    plt.imshow(cv2.cvtColor(image, cv2.COLOR_BGR2RGB))
    plt.title('Sample image')
    plt.xticks([]), plt.yticks([])
    plt.show()

 

3. colab에서 실행시키는 경우

from google.colab.patches import cv2_imshow

def print_image(image):
    if isinstance(image, str):
        image = cv2.imread(image)
        
    cv2_imshow(image)
    cv2.waitKey()
    cv2.destroyAllWindows()