COLAB에서 OpenCV 함수 사용하기

이전 포스팅은 구글 드라이브와 연동에 대해서 포스팅 했다 이번에는 OpenCV 함수를 사용하는 방법에 대해서 정리 하도록 한다.

overface.tistory.com/572

COLAB 구글드라이브(Google Drive) 연동

 

이미지를 표기하기 위해서 구글드라이브에 lenna.bmp 파일을 업로드 하였다.

 

코드+ 탭을 클릭하여 소스코드를 아래와 같이 입력하였다.

기본 경로는 /cotent/gdrive/My Drive 이다. lenna.bmp 사진파일을 Colab Notebooks 하위 폴더에 업로드를 하였기 때문에 아래와 같이 전체 경로를 입력 하였다.

 

위코드를 실행 하면은 아래와 같은 오류가 나온다.

 

Jupyter notebook 에서 사용하는 imshow 함수가 호환되지 않는 문제로 발생한다. 아래 코드로 수정한다.

#최상단에 import 를 추가한다.
from google.colab.patches import cv2_imshow

#cv2.imshow 함수를 cv2_imshow로 변경한다.
cv2_imshow(img)

 

수정 된 코드

from google.colab.patches import cv2_imshow
import sys
import numpy as np
import cv2

src = cv2.imread('/content/gdrive/My Drive/Colab Notebooks/lenna.bmp')

if src is None:
    print('Image load failed!')
    sys.exit()

dst = cv2.bilateralFilter(src, -1, 10, 5)

cv2_imshow(src)
cv2_imshow(dst)
cv2.waitKey()

cv2.destroyAllWindows()

 

실행 결과